From f8d3c1d50cb0eddaa466d4bec2c5bb1e528c8f9b Mon Sep 17 00:00:00 2001 From: Shreyas Goenka Date: Wed, 19 Apr 2023 16:32:50 +0200 Subject: [PATCH] Added tests --- bundle/deploy/terraform/show_events_test.go | 62 +++++++++++++++++++++ 1 file changed, 62 insertions(+) create mode 100644 bundle/deploy/terraform/show_events_test.go diff --git a/bundle/deploy/terraform/show_events_test.go b/bundle/deploy/terraform/show_events_test.go new file mode 100644 index 00000000..6f039fa7 --- /dev/null +++ b/bundle/deploy/terraform/show_events_test.go @@ -0,0 +1,62 @@ +package terraform + +import ( + "testing" + + tfjson "github.com/hashicorp/terraform-json" + "github.com/stretchr/testify/assert" +) + +func TestResourceChangeEventForCreateJob(t *testing.T) { + tfChange := tfjson.ResourceChange{ + Change: &tfjson.Change{ + Actions: []tfjson.Action{"create"}, + }, + Type: "databricks_job", + Name: "foo", + } + bundleChange := toResourceChangeEvent(&tfChange) + + assert.Equal(t, "create", string(bundleChange.Action)) + assert.Equal(t, "job", string(bundleChange.ResourceType)) + assert.Equal(t, "foo", string(bundleChange.Name)) +} + +func TestResourceChangeEventForReplacePipeline(t *testing.T) { + tfChange := tfjson.ResourceChange{ + Change: &tfjson.Change{ + Actions: []tfjson.Action{"delete", "create"}, + }, + Type: "databricks_pipeline", + Name: "bar", + } + bundleChange := toResourceChangeEvent(&tfChange) + + assert.Equal(t, "replace", string(bundleChange.Action)) + assert.Equal(t, "pipeline", string(bundleChange.ResourceType)) + assert.Equal(t, "bar", string(bundleChange.Name)) +} + +func TesResourceChangeEventForDeleteExperiment(t *testing.T) { + tfChange := tfjson.ResourceChange{ + Change: &tfjson.Change{ + Actions: []tfjson.Action{"delete"}, + }, + Type: "databricks_mlflow_experiment", + Name: "my_exp", + } + bundleChange := toResourceChangeEvent(&tfChange) + + assert.Equal(t, "delete", string(bundleChange.Action)) + assert.Equal(t, "mlflow_experiment", string(bundleChange.ResourceType)) + assert.Equal(t, "my_exp", string(bundleChange.Name)) +} + +func TestResourceChangeEventToString(t *testing.T) { + event := ResourceChangeEvent{ + Name: "foo", + ResourceType: "job", + Action: ActionCreate, + } + assert.Equal(t, " create job foo", event.String()) +}