mirror of https://github.com/databricks/cli.git
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package env
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/internal/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestContext(t *testing.T) {
|
|
testutil.CleanupEnvironment(t)
|
|
t.Setenv("FOO", "bar")
|
|
|
|
ctx0 := context.Background()
|
|
|
|
// Get
|
|
assert.Equal(t, "bar", Get(ctx0, "FOO"))
|
|
assert.Equal(t, "", Get(ctx0, "dontexist"))
|
|
|
|
// Lookup
|
|
v, ok := Lookup(ctx0, "FOO")
|
|
assert.True(t, ok)
|
|
assert.Equal(t, "bar", v)
|
|
v, ok = Lookup(ctx0, "dontexist")
|
|
assert.False(t, ok)
|
|
assert.Equal(t, "", v)
|
|
|
|
// Set and get new context.
|
|
// Verify that the previous context remains unchanged.
|
|
ctx1 := Set(ctx0, "FOO", "baz")
|
|
assert.Equal(t, "baz", Get(ctx1, "FOO"))
|
|
assert.Equal(t, "bar", Get(ctx0, "FOO"))
|
|
|
|
// Set and get new context.
|
|
// Verify that the previous contexts remains unchanged.
|
|
ctx2 := Set(ctx1, "FOO", "qux")
|
|
assert.Equal(t, "qux", Get(ctx2, "FOO"))
|
|
assert.Equal(t, "baz", Get(ctx1, "FOO"))
|
|
assert.Equal(t, "bar", Get(ctx0, "FOO"))
|
|
|
|
ctx3 := Set(ctx2, "BAR", "x=y")
|
|
|
|
all := All(ctx3)
|
|
assert.NotNil(t, all)
|
|
assert.Equal(t, "qux", all["FOO"])
|
|
assert.Equal(t, "x=y", all["BAR"])
|
|
assert.NotEmpty(t, all["PATH"])
|
|
}
|
|
|
|
func TestHome(t *testing.T) {
|
|
ctx := context.Background()
|
|
ctx = WithUserHomeDir(ctx, "...")
|
|
home := UserHomeDir(ctx)
|
|
assert.Equal(t, "...", home)
|
|
}
|