Functions/Connect-AzureRmsAdminAccount.Tests.ps1
|
describe "BitTitan.Runbooks.AzureRMS/Connect-AzureRmsAdminAccount" -Tags "module", "unit" { # Import the function to test . "$($PSScriptRoot)\Connect-AzureRmsAdminAccount.ps1" # Declare our own Get-CredentialFromMSPCompleteEndpoint # If we don't do this the mock will not work function Get-CredentialFromMSPCompleteEndpoint { param ($endpoint) return [PSCredential]::new("username", ("password" | ConvertTo-SecureString -AsPlainText -Force)) } # Declare our own Connect-AadrmService # If we don't do this the mock will not work function Connect-AadrmService { param ([PSCredential]$Credential) } context "when there are no issues" { # Mock Get-CredentialFromMSPCompleteEndpoint mock Get-CredentialFromMSPCompleteEndpoint { return [PSCredential]::new("username", ("password" | ConvertTo-SecureString -AsPlainText -Force)) } # Mock Connect-AadrmService mock Connect-AadrmService {} it "connects to Azure RMS with the provided username and password" { # Call the function $output = Connect-AzureRmsAdminAccount -Username "username" -Password ("password" | ConvertTo-SecureString -AsPlainText -Force) # Verify the mocks Assert-MockCalled Get-CredentialFromMSPCompleteEndpoint -Times 0 -Exactly -Scope it Assert-MockCalled Connect-AadrmService -Times 1 -Exactly -ParameterFilter { $Credential.Username -eq "username" -and $Credential.GetNetworkCredential().Password -eq "password" } -Scope it # Verify the output $output | Should Be $true } it "connects to Azure RMS with the provided endpoint" { # Mock the endpoint $endpoint = "endpoint" # Call the function $output = Connect-AzureRmsAdminAccount -Endpoint $endpoint # Verify the mocks Assert-MockCalled Get-CredentialFromMSPCompleteEndpoint -Times 1 -Exactly -ParameterFilter { $Endpoint -eq "endpoint" } -Scope it Assert-MockCalled Connect-AadrmService -Times 1 -Exactly -ParameterFilter { $Credential.Username -eq "username" -and $Credential.GetNetworkCredential().Password -eq "password" } -Scope it # Verify the output $output | Should Be $true } } context "when there is an exception while connecting to Azure RMS" { # Mock Get-CredentialFromMSPCompleteEndpoint mock Get-CredentialFromMSPCompleteEndpoint { return [PSCredential]::new("username", ("password" | ConvertTo-SecureString -AsPlainText -Force)) } # Mock Connect-AadrmService mock Connect-AadrmService { throw "throws exception" } it "fails to connect to Azure RMS and outputs an error message" { # Call the function $output = Connect-AzureRmsAdminAccount -Username "username" -Password ("password" | ConvertTo-SecureString -AsPlainText -Force) ` -ErrorAction SilentlyContinue -ErrorVariable errorVariable # Verify the mocks Assert-MockCalled Get-CredentialFromMSPCompleteEndpoint -Times 0 -Exactly -Scope it Assert-MockCalled Connect-AadrmService -Times 1 -Exactly -ParameterFilter { $Credential.Username -eq "username" -and $Credential.GetNetworkCredential().Password -eq "password" } -Scope it # Verify the output $errorVariable | Should Not BeNullOrEmpty $output | Should Be $false } } } |