Posh-YBKPIV

0.10.0

Tests/Public/Set-PWSHYBKPIVManagementKey.Tests.ps1

BeforeAll {
    Import-Module (Resolve-Path "$PSScriptRoot\..\..\Posh-YBKPIV.psd1") -Force

    $script:RealConfigPath = (Resolve-Path "$PSScriptRoot\..\..\Config\Posh-YBKPIV.json").Path
    InModuleScope Posh-YBKPIV -Parameters @{ ConfigPath = $script:RealConfigPath } {
        param($ConfigPath)
        $script:ConfigPath = $ConfigPath
    }
}

AfterAll {
    Remove-Module Posh-YBKPIV -ErrorAction SilentlyContinue
}

Describe 'Set-PWSHYBKPIVManagementKey' -Tag Unit {
    BeforeEach {
        $script:InstallExePath = 'C:\Program Files\Yubico\Yubico PIV Tool\bin\yubico-piv-tool.exe'
        Mock -ModuleName Posh-YBKPIV Test-Path { $true } -ParameterFilter { $Path -eq $script:InstallExePath }
        Mock -ModuleName Posh-YBKPIV Resolve-PWSHYBKPIVArchitecture { 'win64' }
        Mock -ModuleName Posh-YBKPIV Get-PWSHYBKPIVInstallPath { $script:InstallExePath }
        Mock -ModuleName Posh-YBKPIV Invoke-PWSHYBKPIVTool { [PSCustomObject]@{ ExitCode = 0; Output = @('Generated new key: 0102030405060708') } }
    }

    It 'Returns the raw output, including a generated key when none was supplied' {
        Set-PWSHYBKPIVManagementKey -Confirm:$false | Should -Be @('Generated new key: 0102030405060708')
    }

    It 'Calls Invoke-PWSHYBKPIVTool with the set-mgm-key action' {
        Set-PWSHYBKPIVManagementKey -Confirm:$false | Out-Null
        Should -Invoke -ModuleName Posh-YBKPIV Invoke-PWSHYBKPIVTool -Times 1 -ParameterFilter { $Action -eq 'set-mgm-key' }
    }

    It 'Does not set the management key when -WhatIf is used' {
        Set-PWSHYBKPIVManagementKey -WhatIf
        Should -Invoke -ModuleName Posh-YBKPIV Invoke-PWSHYBKPIVTool -Times 0
    }

    It 'Throws when Invoke-PWSHYBKPIVTool fails' {
        Mock -ModuleName Posh-YBKPIV Invoke-PWSHYBKPIVTool { throw 'boom' }
        { Set-PWSHYBKPIVManagementKey -Confirm:$false } | Should -Throw -ExpectedMessage '*boom*'
    }

    It 'Throws a clear error when yubico-piv-tool.exe is not installed' {
        Mock -ModuleName Posh-YBKPIV Test-Path { $false } -ParameterFilter { $Path -eq $script:InstallExePath }
        { Set-PWSHYBKPIVManagementKey -Confirm:$false } | Should -Throw -ExpectedMessage '*Install-PWSHYBKPIVTool*'
    }
}