load auto-generated resources and data sources

databricks · Jan 2, 2025 · c5e56ff · c5e56ff
1 parent e11d99a
commit c5e56ff
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 17 deletions.
diff --git a/internal/providers/pluginfw/auto_generation.go b/internal/providers/pluginfw/auto_generation.go
diff --git a/internal/providers/pluginfw/pluginfw_rollout_utils.go b/internal/providers/pluginfw/pluginfw_rollout_utils.go
@@ -41,26 +41,32 @@ var migratedDataSources = []func() datasource.DataSource{
 
 // List of resources that have been onboarded to the plugin framework - not migrated from sdkv2.
 // Keep this list sorted.
-var pluginFwOnlyResources = []func() resource.Resource{
-	app.ResourceApp,
-	sharing.ResourceShare,
-}
+var pluginFwOnlyResources = append(
+	[]func() resource.Resource{
+		app.ResourceApp,
+		sharing.ResourceShare,
+	},
+	autoGeneratedResources...,
+)
 
 // List of data sources that have been onboarded to the plugin framework - not migrated from sdkv2.
 // Keep this list sorted.
-var pluginFwOnlyDataSources = []func() datasource.DataSource{
-	app.DataSourceApp,
-	app.DataSourceApps,
-	catalog.DataSourceFunctions,
-	notificationdestinations.DataSourceNotificationDestinations,
-	registered_model.DataSourceRegisteredModel,
-	registered_model.DataSourceRegisteredModelVersions,
-	serving.DataSourceServingEndpoints,
-	// TODO: Add DataSourceCluster into migratedDataSources after fixing unit tests.
-	cluster.DataSourceCluster, // Using the staging name (with pluginframework suffix)
-	sharing.DataSourceShare,   // Using the staging name (with pluginframework suffix)
-	sharing.DataSourceShares,  // Using the staging name (with pluginframework suffix)
-}
+var pluginFwOnlyDataSources = append(
+	[]func() datasource.DataSource{
+		app.DataSourceApp,
+		app.DataSourceApps,
+		catalog.DataSourceFunctions,
+		notificationdestinations.DataSourceNotificationDestinations,
+		registered_model.DataSourceRegisteredModel,
+		registered_model.DataSourceRegisteredModelVersions,
+		serving.DataSourceServingEndpoints,
+		// TODO: Add DataSourceCluster into migratedDataSources after fixing unit tests.
+		cluster.DataSourceCluster, // Using the staging name (with pluginframework suffix)
+		sharing.DataSourceShare,   // Using the staging name (with pluginframework suffix)
+		sharing.DataSourceShares,  // Using the staging name (with pluginframework suffix)
+	},
+	autoGeneratedDataSources...,
+)
 
 type sdkV2FallbackOptions struct {
 	resourceFallbacks   []string