summaryrefslogtreecommitdiff
path: root/go/internal/testhelper/testhelper.go
diff options
context:
space:
mode:
Diffstat (limited to 'go/internal/testhelper/testhelper.go')
-rw-r--r--go/internal/testhelper/testhelper.go17
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)
+ }
+ }
+}