add test

databricks · May 31, 2022 · 638b9f5 · 638b9f5
1 parent 7c53c42
commit 638b9f5
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/storage/gs_test.go b/storage/gs_test.go
@@ -94,3 +94,22 @@ func TestCreateOrValidateClusterForGoogleStorage_WorksOnDeletedCluster(t *testin
 		assert.Equal(t, "new-cluster", d.Get("cluster_id"))
 	})
 }
+
+func TestCreateOrValidateClusterForGoogleStorage_FailsOnErrorGettingCluster(t *testing.T) {
+	qa.HTTPFixturesApply(t, []qa.HTTPFixture{
+		{
+			Method:   "GET",
+			Resource: "/api/2.0/clusters/get?cluster_id=my-cluster",
+			Status:   500,
+			Response: common.APIError{
+				ErrorCode:  "SERVER_ERROR",
+				StatusCode: 500,
+				Message:    "Server error",
+			},
+		},
+	}, func(ctx context.Context, client *common.DatabricksClient) {
+		d := ResourceMount().TestResourceData()
+		err := createOrValidateClusterForGoogleStorage(ctx, client, d, "my-cluster", "service-account")
+		assert.EqualError(t, err, "cannot get mounting cluster: Server error")
+	})
+}