fix: Fix resource acceptance tests setup part5

pkg/acceptance/bettertestspoc/assert/resourceassert/gen/resource_schema_def.go

@@ -213,4 +213,8 @@ var allResourceSchemaDefs = []ResourceSchemaDef{
 		name:   "SecondaryDatabase",
 		schema: resources.SecondaryDatabase().Schema,
 	},
+	{
+		name:   "SharedDatabase",
+		schema: resources.SharedDatabase().Schema,
+	},
 }

pkg/acceptance/bettertestspoc/config/model/stream_on_view_model_ext.go

@@ -0,0 +1,9 @@
+package model
+
+import (
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk"
+)
+
+func StreamOnViewBase(resourceName string, id sdk.SchemaObjectIdentifier, viewId sdk.SchemaObjectIdentifier) *StreamOnViewModel {
+	return StreamOnView(resourceName, id.DatabaseName(), id.Name(), id.SchemaName(), viewId.FullyQualifiedName())
+}

pkg/acceptance/bettertestspoc/config/model/warehouse_model_ext.go

@@ -1,14 +1,33 @@
 package model
 
 import (
+	r "github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/resources"
+
 	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk"
 )
 
 func BasicWarehouseModel(
-	name string,
+	id sdk.AccountObjectIdentifier,
+	comment string,
+) *WarehouseModel {
+	return WarehouseWithDefaultMeta(id.Name()).WithComment(comment)
+}
+
+func WarehouseSnowflakeDefaultWithoutParameters(
+	id sdk.AccountObjectIdentifier,
 	comment string,
 ) *WarehouseModel {
-	return WarehouseWithDefaultMeta(name).WithComment(comment)
+	return BasicWarehouseModel(id, comment).
+		WithWarehouseTypeEnum(sdk.WarehouseTypeStandard).
+		WithWarehouseSizeEnum(sdk.WarehouseSizeXSmall).
+		WithMinClusterCount(1).
+		WithMaxClusterCount(1).
+		WithScalingPolicyEnum(sdk.ScalingPolicyStandard).
+		WithAutoSuspend(600).
+		WithAutoResume(r.BooleanTrue).
+		WithInitiallySuspended(false).
+		WithEnableQueryAcceleration(r.BooleanFalse).
+		WithQueryAccelerationMaxScaleFactor(8)
 }
 
 // TODO [SNOW-1501905]: currently config builder are generated from the resource schema, so there is no direct connection to the source enum (like sdk.WarehouseSize)
@@ -17,3 +36,11 @@ func BasicWarehouseModel(
 func (w *WarehouseModel) WithWarehouseSizeEnum(warehouseSize sdk.WarehouseSize) *WarehouseModel {
 	return w.WithWarehouseSize(string(warehouseSize))
 }
+
+func (w *WarehouseModel) WithWarehouseTypeEnum(warehouseType sdk.WarehouseType) *WarehouseModel {
+	return w.WithWarehouseType(string(warehouseType))
+}
+
+func (w *WarehouseModel) WithScalingPolicyEnum(scalingPolicy sdk.ScalingPolicy) *WarehouseModel {
+	return w.WithScalingPolicy(string(scalingPolicy))
+}

pkg/acceptance/helpers/database_client.go

@@ -181,25 +181,12 @@ func (c *DatabaseClient) Describe(t *testing.T, id sdk.AccountObjectIdentifier)
 func (c *DatabaseClient) CreateDatabaseFromShareTemporarily(t *testing.T, externalShareId sdk.ExternalObjectIdentifier) {
 	t.Helper()
 
-	databaseId := c.ids.RandomAccountObjectIdentifier()
-	err := c.client().CreateShared(context.Background(), databaseId, externalShareId, new(sdk.CreateSharedDatabaseOptions))
-	require.NoError(t, err)
-
-	require.Eventually(t, func() bool {
-		database, err := c.Show(t, databaseId)
-		if err != nil {
-			return false
-		}
-		// Origin is returned as "<revoked>" in those cases, because it's not valid sdk.ExternalObjectIdentifier parser sets it as nil.
-		// Once it turns into valid sdk.ExternalObjectIdentifier, we're ready to proceed with the actual test.
-		return database.Origin != nil
-	}, time.Minute, time.Second*6)
+	db, _ := c.CreateDatabaseFromShare(t, externalShareId)
 
-	err = c.DropDatabase(t, databaseId)
+	err := c.DropDatabase(t, db.ID())
 	require.NoError(t, err)
 }
 
-// CreateDatabaseFromShare logic is duplicated from CreateDatabaseFromShareTemporarily.
 func (c *DatabaseClient) CreateDatabaseFromShare(t *testing.T, externalShareId sdk.ExternalObjectIdentifier) (*sdk.Database, func()) {
 	t.Helper()
 

pkg/acceptance/helpers/schema_client.go

@@ -85,6 +85,16 @@ func (c *SchemaClient) UpdateDataRetentionTime(t *testing.T, id sdk.DatabaseObje
 	})
 }
 
+func (c *SchemaClient) UnsetDataRetentionTime(t *testing.T, id sdk.DatabaseObjectIdentifier) {
+	t.Helper()
+
+	c.Alter(t, id, &sdk.AlterSchemaOptions{
+		Unset: &sdk.SchemaUnset{
+			DataRetentionTimeInDays: sdk.Bool(true),
+		},
+	})
+}
+
 func (c *SchemaClient) Show(t *testing.T, id sdk.DatabaseObjectIdentifier) (*sdk.Schema, error) {
 	t.Helper()
 	ctx := context.Background()

pkg/acceptance/helpers/security_integration_client.go

@@ -26,11 +26,16 @@ func (c *SecurityIntegrationClient) client() sdk.SecurityIntegrations {
 }
 
 func (c *SecurityIntegrationClient) CreateApiAuthenticationWithClientCredentialsFlow(t *testing.T) (*sdk.SecurityIntegration, func()) {
+	t.Helper()
+	return c.CreateApiAuthenticationWithClientCredentialsFlowWithEnabled(t, false)
+}
+
+func (c *SecurityIntegrationClient) CreateApiAuthenticationWithClientCredentialsFlowWithEnabled(t *testing.T, enabled bool) (*sdk.SecurityIntegration, func()) {
 	t.Helper()
 	ctx := context.Background()
 
 	id := c.ids.RandomAccountObjectIdentifier()
-	request := sdk.NewCreateApiAuthenticationWithClientCredentialsFlowSecurityIntegrationRequest(id, false, "foo", "foo")
+	request := sdk.NewCreateApiAuthenticationWithClientCredentialsFlowSecurityIntegrationRequest(id, enabled, "foo", "foo")
 	err := c.client().CreateApiAuthenticationWithClientCredentialsFlow(ctx, request)
 	require.NoError(t, err)