Use t.Setenv instead of os.Setenv

mxpv · mxpv · commit 2d59a3944598 · 2022-04-01T13:53:17.000-07:00
Signed-off-by: Maksym Pavlenko &lt;pavlenko.maksym@gmail.com&gt;
diff --git a/archive/compression/compression_test.go b/archive/compression/compression_test.go
@@ -135,18 +135,15 @@ func TestDetectPigz(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	oldPath := os.Getenv("PATH")
-	os.Setenv("PATH", tempPath)
-	defer os.Setenv("PATH", oldPath)
+	t.Setenv("PATH", tempPath)
 
 	if pigzPath := detectPigz(); pigzPath == "" {
 		t.Fatal("failed to detect pigz path")
 	} else if pigzPath != fullPath {
 		t.Fatalf("wrong pigz found: %s != %s", pigzPath, fullPath)
 	}
 
-	os.Setenv(disablePigzEnv, "1")
-	defer os.Unsetenv(disablePigzEnv)
+	t.Setenv(disablePigzEnv, "1")
 
 	if pigzPath := detectPigz(); pigzPath != "" {
 		t.Fatalf("disable via %s doesn't work", disablePigzEnv)
diff --git a/namespaces/context_test.go b/namespaces/context_test.go
@@ -18,7 +18,6 @@ package namespaces
 
 import (
 	"context"
-	"os"
 	"testing"
 )
 
@@ -47,9 +46,6 @@ func TestContext(t *testing.T) {
 }
 
 func TestNamespaceFromEnv(t *testing.T) {
-	oldenv := os.Getenv(NamespaceEnvVar)
-	defer os.Setenv(NamespaceEnvVar, oldenv) // restore old env var
-
 	ctx := context.Background()
 	namespace, ok := Namespace(ctx)
 	if ok {
@@ -61,7 +57,7 @@ func TestNamespaceFromEnv(t *testing.T) {
 	}
 
 	expected := "test-namespace"
-	os.Setenv(NamespaceEnvVar, expected)
+	t.Setenv(NamespaceEnvVar, expected)
 	nctx := NamespaceFromEnv(ctx)
 
 	namespace, ok = Namespace(nctx)
diff --git a/runtime/v2/shim/shim_test.go b/runtime/v2/shim/shim_test.go
@@ -18,7 +18,6 @@ package shim
 
 import (
 	"context"
-	"os"
 	"runtime"
 	"testing"
 )
@@ -27,7 +26,7 @@ func TestRuntimeWithEmptyMaxEnvProcs(t *testing.T) {
 	var oldGoMaxProcs = runtime.GOMAXPROCS(0)
 	defer runtime.GOMAXPROCS(oldGoMaxProcs)
 
-	os.Setenv("GOMAXPROCS", "")
+	t.Setenv("GOMAXPROCS", "")
 	setRuntime()
 
 	var currentGoMaxProcs = runtime.GOMAXPROCS(0)
@@ -37,7 +36,7 @@ func TestRuntimeWithEmptyMaxEnvProcs(t *testing.T) {
 }
 
 func TestRuntimeWithNonEmptyMaxEnvProcs(t *testing.T) {
-	os.Setenv("GOMAXPROCS", "not_empty")
+	t.Setenv("GOMAXPROCS", "not_empty")
 	setRuntime()
 	var oldGoMaxProcs2 = runtime.GOMAXPROCS(0)
 	if oldGoMaxProcs2 != runtime.NumCPU() {
