fix: Fix datasource acceptance tests setup part3

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

@@ -177,4 +177,28 @@ var allResourceSchemaDefs = []ResourceSchemaDef{
 		name:   "ManagedAccount",
 		schema: resources.ManagedAccount().Schema,
 	},
+	{
+		name:   "NetworkPolicy",
+		schema: resources.NetworkPolicy().Schema,
+	},
+	{
+		name:   "ScimSecurityIntegration",
+		schema: resources.SCIMIntegration().Schema,
+	},
+	{
+		name:   "Saml2SecurityIntegration",
+		schema: resources.SAML2Integration().Schema,
+	},
+	{
+		name:   "ApiAuthenticationIntegrationWithAuthorizationCodeGrant",
+		schema: resources.ApiAuthenticationIntegrationWithAuthorizationCodeGrant().Schema,
+	},
+	{
+		name:   "ApiAuthenticationIntegrationWithClientCredentials",
+		schema: resources.ApiAuthenticationIntegrationWithClientCredentials().Schema,
+	},
+	{
+		name:   "ExternalOauthSecurityIntegration",
+		schema: resources.ExternalOauthIntegration().Schema,
+	},
 }

pkg/acceptance/bettertestspoc/config/datasourcemodel/gen/datasource_schema_def.go

@@ -46,4 +46,24 @@ var allDatasourcesSchemaDefs = []DatasourceSchemaDef{
 		name:   "Grants",
 		schema: datasources.Grants().Schema,
 	},
+	{
+		name:   "NetworkPolicies",
+		schema: datasources.NetworkPolicies().Schema,
+	},
+	{
+		name:   "ResourceMonitors",
+		schema: datasources.ResourceMonitors().Schema,
+	},
+	{
+		name:   "Schemas",
+		schema: datasources.Schemas().Schema,
+	},
+	{
+		name:   "Secrets",
+		schema: datasources.Secrets().Schema,
+	},
+	{
+		name:   "SecurityIntegrations",
+		schema: datasources.SecurityIntegrations().Schema,
+	},
 }

pkg/acceptance/bettertestspoc/config/datasourcemodel/schemas_model_ext.go

@@ -0,0 +1,32 @@
+package datasourcemodel
+
+import (
+	tfconfig "github.com/hashicorp/terraform-plugin-testing/config"
+
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/sdk"
+)
+
+func (s *SchemasModel) WithLimit(rows int) *SchemasModel {
+	return s.WithLimitValue(
+		tfconfig.ObjectVariable(map[string]tfconfig.Variable{
+			"rows": tfconfig.IntegerVariable(rows),
+		}),
+	)
+}
+
+func (s *SchemasModel) WithRowsAndFrom(rows int, from string) *SchemasModel {
+	return s.WithLimitValue(
+		tfconfig.ObjectVariable(map[string]tfconfig.Variable{
+			"rows": tfconfig.IntegerVariable(rows),
+			"from": tfconfig.StringVariable(from),
+		}),
+	)
+}
+
+func (s *SchemasModel) WithIn(databaseId sdk.AccountObjectIdentifier) *SchemasModel {
+	return s.WithInValue(
+		tfconfig.ObjectVariable(map[string]tfconfig.Variable{
+			"database": tfconfig.StringVariable(databaseId.Name()),
+		}),
+	)
+}