summaryrefslogtreecommitdiffhomepage
path: root/pkg
diff options
context:
space:
mode:
authorDean Deng <deandeng@google.com>2021-01-13 15:10:03 -0800
committergVisor bot <gvisor-bot@google.com>2021-01-13 15:15:20 -0800
commit1efe0ebc5973ec8a06b881c087dae2183898504b (patch)
tree6587b91164883c87324abe602b438f3e3ace19cf /pkg
parentf34aaf7ef17aa10c7ba1923d0694347e47634192 (diff)
Switch uses of os.Getenv that check for empty string to os.LookupEnv.
Whether the variable was found is already returned by syscall.Getenv. os.Getenv drops this value while os.Lookupenv passes it along. PiperOrigin-RevId: 351674032
Diffstat (limited to 'pkg')
-rw-r--r--pkg/tcpip/link/sharedmem/sharedmem_test.go4
-rw-r--r--pkg/test/testutil/testutil.go14
2 files changed, 9 insertions, 9 deletions
diff --git a/pkg/tcpip/link/sharedmem/sharedmem_test.go b/pkg/tcpip/link/sharedmem/sharedmem_test.go
index dd2e1a125..9f5ee43d7 100644
--- a/pkg/tcpip/link/sharedmem/sharedmem_test.go
+++ b/pkg/tcpip/link/sharedmem/sharedmem_test.go
@@ -191,8 +191,8 @@ func shuffle(b []int) {
}
func createFile(t *testing.T, size int64, initQueue bool) int {
- tmpDir := os.Getenv("TEST_TMPDIR")
- if tmpDir == "" {
+ tmpDir, ok := os.LookupEnv("TEST_TMPDIR")
+ if !ok {
tmpDir = os.Getenv("TMPDIR")
}
f, err := ioutil.TempFile(tmpDir, "sharedmem_test")
diff --git a/pkg/test/testutil/testutil.go b/pkg/test/testutil/testutil.go
index 2b2f95c13..a35c7ffa6 100644
--- a/pkg/test/testutil/testutil.go
+++ b/pkg/test/testutil/testutil.go
@@ -83,11 +83,10 @@ func ConfigureExePath() error {
// TmpDir returns the absolute path to a writable directory that can be used as
// scratch by the test.
func TmpDir() string {
- dir := os.Getenv("TEST_TMPDIR")
- if dir == "" {
- dir = "/tmp"
+ if dir, ok := os.LookupEnv("TEST_TMPDIR"); ok {
+ return dir
}
- return dir
+ return "/tmp"
}
// Logger is a simple logging wrapper.
@@ -543,7 +542,7 @@ func IsStatic(filename string) (bool, error) {
//
// See https://docs.bazel.build/versions/master/test-encyclopedia.html#role-of-the-test-runner.
func TouchShardStatusFile() error {
- if statusFile := os.Getenv("TEST_SHARD_STATUS_FILE"); statusFile != "" {
+ if statusFile, ok := os.LookupEnv("TEST_SHARD_STATUS_FILE"); ok {
cmd := exec.Command("touch", statusFile)
if b, err := cmd.CombinedOutput(); err != nil {
return fmt.Errorf("touch %q failed:\n output: %s\n error: %s", statusFile, string(b), err.Error())
@@ -565,8 +564,9 @@ func TestIndicesForShard(numTests int) ([]int, error) {
shardTotal = 1
)
- indexStr, totalStr := os.Getenv("TEST_SHARD_INDEX"), os.Getenv("TEST_TOTAL_SHARDS")
- if indexStr != "" && totalStr != "" {
+ indexStr, indexOk := os.LookupEnv("TEST_SHARD_INDEX")
+ totalStr, totalOk := os.LookupEnv("TEST_TOTAL_SHARDS")
+ if indexOk && totalOk {
// Parse index and total to ints.
var err error
shardIndex, err = strconv.Atoi(indexStr)