Functions/Assertions/HaveCount.ps1
|
function PesterHaveCount($ActualValue, [int] $ExpectedValue, [switch] $Negate, [string] $Because) { if ($ExpectedValue -lt 0) { throw [ArgumentException]"Excpected collection size must be greater than or equal to 0." } $count = if ($null -eq $ActualValue) { 0 } else { $ActualValue.Count } $expectingEmpty = $ExpectedValue -eq 0 [bool] $succeeded = $count -eq $ExpectedValue if ($Negate) { $succeeded = -not $succeeded } if (-not $succeeded) { if ($Negate) { $expect = if ($expectingEmpty) { "Expected a non-empty collection" } else { "Expected a collection with size different from $(Format-Nicely $ExpectedValue)" } $but = if ($count -ne 0) { "but got collection with that size $(Format-Nicely $ActualValue)." } else { "but got an empty collection."} return New-Object psobject -Property @{ Succeeded = $false FailureMessage = "$expect,$(Format-Because $Because) $but" } } else { $expect = if ($expectingEmpty) { "Expected an empty collection" } else { "Expected a collection with size $(Format-Nicely $ExpectedValue)" } $but = if ($count -ne 0) { "but got collection with size $(Format-Nicely $count) $(Format-Nicely $ActualValue)." } else { "but got an empty collection."} return New-Object psobject -Property @{ Succeeded = $false FailureMessage = "$expect,$(Format-Because $Because) $but" } } } return New-Object psobject -Property @{ Succeeded = $true } } Add-AssertionOperator -Name HaveCount ` -Test $function:PesterHaveCount ` -SupportsArrayInput function PesterHaveCountFailureMessage() {} function NotPesterHaveCountFailureMessage() {} |