Use t.Setenv instead of os.Setenv
Signed-off-by: Maksym Pavlenko <pavlenko.maksym@gmail.com>
This commit is contained in:
parent
887615c7d0
commit
2d59a39445
@ -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)
|
||||
|
@ -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)
|
||||
|
@ -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() {
|
||||
|
Loading…
Reference in New Issue
Block a user