mirror of https://github.com/databricks/cli.git
24 lines
340 B
Go
24 lines
340 B
Go
package tests
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func SetTestEnv(t *testing.T, key, value string) {
|
|
originalValue, isSet := os.LookupEnv(key)
|
|
|
|
err := os.Setenv(key, value)
|
|
assert.NoError(t, err)
|
|
|
|
t.Cleanup(func() {
|
|
if isSet {
|
|
os.Setenv(key, originalValue)
|
|
} else {
|
|
os.Unsetenv(key)
|
|
}
|
|
})
|
|
}
|