Merge pull request #1063 from PowerShell/GetPsCallStackTests

Add Unit Test for Get-PSCallSta

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
+    }
+}