Tests/MSFT_xServiceResource.TestHelper.psm1
|
<#
.SYNOPSIS Creates a service binary file. .PARAMETER ServiceName The name of the service to create the binary file for. .PARAMETER ServiceCodePath The path to the code for the service to create the binary file for. .PARAMETER ServiceDisplayName The display name of the service to create the binary file for. .PARAMETER ServiceDescription The description of the service to create the binary file for. .PARAMETER ServiceDependsOn Dependencies of the service to create the binary file for. .PARAMETER ServiceExecutablePath The path to write the service executable to. #> function New-ServiceBinary { [CmdletBinding()] param ( [Parameter(Mandatory = $true)] [String] $ServiceName, [Parameter(Mandatory = $true)] [String] $ServiceCodePath, [Parameter(Mandatory = $true)] [String] $ServiceDisplayName, [Parameter(Mandatory = $true)] [String] $ServiceDescription, [Parameter(Mandatory = $true)] [String] $ServiceDependsOn, [Parameter(Mandatory = $true)] [String] $ServiceExecutablePath ) if (Get-Service $ServiceName -ErrorAction Ignore) { Stop-Service $ServiceName -ErrorAction SilentlyContinue Remove-TestService -ServiceName $ServiceName -ServiceExecutablePath $ServiceExecutablePath } $fileText = Get-Content $ServiceCodePath -Raw $fileText = $fileText.Replace('TestServiceReplacementName', $ServiceName) $fileText = $fileText.Replace('TestServiceReplacementDisplayName', $ServiceDisplayName) $fileText = $fileText.Replace('TestServiceReplacementDescription', $ServiceDescription) $fileText = $fileText.Replace('TestServiceReplacementDependsOn', $ServiceDependsOn) Add-Type $fileText ` -OutputAssembly $ServiceExecutablePath ` -OutputType WindowsApplication ` -ReferencedAssemblies @('System.ServiceProcess', 'System.Configuration.Install') } <# .SYNOPSIS Creates a new service for testing. .PARAMETER ServiceName The name of the service to create. .PARAMETER ServiceCodePath The path to the code for the service to create. .PARAMETER ServiceDisplayName The display name of the service to create. .PARAMETER ServiceDescription The description of the service to create. .PARAMETER ServiceDependsOn Dependencies of the service to create. .PARAMETER ServiceExecutablePath The path to write the service executable to. #> function New-TestService { [CmdletBinding()] param ( [Parameter(Mandatory = $true)] [String] $ServiceName, [Parameter(Mandatory = $true)] [String] $ServiceCodePath, [Parameter(Mandatory = $true)] [String] $ServiceDisplayName, [Parameter(Mandatory = $true)] [String] $ServiceDescription, [Parameter(Mandatory = $true)] [String] $ServiceDependsOn, [Parameter(Mandatory = $true)] [String] $ServiceExecutablePath ) New-ServiceBinary ` -ServiceName $ServiceName ` -ServiceCodePath $ServiceCodePath ` -ServiceDisplayName $ServiceDisplayName ` -ServiceDescription $ServiceDescription ` -ServiceDependsOn $ServiceDependsOn ` -ServiceExecutablePath $ServiceExecutablePath if (-not (Test-Path $ServiceExecutablePath)) { throw 'Failed to create service executable file.' } $configurationName = 'TestServiceConfig' $configurationPath = Join-Path -Path (Get-Location) -ChildPath $ServiceName Configuration $configurationName { Import-DscResource -ModuleName xPSDesiredStateConfiguration xService Service1 { Name = $ServiceName Path = $ServiceExecutablePath DisplayName = $ServiceDisplayName Description = $ServiceDescription Dependencies = $ServiceDependsOn StartupType = 'Manual' BuiltInAccount = 'LocalSystem' State = 'Stopped' Ensure = 'Present' } } & $configurationName -OutputPath $configurationPath Start-DscConfiguration -Path $configurationPath -Wait -Force -Verbose } <# .SYNOPSIS Retrieves the path to the install utility. #> function Get-InstallUtilPath { [CmdletBinding()] param () if ($env:Processor_Architecture -ieq 'amd64') { $frameworkName = 'Framework64' } else { $frameworkName = 'Framework' } return Join-Path -Path (Resolve-Path "$env:WinDir\Microsoft.Net\$frameworkName\v4*") ` -ChildPath 'installUtil.exe' } <# .SYNOPSIS Removes a service. .PARAMETER ServiceName The name of the service to remove. .PARAMETER ServiceExecutablePath The path to the executable of the service to remove. #> function Remove-TestService { [CmdletBinding()] param ( [Parameter(Mandatory = $true)] [String] $ServiceName, [Parameter(Mandatory = $true)] [String] $ServiceExecutablePath ) if (Get-Service -Name $ServiceName -ErrorAction SilentlyContinue) { $installUtility = Get-InstallUtilPath & $installUtility /u $ServiceExecutablePath } Remove-Item $ServiceExecutablePath -Force -ErrorAction SilentlyContinue Remove-Item *.InstallLog -Force -ErrorAction SilentlyContinue Remove-Item $ServiceName -Force -Recurse -ErrorAction SilentlyContinue } function Get-InvalidArgumentRecord { [CmdletBinding()] param ( [Parameter(Mandatory)] [ValidateNotNullOrEmpty()] [System.String] $ErrorId, [Parameter(Mandatory)] [ValidateNotNullOrEmpty()] [System.String] $ErrorMessage ) $exception = New-Object -TypeName System.InvalidOperationException ` -ArgumentList $ErrorMessage $errorCategory = [System.Management.Automation.ErrorCategory]::InvalidOperation $errorRecord = New-Object -TypeName System.Management.Automation.ErrorRecord ` -ArgumentList $exception, $ErrorId, $errorCategory, $null return $errorRecord } Export-ModuleMember -Function ` New-ServiceBinary, ` New-TestService, ` Remove-TestService, ` Get-InvalidArgumentRecord |