CLA-Assessment

1.7.2

Testing/Unit/PowerShell/Connection/Disconnect-CyberAssessmentTenant.Tests.ps1

BeforeDiscovery {
    $ModuleRootPath = Join-Path -Path $PSScriptRoot -ChildPath '..\..\..\..\Modules\Connection' -Resolve
    Import-Module (Join-Path -Path $ModuleRootPath -ChildPath 'Connection.psm1') -Function 'Disconnect-CyberAssessmentTenant' -Force
}

InModuleScope Connection {
    Describe -Tag 'Connection' -Name 'Disconnect-CyberAssessmentTenant' {
        BeforeAll {
            function Disconnect-MgGraph {throw 'this will be mocked'}
            Mock -ModuleName Connection Disconnect-MgGraph {}
            function Disconnect-ExchangeOnline {throw 'this will be mocked'}
            Mock -ModuleName Connection Disconnect-ExchangeOnline {}
            function Disconnect-SPOService {throw 'this will be mocked'}
            Mock -ModuleName Connection Disconnect-SPOService {}
            function Remove-PowerAppsAccount {throw 'this will be mocked'}
            Mock  -ModuleName Connection Remove-PowerAppsAccount {}
            function Disconnect-MicrosoftTeams {throw 'this will be mocked'}
            Mock  -ModuleName Connection Disconnect-MicrosoftTeams {}
            Mock -CommandName Write-Progress {}
        }
        It 'Disconnects from Microsoft Graph' {
            Disconnect-CyberAssessmentTenant -ProductNames 'aad'
            Should -Invoke -ModuleName Connection -CommandName Disconnect-MgGraph -Times 1 -Exactly
        }
        It 'Disconnects from Exchange Online' {
            Disconnect-CyberAssessmentTenant -ProductNames 'exo'
            Should -Invoke -ModuleName Connection -CommandName Disconnect-ExchangeOnline -Times 1 -Exactly
        }
        It 'Disconnects from Defender (Exchange Online and Security & Compliance)' {
            {Disconnect-CyberAssessmentTenant -ProductNames 'defender'} | Should -Not -Throw
        }
        It 'Disconnects from Power Platform' {
            {Disconnect-CyberAssessmentTenant -ProductNames 'powerplatform'} | Should -Not -Throw
        }
        It 'Disconnects from SharePoint Online' {
            {Disconnect-CyberAssessmentTenant -ProductNames 'sharepoint'} | Should -Not -Throw
        }
        It 'Disconnects from Microsoft Teams' {
            {Disconnect-CyberAssessmentTenant -ProductNames 'sharepoint'} | Should -Not -Throw
        }
        It 'Disconnects from all products' {
            {Disconnect-CyberAssessmentTenant} | Should -Not -Throw
        }
    }
}

AfterAll {
    Remove-Module Connection -ErrorAction SilentlyContinue
}