Update from OpenAPI (#328)

databricks · Mar 8, 2023 · 3adcf9c · 3adcf9c
1 parent e54da35
commit 3adcf9c
Show file tree

Hide file tree

Showing 28 changed files with 455 additions and 833 deletions.
diff --git a/account_client.go b/account_client.go
diff --git a/internal/billing_test.go b/internal/billing_test.go
@@ -29,8 +29,8 @@ func TestMwsAccLogDelivery(t *testing.T) {
 	}
 	creds, err := a.Credentials.Create(ctx, deployment.CreateCredentialRequest{
 		CredentialsName: RandomName("sdk-"),
-		AwsCredentials: deployment.AwsCredentials{
-			StsRole: &deployment.StsRole{
+		AwsCredentials: deployment.CreateCredentialAwsCredentials{
+			StsRole: &deployment.CreateCredentialStsRole{
 				RoleArn: GetEnvOrSkipTest(t, "TEST_LOGDELIVERY_ARN"),
 			},
 		},

diff --git a/internal/dbsql_test.go b/internal/dbsql_test.go
@@ -1,7 +1,6 @@
 package internal
 
 import (
-	"strconv"
 	"testing"
 
 	"github.com/databricks/databricks-sdk-go/service/sql"
@@ -105,37 +104,6 @@ func TestAccAlerts(t *testing.T) {
 	require.NoError(t, err)
 	assert.Equal(t, len(all), len(names))
 	assert.Equal(t, alert.Id, names[byId.Name])
-
-	schedule, err := w.Alerts.CreateSchedule(ctx, sql.CreateRefreshSchedule{
-		AlertId:      alert.Id,
-		Cron:         "5 4 * * *",
-		DataSourceId: srcs[0].Id,
-	})
-	require.NoError(t, err)
-	defer w.Alerts.DeleteScheduleByAlertIdAndScheduleId(ctx, alert.Id, schedule.Id)
-
-	schedules, err := w.Alerts.ListSchedulesByAlertId(ctx, alert.Id)
-	require.NoError(t, err)
-	assert.True(t, len(schedules) >= 1)
-
-	me, err := w.CurrentUser.Me(ctx)
-	require.NoError(t, err)
-
-	userId, err := strconv.ParseInt(me.Id, 10, 64)
-	require.NoError(t, err)
-
-	sub, err := w.Alerts.Subscribe(ctx, sql.CreateSubscription{
-		AlertId: alert.Id,
-		UserId:  userId,
-	})
-	require.NoError(t, err)
-
-	allSubs, err := w.Alerts.GetSubscriptionsByAlertId(ctx, alert.Id)
-	require.NoError(t, err)
-	assert.True(t, len(allSubs) >= 1)
-
-	err = w.Alerts.UnsubscribeByAlertIdAndSubscriptionId(ctx, alert.Id, sub.Id)
-	require.NoError(t, err)
 }
 
 func TestAccDashboards(t *testing.T) {

diff --git a/internal/deployment_test.go b/internal/deployment_test.go
@@ -75,8 +75,8 @@ func TestMwsAccCredentials(t *testing.T) {
 	}
 	role, err := a.Credentials.Create(ctx, deployment.CreateCredentialRequest{
 		CredentialsName: RandomName("sdk-"),
-		AwsCredentials: deployment.AwsCredentials{
-			StsRole: &deployment.StsRole{
+		AwsCredentials: deployment.CreateCredentialAwsCredentials{
+			StsRole: &deployment.CreateCredentialStsRole{
 				RoleArn: GetEnvOrSkipTest(t, "TEST_CROSSACCOUNT_ARN"),
 			},
 		},
@@ -217,8 +217,8 @@ func TestMwsAccWorkspaces(t *testing.T) {
 	// See https://github.com/databricks/terraform-provider-databricks/issues/1424
 	role, err := a.Credentials.Create(ctx, deployment.CreateCredentialRequest{
 		CredentialsName: RandomName("go-sdk-"),
-		AwsCredentials: deployment.AwsCredentials{
-			StsRole: &deployment.StsRole{
+		AwsCredentials: deployment.CreateCredentialAwsCredentials{
+			StsRole: &deployment.CreateCredentialStsRole{
 				RoleArn: GetEnvOrSkipTest(t, "TEST_CROSSACCOUNT_ARN"),
 			},
 		},
@@ -244,8 +244,8 @@ func TestMwsAccWorkspaces(t *testing.T) {
 
 	updateRole, err := a.Credentials.Create(ctx, deployment.CreateCredentialRequest{
 		CredentialsName: RandomName("go-sdk-"),
-		AwsCredentials: deployment.AwsCredentials{
-			StsRole: &deployment.StsRole{
+		AwsCredentials: deployment.CreateCredentialAwsCredentials{
+			StsRole: &deployment.CreateCredentialStsRole{
 				RoleArn: GetEnvOrSkipTest(t, "TEST_CROSSACCOUNT_ARN"),
 			},
 		},

diff --git a/service/clusters/api.go b/service/clusters/api.go
diff --git a/service/clusters/interface.go b/service/clusters/interface.go
diff --git a/service/dbfs/api.go b/service/dbfs/api.go
diff --git a/service/dbfs/interface.go b/service/dbfs/interface.go