Fixed databricks_metastore_assignment resource (#1900)

guillaumestevens · web-flow · commit 5b12bdd68589 · 2023-01-05T17:56:44.000+01:00
* fix metastore assignment

* Update metastore assignment test
diff --git a/catalog/resource_metastore_assignment.go b/catalog/resource_metastore_assignment.go
@@ -34,9 +34,9 @@ func (a MetastoreAssignmentAPI) updateMetastoreAssignment(ma MetastoreAssignment
 }
 
 func (a MetastoreAssignmentAPI) getAssignedMetastoreID() (string, error) {
-	var ma MetastoreAssignment
-	err := a.client.Get(a.context, "/unity-catalog/metastore_summary", nil, &ma)
-	return ma.MetastoreID, err
+	var mi MetastoreInfo
+	err := a.client.Get(a.context, "/unity-catalog/metastore_summary", nil, &mi)
+	return mi.MetastoreID, err
 }
 
 func (a MetastoreAssignmentAPI) deleteMetastoreAssignment(workspaceID, metastoreID string) error {
diff --git a/catalog/resource_metastore_assignment_test.go b/catalog/resource_metastore_assignment_test.go
@@ -27,7 +27,7 @@ func TestMetastoreAssignment_Create(t *testing.T) {
 			{
 				Method:   "GET",
 				Resource: "/api/2.1/unity-catalog/metastore_summary",
-				Response: MetastoreAssignment{
+				Response: MetastoreInfo{
 					MetastoreID: "a",
 				},
 			},

Original file line number	Diff line number	Diff line change
`@@ -34,9 +34,9 @@ func (a MetastoreAssignmentAPI) updateMetastoreAssignment(ma MetastoreAssignment`
`34`	`34`	`}`
`35`	`35`
`36`	`36`	`func (a MetastoreAssignmentAPI) getAssignedMetastoreID() (string, error) {`
`37`		`- var ma MetastoreAssignment`
`38`		`- err := a.client.Get(a.context, "/unity-catalog/metastore_summary", nil, &ma)`
`39`		`- return ma.MetastoreID, err`
	`37`	`+ var mi MetastoreInfo`
	`38`	`+ err := a.client.Get(a.context, "/unity-catalog/metastore_summary", nil, &mi)`
	`39`	`+ return mi.MetastoreID, err`
`40`	`40`	`}`
`41`	`41`
`42`	`42`	`func (a MetastoreAssignmentAPI) deleteMetastoreAssignment(workspaceID, metastoreID string) error {`
Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ func TestMetastoreAssignment_Create(t *testing.T) {`
`27`	`27`	`{`
`28`	`28`	`Method: "GET",`
`29`	`29`	`Resource: "/api/2.1/unity-catalog/metastore_summary",`
`30`		`- Response: MetastoreAssignment{`
	`30`	`+ Response: MetastoreInfo{`
`31`	`31`	`MetastoreID: "a",`
`32`	`32`	`},`
`33`	`33`	`},`