feat: Update Mock-EnvironmentVariable.Tests.ps1

Add a test to ensure that the environment variable got destroyed.

Author: BusHero

tests/PesterExtensions/Public/Mock-EnvironmentVariable.Tests.ps1

@@ -5,31 +5,51 @@ BeforeAll {
 }
 
 Describe 'Mock an environment variable' {
-	BeforeAll {
-		$script:environmentVariable = "test$(New-Guid)"
-		$script:InitialValue = 'Some value here and there'
+	Describe 'Code is called' {
+		BeforeAll {
+			$script:EnvironmentVariableName = "test$(New-Guid)"
+			$script:EnvironmentVariable = "env:${environmentVariableName}"
+			$script:InitialValue = 'Some value here and there'
+		}
+	
+		It 'Code is called' {
+			$script:InitialValue = 'Some value here and there'
+			$script:called = $false
+			Mock-EnvironmentVariable -Variable $environmentVariableName -Value $InitialValue {
+				$script:called = $true
+			}
+			$script:called | Should -BeTrue
+		}
 
+		AfterAll {
+			Remove-Item `
+				-Path $EnvironmentVariable `
+				-Recurse `
+				-Force `
+				-ErrorAction Ignore
+		}
 	}
-	It 'Code is called' {
-		$script:environmentVariable = "test$(New-Guid)"
-		$script:InitialValue = 'Some value here and there'
-		$script:called = $false
-		Mock-EnvironmentVariable -Variable $environmentVariable -Value $InitialValue {
-			$script:called = $true
-		}
-		$script:called | Should -BeTrue
-	}
+	Describe 'Environment variable is set up' {
+		BeforeAll {
+			$script:environmentVariable = "test$(New-Guid)"
+			$script:InitialValue = 'Some value here and there'
+		}
+	
+		It 'Environment variable is set up' {
+			$environmentVariable = "test$(New-Guid)"
+			$InitialValue = 'Some value here and there'
+			Test-Path -Path $environmentVariable | Should -BeFalse 
+			Mock-EnvironmentVariable -Variable $environmentVariable -Value $InitialValue {
+				(Get-ChildItem -Path "env:${environmentVariable}").Value | Should -Be $InitialValue
+			}
+			Test-Path -Path $environmentVariable | Should -BeFalse 
+		}
 
-	It 'Environment variable is set up' {
-		$environmentVariable = "test$(New-Guid)"
-		$InitialValue = 'Some value here and there'
-		Test-Path -Path $environmentVariable | Should -BeFalse 
-		Mock-EnvironmentVariable -Variable $environmentVariable -Value $InitialValue {
-			(Get-ChildItem -Path "env:${environmentVariable}").Value | Should -Be $InitialValue
-		}
-		Test-Path -Path $environmentVariable | Should -BeFalse 
-	}
+		AfterAll {
 
+		}
+	}
+		
 	Describe 'Environment variable is set' {
 		BeforeAll {
 			$script:environmentVariableName = "test$(New-Guid)"
@@ -167,4 +187,27 @@ Describe 'Mock an environment variable' {
 				-ErrorAction Ignore
 		}
 	}
+
+	Describe 'Set up environment variable is cleared up' {
+		BeforeAll {
+			$environmentVariableName = "test$(New-Guid)"
+			$script:environmentVariable = "env:${environmentVariableName}"
+		}
+
+		It 'Set up environment variable is cleared up' {
+			Mock-EnvironmentVariable `
+				-Variable $environmentVariableName { 
+				New-Item -Path $environmentVariable -Value 'Some value'
+			} 
+			Test-Path -Path $environmentVariable | Should -BeFalse 
+		}
+
+		AfterAll {
+			Remove-Item `
+				-Path $environmentVariable `
+				-Recurse `
+				-Force `
+				-ErrorAction Ignore
+		}
+	}
 }