New resource azurerm_azure_ai_services

internal/features/defaults.go

@@ -20,6 +20,9 @@ func Default() UserFeatures {
 		CognitiveAccount: CognitiveAccountFeatures{
 			PurgeSoftDeleteOnDestroy: true,
 		},
+		CognitiveAccountAIServices: CognitiveAccountAIServicesFeatures{
+			PurgeSoftDeleteOnDestroy: true,
+		},
 		KeyVault: KeyVaultFeatures{
 			PurgeSoftDeleteOnDestroy:         true,
 			PurgeSoftDeletedKeysOnDestroy:    true,

internal/features/user_flags.go

@@ -4,28 +4,33 @@
 package features
 
 type UserFeatures struct {
-	ApiManagement            ApiManagementFeatures
-	AppConfiguration         AppConfigurationFeatures
-	ApplicationInsights      ApplicationInsightFeatures
-	CognitiveAccount         CognitiveAccountFeatures
-	VirtualMachine           VirtualMachineFeatures
-	VirtualMachineScaleSet   VirtualMachineScaleSetFeatures
-	KeyVault                 KeyVaultFeatures
-	TemplateDeployment       TemplateDeploymentFeatures
-	LogAnalyticsWorkspace    LogAnalyticsWorkspaceFeatures
-	ResourceGroup            ResourceGroupFeatures
-	RecoveryServicesVault    RecoveryServicesVault
-	ManagedDisk              ManagedDiskFeatures
-	Subscription             SubscriptionFeatures
-	PostgresqlFlexibleServer PostgresqlFlexibleServerFeatures
-	MachineLearning          MachineLearningFeatures
-	RecoveryService          RecoveryServiceFeatures
+	ApiManagement              ApiManagementFeatures
+	AppConfiguration           AppConfigurationFeatures
+	ApplicationInsights        ApplicationInsightFeatures
+	CognitiveAccount           CognitiveAccountFeatures
+	CognitiveAccountAIServices CognitiveAccountAIServicesFeatures
+	VirtualMachine             VirtualMachineFeatures
+	VirtualMachineScaleSet     VirtualMachineScaleSetFeatures
+	KeyVault                   KeyVaultFeatures
+	TemplateDeployment         TemplateDeploymentFeatures
+	LogAnalyticsWorkspace      LogAnalyticsWorkspaceFeatures
+	ResourceGroup              ResourceGroupFeatures
+	RecoveryServicesVault      RecoveryServicesVault
+	ManagedDisk                ManagedDiskFeatures
+	Subscription               SubscriptionFeatures
+	PostgresqlFlexibleServer   PostgresqlFlexibleServerFeatures
+	MachineLearning            MachineLearningFeatures
+	RecoveryService            RecoveryServiceFeatures
 }
 
 type CognitiveAccountFeatures struct {
 	PurgeSoftDeleteOnDestroy bool
 }
 
+type CognitiveAccountAIServicesFeatures struct {
+	PurgeSoftDeleteOnDestroy bool
+}
+
 type VirtualMachineFeatures struct {
 	DeleteOSDiskOnDeletion     bool
 	GracefulShutdown           bool

internal/provider/features.go

@@ -88,6 +88,21 @@ func schemaFeatures(supportLegacyTestSuite bool) *pluginsdk.Schema {
 			},
 		},
 
+		"cognitive_account_ai_services": {
+			Type:     pluginsdk.TypeList,
+			Optional: true,
+			MaxItems: 1,
+			Elem: &pluginsdk.Resource{
+				Schema: map[string]*pluginsdk.Schema{
+					"purge_soft_delete_on_destroy": {
+						Type:     pluginsdk.TypeBool,
+						Optional: true,
+						Default:  true,
+					},
+				},
+			},
+		},
+
 		"key_vault": {
 			Type:     pluginsdk.TypeList,
 			Optional: true,
@@ -448,6 +463,16 @@ func expandFeatures(input []interface{}) features.UserFeatures {
 		}
 	}
 
+	if raw, ok := val["cognitive_account_ai_services"]; ok {
+		items := raw.([]interface{})
+		if len(items) > 0 && items[0] != nil {
+			cognitiveAIServicesRaw := items[0].(map[string]interface{})
+			if v, ok := cognitiveAIServicesRaw["purge_soft_delete_on_destroy"]; ok {
+				featuresMap.CognitiveAccountAIServices.PurgeSoftDeleteOnDestroy = v.(bool)
+			}
+		}
+	}
+
 	if raw, ok := val["key_vault"]; ok {
 		items := raw.([]interface{})
 		if len(items) > 0 && items[0] != nil {

internal/provider/features_test.go

@@ -35,6 +35,9 @@ func TestExpandFeatures(t *testing.T) {
 				CognitiveAccount: features.CognitiveAccountFeatures{
 					PurgeSoftDeleteOnDestroy: true,
 				},
+				CognitiveAccountAIServices: features.CognitiveAccountAIServicesFeatures{
+					PurgeSoftDeleteOnDestroy: true,
+				},
 				KeyVault: features.KeyVaultFeatures{
 					PurgeSoftDeletedCertsOnDestroy:   true,
 					PurgeSoftDeletedKeysOnDestroy:    true,
@@ -115,6 +118,11 @@ func TestExpandFeatures(t *testing.T) {
 							"purge_soft_delete_on_destroy": true,
 						},
 					},
+					"cognitive_account_ai_services": []interface{}{
+						map[string]interface{}{
+							"purge_soft_delete_on_destroy": true,
+						},
+					},
 					"key_vault": []interface{}{
 						map[string]interface{}{
 							"purge_soft_deleted_certificates_on_destroy":                  true,
@@ -208,6 +216,9 @@ func TestExpandFeatures(t *testing.T) {
 				CognitiveAccount: features.CognitiveAccountFeatures{
 					PurgeSoftDeleteOnDestroy: true,
 				},
+				CognitiveAccountAIServices: features.CognitiveAccountAIServicesFeatures{
+					PurgeSoftDeleteOnDestroy: true,
+				},
 				KeyVault: features.KeyVaultFeatures{
 					PurgeSoftDeletedCertsOnDestroy:   true,
 					PurgeSoftDeletedKeysOnDestroy:    true,
@@ -288,6 +299,11 @@ func TestExpandFeatures(t *testing.T) {
 							"purge_soft_delete_on_destroy": false,
 						},
 					},
+					"cognitive_account_ai_services": []interface{}{
+						map[string]interface{}{
+							"purge_soft_delete_on_destroy": false,
+						},
+					},
 					"key_vault": []interface{}{
 						map[string]interface{}{
 							"purge_soft_deleted_certificates_on_destroy":                  false,
@@ -381,6 +397,9 @@ func TestExpandFeatures(t *testing.T) {
 				CognitiveAccount: features.CognitiveAccountFeatures{
 					PurgeSoftDeleteOnDestroy: false,
 				},
+				CognitiveAccountAIServices: features.CognitiveAccountAIServicesFeatures{
+					PurgeSoftDeleteOnDestroy: false,
+				},
 				KeyVault: features.KeyVaultFeatures{
 					PurgeSoftDeletedCertsOnDestroy:   false,
 					PurgeSoftDeletedKeysOnDestroy:    false,
@@ -716,6 +735,71 @@ func TestExpandFeaturesCognitiveServices(t *testing.T) {
 	}
 }
 
+func TestExpandFeaturesCognitiveAIServicesAccount(t *testing.T) {
+	testData := []struct {
+		Name     string
+		Input    []interface{}
+		EnvVars  map[string]interface{}
+		Expected features.UserFeatures
+	}{
+		{
+			Name: "Empty Block",
+			Input: []interface{}{
+				map[string]interface{}{
+					"cognitive_account_ai_services": []interface{}{},
+				},
+			},
+			Expected: features.UserFeatures{
+				CognitiveAccountAIServices: features.CognitiveAccountAIServicesFeatures{
+					PurgeSoftDeleteOnDestroy: true,
+				},
+			},
+		},
+		{
+			Name: "Purge on Destroy Enabled",
+			Input: []interface{}{
+				map[string]interface{}{
+					"cognitive_account_ai_services": []interface{}{
+						map[string]interface{}{
+							"purge_soft_delete_on_destroy": true,
+						},
+					},
+				},
+			},
+			Expected: features.UserFeatures{
+				CognitiveAccountAIServices: features.CognitiveAccountAIServicesFeatures{
+					PurgeSoftDeleteOnDestroy: true,
+				},
+			},
+		},
+		{
+			Name: "Purge on Destroy Disabled",
+			Input: []interface{}{
+				map[string]interface{}{
+					"cognitive_account_ai_services": []interface{}{
+						map[string]interface{}{
+							"purge_soft_delete_on_destroy": false,
+						},
+					},
+				},
+			},
+			Expected: features.UserFeatures{
+				CognitiveAccountAIServices: features.CognitiveAccountAIServicesFeatures{
+					PurgeSoftDeleteOnDestroy: false,
+				},
+			},
+		},
+	}
+
+	for _, testCase := range testData {
+		t.Logf("[DEBUG] Test Case: %q", testCase.Name)
+		result := expandFeatures(testCase.Input)
+		if !reflect.DeepEqual(result.CognitiveAccountAIServices, testCase.Expected.CognitiveAccountAIServices) {
+			t.Fatalf("Expected %+v but got %+v", result.CognitiveAccountAIServices, testCase.Expected.CognitiveAccountAIServices)
+		}
+	}
+}
+
 func TestExpandFeaturesKeyVault(t *testing.T) {
 	testData := []struct {
 		Name     string