mirror of https://github.com/databricks/cli.git
43 lines
1.1 KiB
Go
43 lines
1.1 KiB
Go
package mutator_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/databricks/cli/bundle"
|
|
"github.com/databricks/cli/bundle/config"
|
|
"github.com/databricks/cli/bundle/config/mutator"
|
|
"github.com/databricks/cli/bundle/config/variable"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestInitializeVariables(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: map[string]*variable.Variable{
|
|
"foo": nil,
|
|
"bar": {
|
|
Description: "This is a description",
|
|
},
|
|
},
|
|
},
|
|
}
|
|
err := bundle.Apply(context.Background(), b, mutator.InitializeVariables())
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, b.Config.Variables["foo"])
|
|
assert.NotNil(t, b.Config.Variables["bar"])
|
|
assert.Equal(t, "This is a description", b.Config.Variables["bar"].Description)
|
|
}
|
|
|
|
func TestInitializeVariablesWithoutVariables(t *testing.T) {
|
|
b := &bundle.Bundle{
|
|
Config: config.Root{
|
|
Variables: nil,
|
|
},
|
|
}
|
|
err := bundle.Apply(context.Background(), b, mutator.InitializeVariables())
|
|
require.NoError(t, err)
|
|
assert.Nil(t, b.Config.Variables)
|
|
}
|