chore: Adjust data source tests follow up 2 (snowflakedb#4165)

Follow up for
snowflakedb#4115,
addressing the following data sources
- `secrets`
- `security integrations`
- `streamlits`
- `schemas`

Author: sfc-gh-jcieslak

pkg/acceptance/bettertestspoc/assert/objectassert/gen/sdk_object_def.go

@@ -173,6 +173,11 @@ var allStructs = []SdkObjectDef{
 		ObjectType:   sdk.ObjectTypeSecurityIntegration,
 		ObjectStruct: sdk.SecurityIntegration{},
 	},
+	{
+		IdType:       "sdk.DatabaseObjectIdentifier",
+		ObjectType:   sdk.ObjectTypeSchema,
+		ObjectStruct: sdk.Schema{},
+	},
 	{
 		IdType:       "sdk.SchemaObjectIdentifier",
 		ObjectType:   sdk.ObjectTypeStreamlit,

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/schema_show_output_ext.go

@@ -14,6 +14,11 @@ func (s *SchemaShowOutputAssert) HasOwnerNotEmpty() *SchemaShowOutputAssert {
 	return s
 }
 
+func (s *SchemaShowOutputAssert) HasRetentionTimeNotEmpty() *SchemaShowOutputAssert {
+	s.AddAssertion(assert.ResourceShowOutputValuePresent("retention_time"))
+	return s
+}
+
 func (s *SchemaShowOutputAssert) HasOwnerRoleTypeNotEmpty() *SchemaShowOutputAssert {
 	s.AddAssertion(assert.ResourceShowOutputValuePresent("owner_role_type"))
 	return s

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/schemas_datasource_show_output_ext.go

@@ -0,0 +1,17 @@
+package resourceshowoutputassert
+
+import (
+	"testing"
+
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/acceptance/bettertestspoc/assert"
+)
+
+func SchemasDatasourceShowOutput(t *testing.T, datasourceReference string) *SchemaShowOutputAssert {
+	t.Helper()
+
+	s := SchemaShowOutputAssert{
+		ResourceAssert: assert.NewDatasourceAssert(datasourceReference, "show_output", "schemas.0."),
+	}
+	s.AddAssertion(assert.ValueSet("show_output.#", "1"))
+	return &s
+}

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/secrets_datasource_show_output_ext.go

@@ -7,11 +7,11 @@ import (
 )
 
 // SecretsDatasourceShowOutput is a temporary workaround to have better show output assertions in data source acceptance tests.
-func SecretsDatasourceShowOutput(t *testing.T, name string) *SecretShowOutputAssert {
+func SecretsDatasourceShowOutput(t *testing.T, datasourceReference string) *SecretShowOutputAssert {
 	t.Helper()
 
 	s := SecretShowOutputAssert{
-		ResourceAssert: assert.NewDatasourceAssert("data."+name, "show_output", "secrets.0."),
+		ResourceAssert: assert.NewDatasourceAssert(datasourceReference, "show_output", "secrets.0."),
 	}
 	s.AddAssertion(assert.ValueSet("show_output.#", "1"))
 	return &s

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/security_integration_show_output_ext.go

@@ -0,0 +1,8 @@
+package resourceshowoutputassert
+
+import "github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/acceptance/bettertestspoc/assert"
+
+func (s *SecurityIntegrationShowOutputAssert) HasCreatedOnNotEmpty() *SecurityIntegrationShowOutputAssert {
+	s.AddAssertion(assert.ValuePresent("show_output.0.created_on"))
+	return s
+}

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/security_integrations_datasource_show_output_ext.go

@@ -0,0 +1,16 @@
+package resourceshowoutputassert
+
+import (
+	"testing"
+
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/acceptance/bettertestspoc/assert"
+)
+
+func SecurityIntegrationsDatasourceShowOutput(t *testing.T, datasourceReference string) *SecurityIntegrationShowOutputAssert {
+	t.Helper()
+	s := SecurityIntegrationShowOutputAssert{
+		ResourceAssert: assert.NewDatasourceAssert(datasourceReference, "show_output", "security_integrations.0."),
+	}
+	s.AddAssertion(assert.ValueSet("show_output.#", "1"))
+	return &s
+}

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/streamlit_show_output_ext.go

@@ -13,3 +13,8 @@ func (s *StreamlitShowOutputAssert) HasUrlIdNotEmpty() *StreamlitShowOutputAsser
 	s.AddAssertion(assert.ResourceShowOutputValuePresent("url_id"))
 	return s
 }
+
+func (s *StreamlitShowOutputAssert) HasOwnerNotEmpty() *StreamlitShowOutputAssert {
+	s.AddAssertion(assert.ResourceShowOutputValuePresent("owner"))
+	return s
+}

pkg/acceptance/bettertestspoc/assert/resourceshowoutputassert/streamlits_datasource_show_output_ext.go

@@ -0,0 +1,17 @@
+package resourceshowoutputassert
+
+import (
+	"testing"
+
+	"github.com/Snowflake-Labs/terraform-provider-snowflake/pkg/acceptance/bettertestspoc/assert"
+)
+
+func StreamlitsDatasourceShowOutput(t *testing.T, datasourceReference string) *StreamlitShowOutputAssert {
+	t.Helper()
+
+	s := StreamlitShowOutputAssert{
+		ResourceAssert: assert.NewDatasourceAssert(datasourceReference, "show_output", "streamlits.0."),
+	}
+	s.AddAssertion(assert.ValueSet("show_output.#", "1"))
+	return &s
+}

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

@@ -13,3 +13,11 @@ func (s *StreamlitsModel) WithInDatabase(databaseId sdk.AccountObjectIdentifier)
 		}),
 	)
 }
+
+func (s *StreamlitsModel) WithInSchema(schemaId sdk.DatabaseObjectIdentifier) *StreamlitsModel {
+	return s.WithInValue(
+		tfconfig.ObjectVariable(map[string]tfconfig.Variable{
+			"schema": tfconfig.StringVariable(schemaId.FullyQualifiedName()),
+		}),
+	)
+}