Replace os.Setenv with testing.T.Setenv in tests

T.Setenv ensures that the environment is returned to its prior state
when the test ends. It also panics when called from a parallel test to
prevent racy test interdependencies.
This commit is contained in:
Chris Bandy
2023-04-15 10:09:47 -05:00
parent 53cccbe4f9
commit d38ac7e7c6
2 changed files with 6 additions and 15 deletions

View File

@@ -20,7 +20,6 @@ import (
"context"
"errors"
"fmt"
"os"
"reflect"
"strings"
"testing"
@@ -705,8 +704,6 @@ func TestGCEPDLimits(t *testing.T) {
}
func TestGetMaxVols(t *testing.T) {
previousValue := os.Getenv(KubeMaxPDVols)
tests := []struct {
rawMaxVols string
expected int
@@ -731,18 +728,13 @@ func TestGetMaxVols(t *testing.T) {
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
os.Setenv(KubeMaxPDVols, test.rawMaxVols)
t.Setenv(KubeMaxPDVols, test.rawMaxVols)
result := getMaxVolLimitFromEnv()
if result != test.expected {
t.Errorf("expected %v got %v", test.expected, result)
}
})
}
os.Unsetenv(KubeMaxPDVols)
if previousValue != "" {
os.Setenv(KubeMaxPDVols, previousValue)
}
}
func getFakePVCLister(filterName string) fakeframework.PersistentVolumeClaimLister {