mirror of https://github.com/databricks/cli.git
132 lines
3.9 KiB
Go
132 lines
3.9 KiB
Go
package mutator
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/variable"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSetVariableFromProcessEnvVar(t *testing.T) {
|
|
defaultVal := "default"
|
|
variable := variable.Variable{
|
|
Description: "a test variable",
|
|
Default: defaultVal,
|
|
}
|
|
|
|
// set value for variable as an environment variable
|
|
t.Setenv("BUNDLE_VAR_foo", "process-env")
|
|
|
|
diags := setVariable(context.Background(), &variable, "foo")
|
|
require.NoError(t, diags.Error())
|
|
assert.Equal(t, variable.Value, "process-env")
|
|
}
|
|
|
|
func TestSetVariableUsingDefaultValue(t *testing.T) {
|
|
defaultVal := "default"
|
|
variable := variable.Variable{
|
|
Description: "a test variable",
|
|
Default: defaultVal,
|
|
}
|
|
|
|
diags := setVariable(context.Background(), &variable, "foo")
|
|
require.NoError(t, diags.Error())
|
|
assert.Equal(t, variable.Value, "default")
|
|
}
|
|
|
|
func TestSetVariableWhenAlreadyAValueIsAssigned(t *testing.T) {
|
|
defaultVal := "default"
|
|
val := "assigned-value"
|
|
variable := variable.Variable{
|
|
Description: "a test variable",
|
|
Default: defaultVal,
|
|
Value: val,
|
|
}
|
|
|
|
// since a value is already assigned to the variable, it would not be overridden
|
|
// by the default value
|
|
diags := setVariable(context.Background(), &variable, "foo")
|
|
require.NoError(t, diags.Error())
|
|
assert.Equal(t, variable.Value, "assigned-value")
|
|
}
|
|
|
|
func TestSetVariableEnvVarValueDoesNotOverridePresetValue(t *testing.T) {
|
|
defaultVal := "default"
|
|
val := "assigned-value"
|
|
variable := variable.Variable{
|
|
Description: "a test variable",
|
|
Default: defaultVal,
|
|
Value: val,
|
|
}
|
|
|
|
// set value for variable as an environment variable
|
|
t.Setenv("BUNDLE_VAR_foo", "process-env")
|
|
|
|
// since a value is already assigned to the variable, it would not be overridden
|
|
// by the value from environment
|
|
diags := setVariable(context.Background(), &variable, "foo")
|
|
require.NoError(t, diags.Error())
|
|
assert.Equal(t, variable.Value, "assigned-value")
|
|
}
|
|
|
|
func TestSetVariablesErrorsIfAValueCouldNotBeResolved(t *testing.T) {
|
|
variable := variable.Variable{
|
|
Description: "a test variable with no default",
|
|
}
|
|
|
|
// fails because we could not resolve a value for the variable
|
|
diags := setVariable(context.Background(), &variable, "foo")
|
|
assert.ErrorContains(t, diags.Error(), "no value assigned to required variable foo. Assignment can be done through the \"--var\" flag or by setting the BUNDLE_VAR_foo environment variable")
|
|
}
|
|
|
|
func TestSetVariablesMutator(t *testing.T) {
|
|
defaultValForA := "default-a"
|
|
defaultValForB := "default-b"
|
|
valForC := "assigned-val-c"
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: map[string]*variable.Variable{
|
|
"a": {
|
|
Description: "resolved to default value",
|
|
Default: defaultValForA,
|
|
},
|
|
"b": {
|
|
Description: "resolved from environment vairables",
|
|
Default: defaultValForB,
|
|
},
|
|
"c": {
|
|
Description: "has already been assigned a value",
|
|
Value: valForC,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
t.Setenv("BUNDLE_VAR_b", "env-var-b")
|
|
|
|
diags := bundle.Apply(context.Background(), b, SetVariables())
|
|
require.NoError(t, diags.Error())
|
|
assert.Equal(t, "default-a", b.Config.Variables["a"].Value)
|
|
assert.Equal(t, "env-var-b", b.Config.Variables["b"].Value)
|
|
assert.Equal(t, "assigned-val-c", b.Config.Variables["c"].Value)
|
|
}
|
|
|
|
func TestSetComplexVariablesViaEnvVariablesIsNotAllowed(t *testing.T) {
|
|
defaultVal := "default"
|
|
variable := variable.Variable{
|
|
Description: "a test variable",
|
|
Default: defaultVal,
|
|
Type: variable.VariableTypeComplex,
|
|
}
|
|
|
|
// set value for variable as an environment variable
|
|
t.Setenv("BUNDLE_VAR_foo", "process-env")
|
|
|
|
diags := setVariable(context.Background(), &variable, "foo")
|
|
assert.ErrorContains(t, diags.Error(), "setting via environment variables (BUNDLE_VAR_foo) is not supported for complex variable foo")
|
|
}
|