update unit tests

Author: vandyliu

internal/provider/datasources/data_source_workspaces.go

@@ -3,9 +3,10 @@ package datasources
 import (
 	"context"
 	"fmt"
+	"strings"
+
 	"github.com/hashicorp/terraform-plugin-framework/attr"
 	"github.com/samber/lo"
-	"strings"
 
 	"github.com/astronomer/astronomer-terraform-provider/internal/clients"
 	"github.com/astronomer/astronomer-terraform-provider/internal/clients/platform"

internal/provider/models/subject_profile.go

@@ -2,6 +2,7 @@ package models
 
 import (
 	"context"
+
 	"github.com/astronomer/astronomer-terraform-provider/internal/provider/schemas"
 
 	"github.com/astronomer/astronomer-terraform-provider/internal/clients/iam"

internal/provider/models/workspaces.go

@@ -2,6 +2,7 @@ package models
 
 import (
 	"context"
+
 	"github.com/astronomer/astronomer-terraform-provider/internal/clients/platform"
 	"github.com/astronomer/astronomer-terraform-provider/internal/provider/schemas"
 	"github.com/hashicorp/terraform-plugin-framework/attr"

internal/provider/provider_test.go

@@ -43,6 +43,7 @@ var _ = Describe("Provider Test", func() {
 	It("has expected data sources", func() {
 		expectedDataSources := []string{
 			"astronomer_workspace",
+			"astronomer_workspaces",
 		}
 
 		dataSources := p.DataSources(ctx)

internal/provider/schemas/subject_profile_test.go

@@ -2,6 +2,7 @@ package schemas_test
 
 import (
 	"context"
+
 	"github.com/astronomer/astronomer-terraform-provider/internal/provider/schemas"
 
 	"github.com/astronomer/astronomer-terraform-provider/internal/clients/iam"