Try to preserve Pester function

PowerShell · Jul 15, 2021 · 1f8f793 · 1f8f793
1 parent e7f1c57
commit 1f8f793
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/Tests/PSScriptAnalyzerTestHelper.psm1 b/Tests/PSScriptAnalyzerTestHelper.psm1
@@ -1,3 +1,5 @@
+$script:SafeShould = (Get-Command -Module Pester -Name Should -ErrorAction Stop)
+
 Function Get-ExtentTextFromContent
 {
 	    Param(
@@ -45,14 +47,12 @@ Function Test-CorrectionExtentFromContent {
         [string] $correctionText
     )
 
-	Import-Module Pester -ErrorAction Stop
-
-
 	$corrections = $diagnosticRecord.SuggestedCorrections
 	$corrections.Count | Should -Be $correctionsCount
 	$corrections[0].Text | Should -Be $correctionText
-	Get-ExtentTextFromContent $corrections[0] $rawContent | `
-		       Should -Be $violationText
+	$extent = Get-ExtentTextFromContent $corrections[0] $rawContent
+
+	& $script:SafeShould -ActualValue $extent -Be $violationText
 }
 
 Function Get-Count