Tests/Public/Set-PWSHYBKPIVRetryCount.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-PWSHYBKPIVRetryCount' -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 = @() } }
    }

    It 'Requires the -PinRetries and -PukRetries dynamic parameters' {
        { Set-PWSHYBKPIVRetryCount -Confirm:$false -ErrorAction Stop } | Should -Throw
    }

    It 'Calls Invoke-PWSHYBKPIVTool with the pin-retries action' {
        Set-PWSHYBKPIVRetryCount -PinRetries 3 -PukRetries 5 -Confirm:$false
        Should -Invoke -ModuleName Posh-YBKPIV Invoke-PWSHYBKPIVTool -Times 1 -ParameterFilter { $Action -eq 'pin-retries' }
    }

    It 'Does not change the retries when -WhatIf is used' {
        Set-PWSHYBKPIVRetryCount -PinRetries 3 -PukRetries 5 -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-PWSHYBKPIVRetryCount -PinRetries 3 -PukRetries 5 -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-PWSHYBKPIVRetryCount -PinRetries 3 -PukRetries 5 -Confirm:$false } | Should -Throw -ExpectedMessage '*Install-PWSHYBKPIVTool*'
    }
}