From 325c39f90a46cab062551e59f1b95b579b674fc6 Mon Sep 17 00:00:00 2001
From: TingLiu6 <v-litin@microsoft.com>
Date: Wed, 1 Jun 2016 02:29:44 -0700
Subject: [PATCH] Add Unit Test for Get-PSCallSta

---
 test/powershell/Get-PSCallStack.Tests.ps1 | 87 +++++++++++++++++++++++
 1 file changed, 87 insertions(+)
 create mode 100644 test/powershell/Get-PSCallStack.Tests.ps1

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