diff options
Diffstat (limited to 'go/internal/testhelper/testhelper.go')
| -rw-r--r-- | go/internal/testhelper/testhelper.go | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/go/internal/testhelper/testhelper.go b/go/internal/testhelper/testhelper.go index 5c900aa..a925c79 100644 --- a/go/internal/testhelper/testhelper.go +++ b/go/internal/testhelper/testhelper.go @@ -85,3 +85,9 @@ func getTestDataDir() (string, error) { return path.Join(path.Dir(currentFile), "testdata"), nil } + +func Setenv(key, value string) (func(), error) { + oldValue := os.Getenv(key) + err := os.Setenv(key, value) + return func() { os.Setenv(key, oldValue) }, err +} |