mirror of https://github.com/databricks/cli.git
47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package testutil
|
|
|
|
import (
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// CleanupEnvironment sets up a pristine environment containing only $PATH and $HOME.
|
|
// The original environment is restored upon test completion.
|
|
// Note: use of this function is incompatible with parallel execution.
|
|
func CleanupEnvironment(t *testing.T) {
|
|
// Restore environment when test finishes.
|
|
environ := os.Environ()
|
|
t.Cleanup(func() {
|
|
// Restore original environment.
|
|
for _, kv := range environ {
|
|
kvs := strings.SplitN(kv, "=", 2)
|
|
os.Setenv(kvs[0], kvs[1])
|
|
}
|
|
})
|
|
|
|
path := os.Getenv("PATH")
|
|
pwd := os.Getenv("PWD")
|
|
os.Clearenv()
|
|
|
|
// We use t.Setenv instead of os.Setenv because the former actively
|
|
// prevents a test being run with t.Parallel. Modifying the environment
|
|
// within a test is not compatible with running tests in parallel
|
|
// because of isolation; the environment is scoped to the process.
|
|
t.Setenv("PATH", path)
|
|
t.Setenv("HOME", pwd)
|
|
if runtime.GOOS == "windows" {
|
|
t.Setenv("USERPROFILE", pwd)
|
|
}
|
|
}
|
|
|
|
// GetEnvOrSkipTest proceeds with test only with that env variable
|
|
func GetEnvOrSkipTest(t *testing.T, name string) string {
|
|
value := os.Getenv(name)
|
|
if value == "" {
|
|
t.Skipf("Environment variable %s is missing", name)
|
|
}
|
|
return value
|
|
}
|