Use t.Setenv instead of os.Setenv

Signed-off-by: Maksym Pavlenko <pavlenko.maksym@gmail.com>
This commit is contained in:
Maksym Pavlenko 2022-04-01 13:53:17 -07:00
parent 887615c7d0
commit 2d59a39445
3 changed files with 5 additions and 13 deletions

View File

@ -135,9 +135,7 @@ 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")
@ -145,8 +143,7 @@ func TestDetectPigz(t *testing.T) {
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)

View File

@ -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)

View File

@ -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() {