56 lines
1.6 KiB
PowerShell
56 lines
1.6 KiB
PowerShell
|
. $PSScriptRoot\..\..\Load-PesterModules.ps1
|
||
|
$here = Split-Path -Parent $MyInvocation.MyCommand.Path
|
||
|
$sut = (Split-Path -Leaf $MyInvocation.MyCommand.Path) -replace '\.tests\.', '.'
|
||
|
$functionPath = Join-Path -Path $here -ChildPath $sut
|
||
|
Write-Host "Overriding SUT: $functionPath"
|
||
|
Import-Module $functionPath -Force
|
||
|
$moduleForMock = ""
|
||
|
|
||
|
Describe "Get-ImdsV2Token" {
|
||
|
|
||
|
Context "Parameter Validation" {
|
||
|
|
||
|
It "Too Low TTL" {
|
||
|
|
||
|
{ Get-ImdsV2Token -TTL 0 } | Should Throw
|
||
|
}
|
||
|
|
||
|
It "Too High TTL" {
|
||
|
|
||
|
{ Get-ImdsV2Token -TTL 21601 } | Should Throw
|
||
|
}
|
||
|
}
|
||
|
|
||
|
Context "Cache Handling" {
|
||
|
|
||
|
$Global:AlkamiImdsSessionToken = $null
|
||
|
Mock -ModuleName $moduleForMock Invoke-RestMethod { return "1234567890" }
|
||
|
|
||
|
It "Sets Token In Cache" {
|
||
|
|
||
|
$token = Get-ImdsV2Token
|
||
|
|
||
|
Assert-MockCalled -ModuleName $moduleForMock Invoke-RestMethod -Times 1 -Exactly -Scope It
|
||
|
$token | Should -Be $Global:AlkamiImdsSessionToken
|
||
|
}
|
||
|
|
||
|
It "Retrieves Token From Cache" {
|
||
|
|
||
|
$token = Get-ImdsV2Token
|
||
|
|
||
|
Assert-MockCalled -ModuleName $moduleForMock Invoke-RestMethod -Times 0 -Exactly -Scope It
|
||
|
$token | Should -Be $Global:AlkamiImdsSessionToken
|
||
|
}
|
||
|
|
||
|
It "Breaks The Cache" {
|
||
|
|
||
|
$Global:AlkamiImdsSessionToken = "1234567890"
|
||
|
Mock -ModuleName $moduleForMock Invoke-RestMethod { return "0987654321" }
|
||
|
|
||
|
$token = Get-ImdsV2Token -InvalidateCache
|
||
|
|
||
|
Assert-MockCalled -ModuleName $moduleForMock Invoke-RestMethod -Times 1 -Exactly -Scope It
|
||
|
$token | Should -Be $Global:AlkamiImdsSessionToken
|
||
|
}
|
||
|
}
|
||
|
}
|