diff --git a/test/powershell/Get-PSCallStack.Tests.ps1 b/test/powershell/Get-PSCallStack.Tests.ps1 new file mode 100644 index 000000000..52124e8a0 --- /dev/null +++ b/test/powershell/Get-PSCallStack.Tests.ps1 @@ -0,0 +1,87 @@ +Describe "Get-PSCallStack DRT Unit Tests" -Tags DRT { + BeforeAll { + $scriptFileName = "GetTryCatchCallStack.ps1" + $scriptFilePath = Join-Path $TestDrive -ChildPath $scriptFileName + } + It "Verifies that the script block of a catch clause does not show up on the call stack" { + $fileStream = @" + function foo() + { + try + { + throw 1 + } + catch + { + bar + } + } + + function bar() + { + try + { + throw 1 + } + catch + { + try + { + throw 1 + } + catch + { + Get-PSCallStack + } + } + } + + foo +"@ + + $fileStream > $scriptFilePath + + $results = & "$scriptFilePath" + $results.Count | Should BeGreaterThan 3 + $results[0].Command | Should be "bar" + $results[0].ScriptName | Should be $scriptFilePath + $results[0].ScriptLineNumber | Should be 27 + $results[0].InvocationInfo.ScriptLineNumber | Should be 9 + + $results[1].Command | Should be "foo" + $results[1].ScriptName | Should be $scriptFilePath + $results[1].ScriptLineNumber | Should be 9 + $results[1].InvocationInfo.ScriptLineNumber | Should be 32 + + #InvocationInfo.ScriptLineNumber: Gets the line number of the script that contains the command + $results[2].Command | Should be $scriptFileName + $results[2].ScriptName | Should be $scriptFilePath + $results[2].ScriptLineNumber | Should be 32 + $results[2].InvocationInfo.ScriptLineNumber | Should be 44 + } + + It "Verify that the script block of a trap statement shows up on the call stack" { + $fileStream = @" + trap + { + Get-PSCallStack + continue + } + + throw 1 +"@ + + $fileStream > $scriptFilePath + $results = & "$scriptFilePath" + $results.Count | Should BeGreaterThan 2 + $results[0].Command | Should be $scriptFileName + $results[0].ScriptName | Should be $scriptFilePath + $results[0].ScriptLineNumber | Should be 3 + $results[0].InvocationInfo.ScriptLineNumber | Should be 75 + + $results[1].Command | Should be $scriptFileName + $results[1].ScriptName | Should be $scriptFilePath + $results[1].ScriptLineNumber | Should be 7 + $results[1].InvocationInfo.ScriptLineNumber | Should be 75 + } +} \ No newline at end of file