machine_learning - Add new machine_learning block that supports purge_soft_deleted_workspace_on_destroy

internal/features/defaults.go

@@ -60,6 +60,9 @@ func Default() UserFeatures {
 		PostgresqlFlexibleServer: PostgresqlFlexibleServerFeatures{
 			RestartServerOnConfigurationValueChange: true,
 		},
+		MachineLearning: MachineLearningFeatures{
+			PurgeSoftDeletedWorkspaceOnDestroy: false,
+		},
 		RecoveryService: RecoveryServiceFeatures{
 			VMBackupStopProtectionAndRetainDataOnDestroy: false,
 			PurgeProtectedItemsFromVaultOnDestroy:        false,

internal/features/user_flags.go

@@ -17,6 +17,7 @@ type UserFeatures struct {
 	ManagedDisk              ManagedDiskFeatures
 	Subscription             SubscriptionFeatures
 	PostgresqlFlexibleServer PostgresqlFlexibleServerFeatures
+	MachineLearning          MachineLearningFeatures
 	RecoveryService          RecoveryServiceFeatures
 }
 
@@ -87,6 +88,10 @@ type PostgresqlFlexibleServerFeatures struct {
 	RestartServerOnConfigurationValueChange bool
 }
 
+type MachineLearningFeatures struct {
+	PurgeSoftDeletedWorkspaceOnDestroy bool
+}
+
 type RecoveryServiceFeatures struct {
 	VMBackupStopProtectionAndRetainDataOnDestroy bool
 	PurgeProtectedItemsFromVaultOnDestroy        bool

internal/provider/features.go

@@ -304,6 +304,20 @@ func schemaFeatures(supportLegacyTestSuite bool) *pluginsdk.Schema {
 				},
 			},
 		},
+		"machine_learning": {
+			Type:     pluginsdk.TypeList,
+			Optional: true,
+			MaxItems: 1,
+			Elem: &pluginsdk.Resource{
+				Schema: map[string]*pluginsdk.Schema{
+					"purge_soft_deleted_workspace_on_destroy": {
+						Type:     pluginsdk.TypeBool,
+						Optional: true,
+						Default:  false,
+					},
+				},
+			},
+		},
 
 		"recovery_service": {
 			Type:     pluginsdk.TypeList,
@@ -534,6 +548,16 @@ func expandFeatures(input []interface{}) features.UserFeatures {
 		}
 	}
 
+	if raw, ok := val["machine_learning"]; ok {
+		items := raw.([]interface{})
+		if len(items) > 0 {
+			subscriptionRaw := items[0].(map[string]interface{})
+			if v, ok := subscriptionRaw["purge_soft_deleted_workspace_on_destroy"]; ok {
+				featuresMap.MachineLearning.PurgeSoftDeletedWorkspaceOnDestroy = v.(bool)
+			}
+		}
+	}
+
 	if raw, ok := val["recovery_service"]; ok {
 		items := raw.([]interface{})
 		if len(items) > 0 {

internal/provider/features_test.go

@@ -75,6 +75,9 @@ func TestExpandFeatures(t *testing.T) {
 				PostgresqlFlexibleServer: features.PostgresqlFlexibleServerFeatures{
 					RestartServerOnConfigurationValueChange: true,
 				},
+				MachineLearning: features.MachineLearningFeatures{
+					PurgeSoftDeletedWorkspaceOnDestroy: false,
+				},
 				RecoveryService: features.RecoveryServiceFeatures{
 					VMBackupStopProtectionAndRetainDataOnDestroy: false,
 					PurgeProtectedItemsFromVaultOnDestroy:        false,
@@ -165,6 +168,11 @@ func TestExpandFeatures(t *testing.T) {
 							"scale_to_zero_before_deletion": true,
 						},
 					},
+					"machine_learning": []interface{}{
+						map[string]interface{}{
+							"purge_soft_deleted_workspace_on_destroy": true,
+						},
+					},
 					"recovery_service": []interface{}{
 						map[string]interface{}{
 							"vm_backup_stop_protection_and_retain_data_on_destroy": true,
@@ -228,6 +236,9 @@ func TestExpandFeatures(t *testing.T) {
 				PostgresqlFlexibleServer: features.PostgresqlFlexibleServerFeatures{
 					RestartServerOnConfigurationValueChange: true,
 				},
+				MachineLearning: features.MachineLearningFeatures{
+					PurgeSoftDeletedWorkspaceOnDestroy: true,
+				},
 				RecoveryService: features.RecoveryServiceFeatures{
 					VMBackupStopProtectionAndRetainDataOnDestroy: true,
 					PurgeProtectedItemsFromVaultOnDestroy:        true,
@@ -318,6 +329,11 @@ func TestExpandFeatures(t *testing.T) {
 							"scale_to_zero_before_deletion": false,
 						},
 					},
+					"machine_learning": []interface{}{
+						map[string]interface{}{
+							"purge_soft_deleted_workspace_on_destroy": false,
+						},
+					},
 					"recovery_service": []interface{}{
 						map[string]interface{}{
 							"vm_backup_stop_protection_and_retain_data_on_destroy": false,
@@ -381,6 +397,9 @@ func TestExpandFeatures(t *testing.T) {
 				PostgresqlFlexibleServer: features.PostgresqlFlexibleServerFeatures{
 					RestartServerOnConfigurationValueChange: false,
 				},
+				MachineLearning: features.MachineLearningFeatures{
+					PurgeSoftDeletedWorkspaceOnDestroy: false,
+				},
 				RecoveryService: features.RecoveryServiceFeatures{
 					VMBackupStopProtectionAndRetainDataOnDestroy: false,
 					PurgeProtectedItemsFromVaultOnDestroy:        false,
@@ -1383,6 +1402,71 @@ func TestExpandFeaturesPosgresqlFlexibleServer(t *testing.T) {
 	}
 }
 
+func TestExpandFeaturesMachineLearning(t *testing.T) {
+	testData := []struct {
+		Name     string
+		Input    []interface{}
+		EnvVars  map[string]interface{}
+		Expected features.UserFeatures
+	}{
+		{
+			Name: "Empty Block",
+			Input: []interface{}{
+				map[string]interface{}{
+					"machine_learning": []interface{}{},
+				},
+			},
+			Expected: features.UserFeatures{
+				MachineLearning: features.MachineLearningFeatures{
+					PurgeSoftDeletedWorkspaceOnDestroy: false,
+				},
+			},
+		},
+		{
+			Name: "MachineLearning Workspace purge soft delete on destroy",
+			Input: []interface{}{
+				map[string]interface{}{
+					"machine_learning": []interface{}{
+						map[string]interface{}{
+							"purge_soft_deleted_workspace_on_destroy": true,
+						},
+					},
+				},
+			},
+			Expected: features.UserFeatures{
+				MachineLearning: features.MachineLearningFeatures{
+					PurgeSoftDeletedWorkspaceOnDestroy: true,
+				},
+			},
+		},
+		{
+			Name: "MachineLearning Workspace does not purge soft delete on destroy",
+			Input: []interface{}{
+				map[string]interface{}{
+					"machine_learning": []interface{}{
+						map[string]interface{}{
+							"purge_soft_deleted_workspace_on_destroy": false,
+						},
+					},
+				},
+			},
+			Expected: features.UserFeatures{
+				MachineLearning: features.MachineLearningFeatures{
+					PurgeSoftDeletedWorkspaceOnDestroy: false,
+				},
+			},
+		},
+	}
+
+	for _, testCase := range testData {
+		t.Logf("[DEBUG] Test Case: %q", testCase.Name)
+		result := expandFeatures(testCase.Input)
+		if !reflect.DeepEqual(result.Subscription, testCase.Expected.Subscription) {
+			t.Fatalf("Expected %+v but got %+v", result.Subscription, testCase.Expected.Subscription)
+		}
+	}
+}
+
 func TestExpandFeaturesRecoveryService(t *testing.T) {
 	testData := []struct {
 		Name     string

internal/services/machinelearning/machine_learning_workspace_resource.go

@@ -461,7 +461,14 @@ func resourceMachineLearningWorkspaceDelete(d *pluginsdk.ResourceData, meta inte
 		return fmt.Errorf("parsing Machine Learning Workspace ID `%q`: %+v", d.Id(), err)
 	}
 
-	future, err := client.Delete(ctx, *id, workspaces.DefaultDeleteOperationOptions())
+	options := workspaces.DefaultDeleteOperationOptions()
+	if meta.(*clients.Client).Features.MachineLearning.PurgeSoftDeletedWorkspaceOnDestroy {
+		options = workspaces.DeleteOperationOptions{
+			ForceToPurge: pointer.To(true),
+		}
+	}
+
+	future, err := client.Delete(ctx, *id, options)
 	if err != nil {
 		return fmt.Errorf("deleting Machine Learning Workspace %q (Resource Group %q): %+v", id.WorkspaceName, id.ResourceGroupName, err)
 	}