feat: use t.Setenv for setting vars in tests

kevincobain2000 · Dec 8, 2023 · 6f03dc7 · 6f03dc7
1 parent 94d3740
commit 6f03dc7
Showing 1 changed file with 2 additions and 13 deletions.
diff --git a/gobrew_test.go b/gobrew_test.go
@@ -36,26 +36,15 @@ func TestNewGobrewHomeDirDefaultsToHome(t *testing.T) {
 		envName = "HOME"
 	}
 
-	oldEnvValue := os.Getenv(envName)
-	defer func() {
-		_ = os.Setenv(envName, oldEnvValue)
-	}()
-
-	_ = os.Unsetenv(envName)
-
+	t.Setenv(envName, "")
 	gobrew := NewGoBrew()
 
 	assert.Equal(t, os.Getenv("HOME"), gobrew.homeDir)
 	t.Log("test finished")
 }
 
 func TestNewGobrewHomeDirUsesGoBrewRoot(t *testing.T) {
-	oldEnvValue := os.Getenv("GOBREW_ROOT")
-	defer func() {
-		_ = os.Setenv("GOBREW_ROOT", oldEnvValue)
-	}()
-
-	_ = os.Setenv("GOBREW_ROOT", "some_fancy_value")
+	t.Setenv("GOBREW_ROOT", "some_fancy_value")
 
 	gobrew := NewGoBrew()