changes after review

pkg/sdk/databases.go

@@ -34,6 +34,7 @@ type Databases interface {
 	Show(ctx context.Context, opts *ShowDatabasesOptions) ([]Database, error)
 	ShowByID(ctx context.Context, id AccountObjectIdentifier) (*Database, error)
 	Describe(ctx context.Context, id AccountObjectIdentifier) (*DatabaseDetails, error)
+	Use(ctx context.Context, id AccountObjectIdentifier) error
 }
 
 var _ Databases = (*databases)(nil)
@@ -682,3 +683,9 @@ func (v *databases) Describe(ctx context.Context, id AccountObjectIdentifier) (*
 	}
 	return &details, err
 }
+
+// Use is based on https://docs.snowflake.com/en/sql-reference/sql/use-database.
+func (v *databases) Use(ctx context.Context, id AccountObjectIdentifier) error {
+	// proxy to sessions
+	return v.client.Sessions.UseDatabase(ctx, id)
+}

pkg/sdk/testint/databases_integration_test.go

@@ -200,10 +200,7 @@ func TestInt_DatabasesCreateShared(t *testing.T) {
 		},
 	})
 	require.NoError(t, err)
-	t.Cleanup(func() {
-		err = client.Databases.Drop(ctx, sharedDatabase.ID(), nil)
-		require.NoError(t, err)
-	})
+	t.Cleanup(testClientHelper().Database.DropDatabaseFunc(t, sharedDatabase.ID()))
 
 	database, err := client.Databases.ShowByID(ctx, sharedDatabase.ID())
 	require.NoError(t, err)
@@ -270,10 +267,7 @@ func TestInt_DatabasesCreateSecondary(t *testing.T) {
 		Comment:                    sdk.String(comment),
 	})
 	require.NoError(t, err)
-	t.Cleanup(func() {
-		err = client.Databases.Drop(ctx, sharedDatabase.ID(), nil)
-		require.NoError(t, err)
-	})
+	t.Cleanup(testClientHelper().Database.DropDatabaseFunc(t, sharedDatabase.ID()))
 
 	database, err := client.Databases.ShowByID(ctx, sharedDatabase.ID())
 	require.NoError(t, err)