go/internal/testhelper/testhelper.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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)
		}
	}
}