added acceptance tests and added the data source to pluginfw.go

databricks · Oct 24, 2024 · 8db0188 · 8db0188
1 parent 4ab1e27
commit 8db0188
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/internal/providers/pluginfw/pluginfw.go b/internal/providers/pluginfw/pluginfw.go
@@ -21,6 +21,7 @@ import (
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/notificationdestinations"
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/qualitymonitor"
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/registered_model"
+	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/user"
 	"github.com/databricks/terraform-provider-databricks/internal/providers/pluginfw/resources/volume"
 
 	"github.com/hashicorp/terraform-plugin-framework/datasource"
@@ -56,6 +57,7 @@ func (p *DatabricksProviderPluginFramework) DataSources(ctx context.Context) []f
 		volume.DataSourceVolumes,
 		registered_model.DataSourceRegisteredModel,
 		notificationdestinations.DataSourceNotificationDestinations,
+		user.DataSourceUsers,
 	}
 }
 

diff --git a/internal/acceptance/data_users_test.go → ...nfw/resources/user/data_users_acc_test.go b/internal/acceptance/data_users_test.go → ...nfw/resources/user/data_users_acc_test.go
@@ -1,8 +1,9 @@
-package acceptance
+package user_test
 
 import (
 	"testing"
 
+	"github.com/databricks/terraform-provider-databricks/internal/acceptance"
 	"github.com/hashicorp/terraform-plugin-testing/terraform"
 	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
@@ -47,7 +48,14 @@ func checkUsersDataSourcePopulated(t *testing.T) func(s *terraform.State) error
 }
 
 func TestAccDataSourceDataUsers(t *testing.T) {
-	AccountLevel(t, Step{
+	acceptance.AccountLevel(t, acceptance.Step{
+		Template: dataSourceTemplate,
+		Check:    checkUsersDataSourcePopulated(t),
+	})
+}
+
+func TestWorkspaceDataSourceDataUsers(t *testing.T) {
+	acceptance.WorkspaceLevel(t, acceptance.Step{
 		Template: dataSourceTemplate,
 		Check:    checkUsersDataSourcePopulated(t),
 	})