refr: Refactor Mock-EnvironmentVariable.Tests.ps1

Remove duplications from tetst
BusHero · Jul 26, 2022 · 5b65653 · 5b65653
1 parent 4beeda3
commit 5b65653
Showing 1 changed file with 114 additions and 142 deletions.
diff --git a/tests/PesterExtensions/Public/Mock-EnvironmentVariable.Tests.ps1 b/tests/PesterExtensions/Public/Mock-EnvironmentVariable.Tests.ps1
@@ -5,170 +5,142 @@ BeforeAll {
 }
 
 Describe 'Mock an environment variable' {
-	Describe 'Code is called' {
-		BeforeAll {
-			$script:EnvironmentVariableName = "test$(New-Guid)"
-			$script:InitialValue = 'Some value here and there'
-		}
+	BeforeAll {
+		$EnvironmentVariableName = "test$(New-Guid)"
+		$InitialValue = 'Some value here and there'
+	}
 
-		It 'Code is called' {
-			$script:called = $false
-			Mock-EnvironmentVariable -Variable $environmentVariableName -Value $InitialValue {
-				$script:called = $true
-			}
-			$script:called | Should -BeTrue
-		}
-
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
+	It 'Code is called' {
+		$script:called = $false
+		Mock-EnvironmentVariable -Variable $environmentVariableName -Value $InitialValue {
+			$script:called = $true
 		}
+		$called | Should -BeTrue
 	}
 
-	Describe 'Environment variable is set up' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-			$script:InitialValue = 'Some value here and there'
-		}
-
-		It 'Environment variable is set up' {
-			[Environment]::GetEnvironmentVariable($EnvironmentVariableName) | Should -Be $null -Because 'environment variable should be reset'
-			Mock-EnvironmentVariable -Variable $environmentVariableName -Value $InitialValue {
-				[Environment]::GetEnvironmentVariable($EnvironmentVariableName) | Should -Be $InitialValue
-			}
-			[Environment]::GetEnvironmentVariable($EnvironmentVariableName) | Should -Be $null -Because 'environment variable should be reset'
-		}
+	AfterAll {
+		[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
+	}
+}
 
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
-		}
+Describe 'Check env mocking' -ForEach @(
+	@{ 
+		InitialValue         = 'Initial Value';
+		MockedValue          = 'Mocked Value'; 
+		ValueInsideTheScript = 'Mocked Value';
+		Script               = { }
 	}
-
-	Describe 'Environment variable is set' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-			$script:InitialValue = 'Some value here and there'
-			$script:UpdatedValue = 'Some updated value'
-			[Environment]::SetEnvironmentVariable($environmentVariableName, $InitialValue)
-		}
-		It 'Environment variable is set up' {
-			Mock-EnvironmentVariable -Variable $environmentVariableName -Value $UpdatedValue {
-				[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $UpdatedValue
-			}
-			[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $InitialValue
-		}
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
-		}
+	@{ 
+		InitialValue         = $null; 
+		MockedValue          = 'Mocked Value'; 
+		ValueInsideTheScript = 'Mocked Value';
+		Script               = { }
 	}
-
-	Describe 'Restore variable that got destroyed' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-			$script:InitialValue = 'Some value here and there'
-			$script:UpdatedValue = 'Some updated value'
-
-			[Environment]::SetEnvironmentVariable($environmentVariableName, $InitialValue)
-			Mock-EnvironmentVariable -Variable $environmentVariableName -Value $UpdatedValue {
-				[Environment]::SetEnvironmentVariable($environmentVariableName, $null)
-			}
-		}
-		It 'Environment variable is set up' {
-			[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $InitialValue
-		}
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
-		}
+	@{ 
+		InitialValue         = 'Initial Value'; 
+		MockedValue          = $null; 
+		ValueInsideTheScript = 'Initial Value';
+		Script               = { }
 	}
-
-	Describe 'Use the original value if value is not set up' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-			$script:environmentVariable = "env:${environmentVariableName}"
-			$script:InitialValue = 'Some value here and there'
-
-			[Environment]::SetEnvironmentVariable($environmentVariableName, $InitialValue)
-		}
-		It 'Test' {
-			Mock-EnvironmentVariable -Variable $environmentVariableName {
-				[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $InitialValue
-			}
-		}
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
-		}
+	@{ 
+		Script = { } 
 	}
-
-	Describe 'Do not create the variable if no value is specified' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-		}
-		It 'Test' {
-			Mock-EnvironmentVariable -Variable $environmentVariableName {
-				[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $null
-			}
-		}
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
+	@{
+		InitialValue         = 'Initial Value';
+		MockedValue          = 'Mocked Value'; 
+		ValueInsideTheScript = 'Mocked Value';
+		Script               = { 
+			[Environment]::SetEnvironmentVariable($args[0], 'Some updated value') 
 		}
 	}
-
-	Describe 'Initial value should be reasigned' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-			$script:InitialValue = 'Some value here and there'
-			$script:UpdatedValue = 'Some updated value'
-			$script:Message = 'Some message here and there'
-			[Environment]::SetEnvironmentVariable($environmentVariableName, $InitialValue)
+	@{
+		InitialValue         = 'Initial Value';
+		MockedValue          = $null; 
+		ValueInsideTheScript = 'Initial Value';
+		Script               = { 
+			[Environment]::SetEnvironmentVariable($args[0], 'Some updated value') 
 		}
-		It 'Throw' {
-			{ Mock-EnvironmentVariable `
-					-Variable $environmentVariableName `
-					-Value $UpdatedValue { throw $Message } 
-			} | Should -Throw -ExpectedMessage $Message
-			[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $InitialValue
+	}
+	@{
+		InitialValue         = $null;
+		MockedValue          = 'Mocked Value'; 
+		ValueInsideTheScript = 'Mocked Value';
+		Script               = { 
+			[Environment]::SetEnvironmentVariable($args[0], 'Some updated value') 
 		}
-
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
+	}
+	@{
+		InitialValue         = $null;
+		MockedValue          = $null; 
+		ValueInsideTheScript = $null;
+		Script               = { 
+			[Environment]::SetEnvironmentVariable($args[0], 'Some updated value') 
 		}
 	}
+) {
+	BeforeAll {
+		$environmentVariableName = "test_$(New-Guid)"
+		[Environment]::SetEnvironmentVariable($environmentVariableName, $InitialValue)
+	}
+	It 'Test' {
+		Mock-EnvironmentVariable `
+			-Variable $environmentVariableName `
+			-Value $MockedValue {
+			[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $ValueInsideTheScript
+			Invoke-Command -ScriptBlock $script -ArgumentList $environmentVariableName
+		} 
+		[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $InitialValue
+	}
+	AfterAll {
+		[Environment]::SetEnvironmentVariable($environmentVariableName, $null)
+	}
+}
 
-	Describe 'Created variable should be destroyed' {
-		BeforeAll {
-			$script:environmentVariableName = "test$(New-Guid)"
-			$script:environmentVariable = "env:${environmentVariableName}"
-			$script:InitialValue = 'Some value here and there'
-			$script:UpdatedValue = 'Some updated value'
-			$script:Message = 'Some message here and there'
+Describe 'Should throw' -ForEach @(
+	@{
+		InitialValue = 'Initial Value';
+		MockedValue  = 'Mocked Value'; 
+		Script       = { 
+			throw 'Some exception here and there'
 		}
-		It 'Throw' {
-			{ Mock-EnvironmentVariable `
-					-Variable $environmentVariableName `
-					-Value $UpdatedValue { throw $Message } 
-			} | Should -Throw -ExpectedMessage $Message
-			[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $null
+	}
+	@{
+		InitialValue = 'Initial Value';
+		MockedValue  = $null; 
+		Script       = { 
+			throw 'Some exception here and there'
 		}
-
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
+	}
+	@{
+		InitialValue = $null;
+		MockedValue  = 'Mocked Value'; 
+		Script       = { 
+			throw 'Some exception here and there'
 		}
 	}
-
-	Describe 'Set up environment variable is cleared up' {
-		BeforeAll {
-			$environmentVariableName = "test$(New-Guid)"
+	@{
+		InitialValue = $null;
+		MockedValue  = $null; 
+		Script       = { 
+			throw 'Some exception here and there'
 		}
-
-		It 'Set up environment variable is cleared up' {
+	}
+) {
+	BeforeAll {
+		$environmentVariableName = "test_$(New-Guid)"
+		[Environment]::SetEnvironmentVariable($environmentVariableName, $InitialValue)
+	}
+	It 'Test' {
+		{ 
 			Mock-EnvironmentVariable `
-				-Variable $environmentVariableName { 
-				New-Item -Path $environmentVariable -Value 'Some value'
+				-Variable $environmentVariableName `
+				-Value $MockedValue {
+				Invoke-Command -ScriptBlock $script -ArgumentList $environmentVariableName
 			} 
-			[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $null
-		}
-
-		AfterAll {
-			[Environment]::SetEnvironmentVariable($EnvironmentVariableName, $null)
-		}
+		} | Should -Throw  
+		[Environment]::GetEnvironmentVariable($environmentVariableName) | Should -Be $InitialValue
+	}
+	AfterAll {
+		[Environment]::SetEnvironmentVariable($environmentVariableName, $null)
 	}
 }