aphrody-labs/PowerShell
1
0 Fork 0
mirror of https://github.com/aphrody-labs/PowerShell.git synced 2026-10-14 05:39:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix flaky Debug-Runspace attach event test (#27885)

Browse Source
This commit is contained in:
Travis Plunk authored and GitHub committed 2026-09-16 14:50:13 -07:00
1 parent 5e35e5ac89
commit 88937ea0a7
1 file changed
+60 -18
Download Patch File Download Diff File Expand all files Collapse all files
test/powershell/Modules/Microsoft.PowerShell.Utility/Debug-Runspace.Tests.ps1
+60 -18
View File
@@ -35,29 +35,71 @@ Describe "Debug-Runspace" -Tag "CI" {
It "Should write attach event and mark runspace as having a remote debugger attached" {
$onAttachName = [System.Management.Automation.PSEngineEvent]::OnDebugAttach
$debugTarget = [PowerShell]::Create()
$null = $debugTarget.AddCommand('Wait-Event').AddParameter('SourceIdentifier', $onAttachName)
$waitTask = $debugTarget.BeginInvoke()
$targetRunspace = $null
$debugTarget = $null
$debugger = $null
$debugTask = $null
$debugTarget.Runspace.IsRemoteDebuggerAttached | Should -BeFalse
try {
# Open the target runspace up front so that 'Debug-Runspace' can never observe it in a
# non-Opened state, and so that its event manager is guaranteed to exist when the
# OnDebugAttach event is generated.
$targetRunspace = [runspacefactory]::CreateRunspace()
$targetRunspace.Open()
$debugger = [PowerShell]::Create()
$null = $debugger.AddCommand('Debug-Runspace').AddParameter('Id', $debugTarget.Runspace.Id)
$debugTask = $debugger.BeginInvoke()
$debugTarget = [PowerShell]::Create()
$debugTarget.Runspace = $targetRunspace
$null = $debugTarget.AddCommand('Wait-Event').AddParameter('SourceIdentifier', $onAttachName)
$waitTask = $debugTarget.BeginInvoke()
$waitTask.AsyncWaitHandle.WaitOne(10000) | Should -BeTrue
$waitInfo = $debugTarget.EndInvoke($waitTask)
$waitInfo.SourceIdentifier | Should -Be $onAttachName
# 'BeginInvoke' only queues the work. Wait until the 'Wait-Event' pipeline is actually
# running in the target runspace before attaching the debugger, so the attach event is
# never generated against a runspace that has not started executing the waiter.
$ready = Wait-UntilTrue -IntervalInMilliseconds 20 -TimeoutInMilliseconds 5000 -sb {
$debugTarget.InvocationStateInfo.State -eq [System.Management.Automation.PSInvocationState]::Running -and
$targetRunspace.RunspaceAvailability -eq [System.Management.Automation.Runspaces.RunspaceAvailability]::Busy
}
$ready | Should -BeTrue -Because "the 'Wait-Event' pipeline should be running in the target runspace"
$debugTarget.Runspace.IsRemoteDebuggerAttached | Should -BeTrue
$targetRunspace.IsRemoteDebuggerAttached | Should -BeFalse
$debugger.Stop()
$exp = {
$debugger.EndInvoke($debugTask)
} | Should -Throw -PassThru
$exp.FullyQualifiedErrorId | Should -Be "PipelineStoppedException"
$debugger = [PowerShell]::Create()
$null = $debugger.AddCommand('Debug-Runspace').AddParameter('Id', $targetRunspace.Id)
$debugTask = $debugger.BeginInvoke()
$debugTarget.Runspace.IsRemoteDebuggerAttached | Should -BeFalse
$waitTask.AsyncWaitHandle.WaitOne(5000) | Should -BeTrue
$waitInfo = $debugTarget.EndInvoke($waitTask)
$waitInfo.SourceIdentifier | Should -Be $onAttachName
$targetRunspace.IsRemoteDebuggerAttached | Should -BeTrue
$debugger.Stop()
$exp = {
$debugger.EndInvoke($debugTask)
} | Should -Throw -PassThru
$exp.FullyQualifiedErrorId | Should -Be "PipelineStoppedException"
# 'IsRemoteDebuggerAttached' is reset by the cmdlet as it unwinds, which happens
# asynchronously with respect to 'Stop' completing.
$detached = Wait-UntilTrue -IntervalInMilliseconds 20 -TimeoutInMilliseconds 5000 -sb {
-not $targetRunspace.IsRemoteDebuggerAttached
}
$detached | Should -BeTrue
$targetRunspace.IsRemoteDebuggerAttached | Should -BeFalse
}
finally {
if ($debugger) {
try { $debugger.Stop() } catch { Write-Warning "Failed to stop the debugger during cleanup: $_" }
$debugger.Dispose()
}
if ($debugTarget) {
try { $debugTarget.Stop() } catch { Write-Warning "Failed to stop the debug target during cleanup: $_" }
$debugTarget.Dispose()
}
if ($targetRunspace) { $targetRunspace.Dispose() }
}
}
}