mirror of https://github.com/databricks/cli.git
54 lines
1.5 KiB
Go
54 lines
1.5 KiB
Go
package terraform
|
|
|
|
import (
|
|
"context"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform-exec/tfexec"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestSomething(t *testing.T) {
|
|
ctx := context.Background()
|
|
tf, err := newTerraform(ctx, "testdata/simplest", map[string]string{
|
|
"DATABRICKS_HOST": "..",
|
|
"DATABRICKS_TOKEN": "..",
|
|
})
|
|
assert.NoError(t, err)
|
|
|
|
err = tf.Init(ctx)
|
|
assert.NoError(t, err)
|
|
|
|
planLoc := path.Join(t.TempDir(), "tfplan.zip")
|
|
hasChanges, err := tf.Plan(ctx, tfexec.Out(planLoc))
|
|
assert.True(t, hasChanges)
|
|
assert.NoError(t, err)
|
|
|
|
plan, err := tf.ShowPlanFile(ctx, planLoc)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, plan)
|
|
|
|
found := false
|
|
for _, r := range plan.Config.RootModule.Resources {
|
|
// TODO: add validator to prevent non-Databricks resources in *.tf files, as
|
|
// we're not replacing terraform, we're wrapping it up for the average user.
|
|
if r.Type != "databricks_job" {
|
|
continue
|
|
}
|
|
// TODO: validate that libraries on jobs defined in *.tf and libraries
|
|
// in `install_requires` defined in setup.py are the same. Exist with
|
|
// the explanatory error otherwise.
|
|
found = true
|
|
// resource "databricks_job" "this"
|
|
assert.Equal(t, "this", r.Name)
|
|
// this is just a PoC to show how to retrieve DBR version from definitions.
|
|
// production code should perform rigorous nil checks...
|
|
nc := r.Expressions["new_cluster"]
|
|
firstBlock := nc.NestedBlocks[0]
|
|
ver := firstBlock["spark_version"].ConstantValue.(string)
|
|
assert.Equal(t, "10.0.1", ver)
|
|
}
|
|
assert.True(t, found)
|
|
}
|