giving a bit of wait time to Assert-RunspaceCount

santisq · Jun 26, 2024 · 7e58b70 · 7e58b70
1 parent 4d08cad
commit 7e58b70
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/tests/PSParallelPipeline.tests.ps1 b/tests/PSParallelPipeline.tests.ps1
@@ -137,6 +137,12 @@ Describe PSParallelPipeline {
 
     Context 'TimeoutSeconds Parameter' {
         It 'Stops processing after the specified seconds' {
+            $wait = 5
+            if (-not $IsCoreCLR) {
+                # because pwsh 5.1 fucking sucks!
+                $wait = 10
+            }
+
             Assert-RunspaceCount {
                 $timer = [Stopwatch]::StartNew()
                 {
@@ -150,7 +156,7 @@ Describe PSParallelPipeline {
                 } | Should -Throw -ExceptionType ([TimeoutException])
                 $timer.Stop()
                 $timer.Elapsed | Should -BeLessOrEqual ([timespan]::FromSeconds(2.2))
-            }
+            } -WaitSeconds $wait
         }
     }
 

diff --git a/tests/common.psm1 b/tests/common.psm1
@@ -20,15 +20,18 @@ function Assert-RunspaceCount {
     [CmdletBinding()]
     param(
         [Parameter(Mandatory)]
-        [scriptblock] $ScriptBlock
+        [scriptblock] $ScriptBlock,
+
+        [Parameter()]
+        [int] $WaitSeconds = 5
     )
 
     try {
         $count = @(Get-Runspace).Count
         & $ScriptBlock
     }
     finally {
-        Start-Sleep 5
+        Start-Sleep $WaitSeconds
         Get-Runspace |
             Should -HaveCount $count -Because 'Runspaces should be correctly disposed'
     }