azurerm_dashboard_grafana: add support for sku_size (hashicorp#31047)

[ENHANCEMENT] * `azurerm_dashboard_grafana`- add support for the `sku_size` property

Author: jiaweitao001

internal/services/dashboard/dashboard_grafana_resource.go

@@ -35,6 +35,7 @@ type DashboardGrafanaModel struct {
 	Location                          string                                            `tfschema:"location"`
 	PublicNetworkAccessEnabled        bool                                              `tfschema:"public_network_access_enabled"`
 	Sku                               string                                            `tfschema:"sku"`
+	SkuSize                           string                                            `tfschema:"sku_size"`
 	Tags                              map[string]string                                 `tfschema:"tags"`
 	ZoneRedundancyEnabled             bool                                              `tfschema:"zone_redundancy_enabled"`
 	Endpoint                          string                                            `tfschema:"endpoint"`
@@ -214,6 +215,14 @@ func (r DashboardGrafanaResource) Arguments() map[string]*pluginsdk.Schema {
 			}, false),
 		},
 
+		"sku_size": {
+			Type:         pluginsdk.TypeString,
+			Optional:     true,
+			Default:      managedgrafanas.SizeXOne,
+			ForceNew:     true,
+			ValidateFunc: validation.StringInSlice(managedgrafanas.PossibleValuesForSize(), false),
+		},
+
 		"tags": commonschema.Tags(),
 
 		"zone_redundancy_enabled": {
@@ -311,6 +320,10 @@ func (r DashboardGrafanaResource) Create() sdk.ResourceFunc {
 				Tags: &model.Tags,
 			}
 
+			if model.SkuSize != "" {
+				properties.Sku.Size = pointer.To(managedgrafanas.Size(model.SkuSize))
+			}
+
 			if err := client.GrafanaCreateThenPoll(ctx, id, *properties); err != nil {
 				return fmt.Errorf("creating %s: %+v", id, err)
 			}
@@ -507,6 +520,9 @@ func (r DashboardGrafanaResource) Read() sdk.ResourceFunc {
 
 			if model.Sku != nil {
 				state.Sku = model.Sku.Name
+				if model.Sku.Size != nil {
+					state.SkuSize = pointer.FromEnum(model.Sku.Size)
+				}
 			}
 
 			if model.Tags != nil {

internal/services/dashboard/dashboard_grafana_resource_test.go

@@ -96,6 +96,20 @@ func TestAccDashboardGrafana_withSku(t *testing.T) {
 	})
 }
 
+func TestAccDashboardGrafana_withSize(t *testing.T) {
+	data := acceptance.BuildTestData(t, "azurerm_dashboard_grafana", "test")
+	r := DashboardGrafanaResource{}
+	data.ResourceSequentialTest(t, r, []acceptance.TestStep{
+		{
+			Config: r.withSize(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).ExistsInAzure(r),
+			),
+		},
+		data.ImportStep(),
+	})
+}
+
 func (r DashboardGrafanaResource) Exists(ctx context.Context, clients *clients.Client, state *pluginsdk.InstanceState) (*bool, error) {
 	id, err := managedgrafanas.ParseGrafanaID(state.ID)
 	if err != nil {
@@ -162,6 +176,23 @@ resource "azurerm_dashboard_grafana" "test" {
 `, template, data.RandomInteger)
 }
 
+func (r DashboardGrafanaResource) withSize(data acceptance.TestData) string {
+	template := r.template(data)
+	return fmt.Sprintf(`
+				%s
+
+resource "azurerm_dashboard_grafana" "test" {
+  name                  = "a-dg-%d"
+  resource_group_name   = azurerm_resource_group.test.name
+  location              = azurerm_resource_group.test.location
+  grafana_major_version = "11"
+
+  sku      = "Standard"
+  sku_size = "X1"
+}
+`, template, data.RandomInteger)
+}
+
 func (r DashboardGrafanaResource) requiresImport(data acceptance.TestData) string {
 	config := r.basic(data)
 	return fmt.Sprintf(`

website/docs/r/dashboard_grafana.html.markdown

@@ -26,6 +26,8 @@ resource "azurerm_dashboard_grafana" "example" {
   api_key_enabled                   = true
   deterministic_outbound_ip_enabled = true
   public_network_access_enabled     = false
+  sku                               = "Standard"
+  sku_size                          = "X1"
 
   identity {
     type = "SystemAssigned"
@@ -65,6 +67,8 @@ The following arguments are supported:
 
 * `sku` - (Optional) The name of the SKU used for the Grafana instance. Possible values are `Standard` and `Essential`. Defaults to `Standard`. Changing this forces a new Dashboard Grafana to be created.
 
+* `sku_size` - (Optional) The size of the SKU used for the Grafana instance. Possible values are `X1` and `X2`. Defaults to `X1`. Changing this forces a new Dashboard Grafana to be created.
+
 * `tags` - (Optional) A mapping of tags which should be assigned to the Dashboard Grafana.
 
 * `zone_redundancy_enabled` - (Optional) Whether to enable the zone redundancy setting of the Grafana instance. Defaults to `false`. Changing this forces a new Dashboard Grafana to be created.