From 9687ba6315d2a4f37224101790a8d74e4963907a Mon Sep 17 00:00:00 2001 From: Wei Fu Date: Wed, 22 Apr 2020 23:15:47 +0800 Subject: [PATCH] test: TestRuntimeWithEmptyMaxEnvProcs should cleanup TestRuntimeWithEmptyMaxEnvProcs should restore the GoMaxProcs after test so that the temporary change of GoMaxProcs will not impact other case, like TestRuntimeWithNonEmptyMaxEnvProcs. Signed-off-by: Wei Fu --- runtime/v2/shim/shim_test.go | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/runtime/v2/shim/shim_test.go b/runtime/v2/shim/shim_test.go index 23a962a82..2ba6fbcd9 100644 --- a/runtime/v2/shim/shim_test.go +++ b/runtime/v2/shim/shim_test.go @@ -24,11 +24,14 @@ import ( ) func TestRuntimeWithEmptyMaxEnvProcs(t *testing.T) { + var oldGoMaxProcs = runtime.GOMAXPROCS(0) + defer runtime.GOMAXPROCS(oldGoMaxProcs) os.Setenv("GOMAXPROCS", "") setRuntime() - var oldGoMaxProcs = runtime.GOMAXPROCS(0) - if oldGoMaxProcs != 2 { + + var currentGoMaxProcs = runtime.GOMAXPROCS(0) + if currentGoMaxProcs != 2 { t.Fatal("the max number of procs should be 2") } }