From: Shengjing Zhu <zhsj@debian.org>
Date: Thu, 24 Feb 2022 01:39:57 +0800
Subject: Fix test with GOPATH mode

GOPATH and GO111MODULE envs need to be reset after test.

Forwarded: no
---
 mockgen/mockgen_test.go | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/mockgen/mockgen_test.go b/mockgen/mockgen_test.go
index 5556600..de52358 100644
--- a/mockgen/mockgen_test.go
+++ b/mockgen/mockgen_test.go
@@ -385,14 +385,10 @@ func TestParsePackageImport_FallbackGoPath(t *testing.T) {
 	}
 	key := "GOPATH"
 	value := goPath
-	if err := os.Setenv(key, value); err != nil {
-		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
-	}
+	t.Setenv(key, value)
 	key = "GO111MODULE"
 	value = "on"
-	if err := os.Setenv(key, value); err != nil {
-		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
-	}
+	t.Setenv(key, value)
 	pkgPath, err := parsePackageImport(srcDir)
 	expected := "example.com/foo"
 	if pkgPath != expected {
@@ -435,14 +431,10 @@ func TestParsePackageImport_FallbackMultiGoPath(t *testing.T) {
 	goPaths := strings.Join(goPathList, string(os.PathListSeparator))
 	key := "GOPATH"
 	value := goPaths
-	if err := os.Setenv(key, value); err != nil {
-		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
-	}
+	t.Setenv(key, value)
 	key = "GO111MODULE"
 	value = "on"
-	if err := os.Setenv(key, value); err != nil {
-		t.Fatalf("unable to set environment variable %q to %q: %v", key, value, err)
-	}
+	t.Setenv(key, value)
 	pkgPath, err := parsePackageImport(srcDir)
 	expected := "example.com/foo"
 	if pkgPath != expected {