diff options
Diffstat (limited to 'go/internal/testhelper/testhelper.go')
-rw-r--r-- | go/internal/testhelper/testhelper.go | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/go/internal/testhelper/testhelper.go b/go/internal/testhelper/testhelper.go new file mode 100644 index 0000000..5cdab89 --- /dev/null +++ b/go/internal/testhelper/testhelper.go @@ -0,0 +1,17 @@ +package testhelper + +import "os" + +func TempEnv(env map[string]string) func() { + var original = make(map[string]string) + for key, value := range env { + original[key] = os.Getenv(key) + os.Setenv(key, value) + } + + return func() { + for key, originalValue := range original { + os.Setenv(key, originalValue) + } + } +} |