golang · codyoss · Aug 18, 2020 · Aug 17, 2020 · Aug 18, 2020 · Aug 18, 2020
diff --git a/mockgen/parse_test.go b/mockgen/parse_test.go
@@ -167,7 +167,10 @@ func TestParsePackageImport(t *testing.T) {
 	} {
 		t.Run(testCase.name, func(t *testing.T) {
 			for key, value := range testCase.envs {
-				os.Setenv(key, value)
+				err = os.Setenv(key, value)
+				if err != nil {
+					t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
+				}
 			}
 			pkgPath, err := parsePackageImport(filepath.Clean(testCase.dir))
 			if err != testCase.err {
@@ -195,8 +198,18 @@ func TestParsePackageImport_FallbackGoPath(t *testing.T) {
 	if err != nil {
 		t.Error(err)
 	}
-	os.Setenv("GOPATH", goPath)
-	os.Setenv("GO111MODULE", "on")
+	key := "GOPATH"
+	value := goPath
+	err = os.Setenv(key, value)
+	if err != nil {
+		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
+	}
+	key = "GO111MODULE"
+	value = "on"
+	err = os.Setenv(key, value)
+	if err != nil {
+		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
+	}
 	pkgPath, err := parsePackageImport(srcDir)
 	expected := "example.com/foo"
 	if pkgPath != expected {
@@ -237,8 +250,18 @@ func TestParsePackageImport_FallbackMultiGoPath(t *testing.T) {
 	}()
 
 	goPaths := strings.Join(goPathList, string(os.PathListSeparator))
-	os.Setenv("GOPATH", goPaths)
-	os.Setenv("GO111MODULE", "on")
+	key := "GOPATH"
+	value := goPaths
+	err = os.Setenv(key, value)
+	if err != nil {
+		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
+	}
+	key = "GO111MODULE"
+	value = "on"
+	err = os.Setenv(key, value)
+	if err != nil {
+		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
+	}
 	pkgPath, err := parsePackageImport(srcDir)
 	expected := "example.com/foo"
 	if pkgPath != expected {