Tests/cTentacleAgent.Tests.ps1
|
#requires -Version 4.0 $moduleName = Split-Path ($PSCommandPath -replace '\.Tests\.ps1$', '') -Leaf $modulePath = Split-Path $PSCommandPath -Parent $modulePath = Resolve-Path "$PSCommandPath/../../DSCResources/$moduleName/$moduleName.psm1" $module = $null try { $prefix = [guid]::NewGuid().Guid -replace '-' $module = Import-Module $modulePath -Prefix $prefix -PassThru -ErrorAction Stop InModuleScope $module.Name { # Get-Service is not available on mac/unix systems - fake it $getServiceCommand = Get-Command "Get-Service" -ErrorAction SilentlyContinue if ($null -eq $getServiceCommand) { function Get-Service {} } Describe 'cTentacleAgent' { BeforeEach { [System.Diagnostics.CodeAnalysis.SuppressMessageAttribute('PSUseDeclaredVarsMoreThanAssignments', '')] $desiredConfiguration = @{ Name = 'Stub' Ensure = 'Present' State = 'Started' } } Context 'Confirm-RegistrationParameter' { It 'Throws if RegisterWithServer is false but environment provided' { { Confirm-RegistrationParameter -RegisterWithServer $False -Environments @('My Env') } | Should Throw "Invalid configuration requested" } It 'Throws if RegisterWithServer is false but roles provided' { { Confirm-RegistrationParameter -RegisterWithServer $False -Roles @('app-server') } | Should Throw "Invalid configuration requested" } It 'Throws if RegisterWithServer is false but tenants provided' { { Confirm-RegistrationParameter -RegisterWithServer $False -Tenants @('Jim-Bob') } | Should Throw "Invalid configuration requested" } It 'Throws if RegisterWithServer is false but tenant Tags provided' { { Confirm-RegistrationParameter -RegisterWithServer $False -TenantTags @('CustomerType/VIP', 'Hosting/OnPrem') } | Should Throw "Invalid configuration requested" } It 'Throws if RegisterWithServer is false but policy provided' { { Confirm-RegistrationParameter -RegisterWithServer $False -Policy "my policy" } | Should Throw "Invalid configuration requested" } It 'Throws if RegisterWithServer is true but no OctopusServerUrl provided' { { Confirm-RegistrationParameter -RegisterWithServer $true -OctopusServerUrl $null } | Should Throw "Invalid configuration requested" } It 'Throws if RegisterWithServer is true but no ApiKey provided' { { Confirm-RegistrationParameter -RegisterWithServer $true -OctopusServerUrl "https://example.octopus.com" -ApiKey $null } | Should Throw "Invalid configuration requested" } It 'Does not throw if RegisterWithServer is false and environment provided as empty string' { Confirm-RegistrationParameter -RegisterWithServer $False -Environments "" } It 'Does not throw if RegisterWithServer is false and environment provided as empty array' { Confirm-RegistrationParameter -RegisterWithServer $False -Environments @() } It 'Does not throw if RegisterWithServer is false and environment provided as array with empty element' { Confirm-RegistrationParameter -RegisterWithServer $False -Environments @('') } It 'Does not throw if RegisterWithServer is false and no environment provided' { Confirm-RegistrationParameter -RegisterWithServer $False } It 'Does not throw if RegisterWithServer is true and OctopusServerUrl and ApiKey provided' { Confirm-RegistrationParameter -RegisterWithServer $true -OctopusServerUrl "https://example.octopus.com" -ApiKey "API-1234" } } Context 'Confirm-RequestedState' { It 'Throws if RegisterWithServer is false but environment provided' { { Confirm-RequestedState -Ensure "Absent" -State "Started" } | Should Throw "Invalid configuration requested" } } Context 'Get-TargetResource' { Mock Get-ItemProperty { return @{ InstallLocation = "c:\Octopus\Tentacle\Stub" }} Mock Get-Service { return @{ Status = "Running" }} It 'Returns the proper data' { $config = Get-TargetResource -Name 'Stub' $config.GetType() | Should Be ([hashtable]) $config['Name'] | Should Be 'Stub' $config['Ensure'] | Should Be 'Present' $config['State'] | Should Be 'Started' } It "Throws if we specify a null or invlid CustomHostName" { { Get-TargetResource -Name "Stub" -PublicHostNameConfiguration "Custom" } | Should Throw "invalid or null" { Get-TargetResource -Name "Stub" -PublicHostNameConfiguration "Custom" -CustomPublicHostName " " } | Should Throw "invalid or null" { Get-TargetResource -Name "Stub" -PublicHostNameConfiguration "Custom" -CustomPublicHostName "mydnsname" } | Should Not Throw } } Context 'Test-TargetResource' { $response = @{ Ensure="Absent"; State="Stopped" } Mock Get-TargetResource { return $response } It 'Returns True when Ensure is set to Absent and Tentacle does not exist' { $desiredConfiguration['Ensure'] = 'Absent' $desiredConfiguration['State'] = 'Stopped' $response['Ensure'] = 'Absent' $response['State'] = 'Stopped' Test-TargetResource @desiredConfiguration | Should Be $true } It 'Returns True when Ensure is set to Present and Tentacle exists' { $desiredConfiguration = @{ Ensure = 'Present' State = 'Started' OctopusServerUrl = 'http://fakeserver1' APIKey = 'API-GRKUQFCFIJM7G2RJM3VMRW43SK' Name = 'Tentacle1' Environments = @() Roles = @() WorkerPools = @() } $response['Ensure'] = 'Present' $response['State'] = 'Started' # Declare mock objects $mockMachine = @{ Name = "MockMachine" EnvironmentIds = @() WorkerPools = @() Roles = @() } # Declare mock calls Mock Get-MachineFromOctopusServer {return $mockMachine} Mock Get-APIResult {return [string]::Empty} Mock Get-TentacleThumbprint { return "ABCDE123456" } Test-TargetResource @desiredConfiguration | Should Be $true } } Context 'Set-TargetResource' { #todo: more tests Mock Invoke-AndAssert { return $true } Mock Install-Tentacle { return $true } } } Describe "Testing Get-MyPublicIPAddress" { $testIP = "54.121.34.56" Mock Invoke-RestMethod { return $testIP } Context "First option works" { It "Should return an IPv4 address" { Get-MyPublicIPAddress | Should Be $testIP } } Context "First option is down" { Mock Invoke-RestMethod { throw } -ParameterFilter { $uri -eq "https://api.ipify.org/"} It "Should return an IPv4 address" { Get-MyPublicIPAddress | Should Be $testIP } } Context "First and Second Options are down" { Mock Invoke-RestMethod { throw } -ParameterFilter { $uri -eq "https://api.ipify.org/" -or $uri -eq 'https://canhazip.com/'} It "Should return an IPv4 address" { Get-MyPublicIPAddress | Should Be $testIP } } Context "All three are down, should throw" { Mock Invoke-RestMethod { throw } It "Should throw" { { Get-MyPublicIPAddress } | Should Throw "Unable to determine" } } Context "A service returns an invalid IP" { Mock Invoke-RestMethod { return "IAMNOTANIPADDRESS" } It "Should Throw" { { Get-MyPublicIpAddress } | Should Throw "couldn't parse" } } } } } finally { if ($module) { Remove-Module -ModuleInfo $module } } |