New data source: azurerm_consumption_budget_subscription

internal/services/consumption/consumption_budget_subscription_data_source.go

@@ -0,0 +1,254 @@
+package consumption
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/hashicorp/terraform-provider-azurerm/internal/clients"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/services/consumption/parse"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/services/consumption/validate"
+	subscriptionParse "github.com/hashicorp/terraform-provider-azurerm/internal/services/subscription/parse"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/tf/pluginsdk"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/timeouts"
+	"github.com/hashicorp/terraform-provider-azurerm/utils"
+)
+
+func resourceArmConsumptionBudgetSubscriptionDataSource() *pluginsdk.Resource {
+	return &pluginsdk.Resource{
+		Read: resourceArmConsumptionBudgetSubscriptionDataSourceRead,
+		Timeouts: &pluginsdk.ResourceTimeout{
+			Read: pluginsdk.DefaultTimeout(5 * time.Minute),
+		},
+		Schema: map[string]*pluginsdk.Schema{
+			"name": {
+				Type:         pluginsdk.TypeString,
+				Required:     true,
+				ValidateFunc: validate.ConsumptionBudgetName(),
+			},
+
+			"subscription_id": {
+				Type:     pluginsdk.TypeString,
+				Required: true,
+			},
+
+			"amount": {
+				Type:     pluginsdk.TypeFloat,
+				Computed: true,
+			},
+
+			"filter": {
+				Type:     pluginsdk.TypeList,
+				Computed: true,
+				Elem: &pluginsdk.Resource{
+					Schema: map[string]*pluginsdk.Schema{
+						"dimension": {
+							Type:     pluginsdk.TypeList,
+							Computed: true,
+							Elem: &pluginsdk.Resource{
+								Schema: map[string]*pluginsdk.Schema{
+									"name": {
+										Type:     pluginsdk.TypeString,
+										Computed: true,
+									},
+									"operator": {
+										Type:     pluginsdk.TypeString,
+										Computed: true,
+									},
+									"values": {
+										Type:     pluginsdk.TypeList,
+										Computed: true,
+										Elem: &pluginsdk.Schema{
+											Type: pluginsdk.TypeString,
+										},
+									},
+								},
+							},
+						},
+						"tag": {
+							Type:     pluginsdk.TypeList,
+							Computed: true,
+							Elem: &pluginsdk.Resource{
+								Schema: map[string]*pluginsdk.Schema{
+									"name": {
+										Type:     pluginsdk.TypeString,
+										Computed: true,
+									},
+									"operator": {
+										Type:     pluginsdk.TypeString,
+										Computed: true,
+									},
+									"values": {
+										Type:     pluginsdk.TypeList,
+										Computed: true,
+										Elem: &pluginsdk.Schema{
+											Type: pluginsdk.TypeString,
+										},
+									},
+								},
+							},
+						},
+						"not": {
+							Type:     pluginsdk.TypeList,
+							Computed: true,
+							Elem: &pluginsdk.Resource{
+								Schema: map[string]*pluginsdk.Schema{
+									"dimension": {
+										Type:     pluginsdk.TypeList,
+										Computed: true,
+										Elem: &pluginsdk.Resource{
+											Schema: map[string]*pluginsdk.Schema{
+												"name": {
+													Type:     pluginsdk.TypeString,
+													Computed: true,
+												},
+												"operator": {
+													Type:     pluginsdk.TypeString,
+													Computed: true,
+												},
+												"values": {
+													Type:     pluginsdk.TypeList,
+													Computed: true,
+													Elem: &pluginsdk.Schema{
+														Type: pluginsdk.TypeString,
+													},
+												},
+											},
+										},
+									},
+									"tag": {
+										Type:     pluginsdk.TypeList,
+										Computed: true,
+										Elem: &pluginsdk.Resource{
+											Schema: map[string]*pluginsdk.Schema{
+												"name": {
+													Type:     pluginsdk.TypeString,
+													Computed: true,
+												},
+												"operator": {
+													Type:     pluginsdk.TypeString,
+													Computed: true,
+												},
+												"values": {
+													Type:     pluginsdk.TypeList,
+													Computed: true,
+													Elem: &pluginsdk.Schema{
+														Type: pluginsdk.TypeString,
+													},
+												},
+											},
+										},
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+
+			"notification": {
+				Type:     pluginsdk.TypeList,
+				Computed: true,
+				Elem: &pluginsdk.Resource{
+					Schema: map[string]*pluginsdk.Schema{
+						"enabled": {
+							Type:     pluginsdk.TypeBool,
+							Computed: true,
+						},
+						"threshold": {
+							Type:     pluginsdk.TypeInt,
+							Computed: true,
+						},
+						"operator": {
+							Type:     pluginsdk.TypeString,
+							Computed: true,
+						},
+
+						"contact_emails": {
+							Type:     pluginsdk.TypeList,
+							Computed: true,
+							Elem: &pluginsdk.Schema{
+								Type: pluginsdk.TypeString,
+							},
+						},
+
+						"contact_groups": {
+							Type:     pluginsdk.TypeList,
+							Computed: true,
+							Elem: &pluginsdk.Schema{
+								Type: pluginsdk.TypeString,
+							},
+						},
+
+						"contact_roles": {
+							Type:     pluginsdk.TypeList,
+							Computed: true,
+							Elem: &pluginsdk.Schema{
+								Type: pluginsdk.TypeString,
+							},
+						},
+					},
+				},
+			},
+
+			"time_grain": {
+				Type:     pluginsdk.TypeString,
+				Computed: true,
+			},
+
+			"time_period": {
+				Type:     pluginsdk.TypeList,
+				Computed: true,
+				Elem: &pluginsdk.Resource{
+					Schema: map[string]*pluginsdk.Schema{
+						"start_date": {
+							Type:     pluginsdk.TypeString,
+							Computed: true,
+						},
+						"end_date": {
+							Type:     pluginsdk.TypeString,
+							Computed: true,
+						},
+					},
+				},
+			},
+		},
+	}
+}
+
+func resourceArmConsumptionBudgetSubscriptionDataSourceRead(d *pluginsdk.ResourceData, meta interface{}) error {
+	client := meta.(*clients.Client).Consumption.BudgetsClient
+	ctx, cancel := timeouts.ForCreateUpdate(meta.(*clients.Client).StopContext, d)
+	defer cancel()
+
+	subscriptionId, err := subscriptionParse.SubscriptionID(d.Get("subscription_id").(string))
+	if err != nil {
+		return err
+	}
+
+	id := parse.NewConsumptionBudgetSubscriptionID(subscriptionId.SubscriptionID, d.Get("name").(string))
+	d.SetId(id.ID())
+
+	resp, err := client.Get(ctx, subscriptionId.ID(), id.BudgetName)
+	if err != nil {
+		if utils.ResponseWasNotFound(resp.Response) {
+			d.SetId("")
+			return nil
+		}
+		return fmt.Errorf("making read request on %s: %+v", id, err)
+	}
+
+	d.Set("name", resp.Name)
+	if resp.Amount != nil {
+		amount, _ := resp.Amount.Float64()
+		d.Set("amount", amount)
+	}
+	d.Set("time_grain", string(resp.TimeGrain))
+	d.Set("time_period", FlattenConsumptionBudgetTimePeriod(resp.TimePeriod))
+	d.Set("notification", FlattenConsumptionBudgetNotifications(resp.Notifications))
+	d.Set("filter", FlattenConsumptionBudgetFilter(resp.Filter))
+
+	// The scope of a Subscription budget resource is the Subscription budget ID
+	d.Set("subscription_id", d.Get("subscription_id").(string))
+
+	return nil
+}

internal/services/consumption/consumption_budget_subscription_data_source_test.go

@@ -0,0 +1,57 @@
+package consumption_test
+
+import (
+	"fmt"
+	"testing"
+
+	"github.com/hashicorp/terraform-provider-azurerm/internal/acceptance"
+	"github.com/hashicorp/terraform-provider-azurerm/internal/acceptance/check"
+)
+
+type ConsumptionBudgetSubscriptionDataSource struct{}
+
+func TestAccDataSourceConsumptionBudgetSubscription_basic(t *testing.T) {
+	data := acceptance.BuildTestData(t, "data.azurerm_consumption_budget_subscription", "test")
+	r := ConsumptionBudgetSubscriptionDataSource{}
+
+	data.DataSourceTest(t, []acceptance.TestStep{
+		{
+			Config: r.basic(data),
+			Check: acceptance.ComposeTestCheckFunc(
+				check.That(data.ResourceName).Key("subscription_id").HasValue(data.Client().SubscriptionID),
+				check.That(data.ResourceName).Key("name").Exists(),
+				check.That(data.ResourceName).Key("amount").HasValue("1000"),
+				check.That(data.ResourceName).Key("time_grain").HasValue("Monthly"),
+				check.That(data.ResourceName).Key("time_period.#").Exists(),
+				check.That(data.ResourceName).Key("time_period.0.start_date").Exists(),
+				check.That(data.ResourceName).Key("time_period.0.end_date").Exists(),
+				check.That(data.ResourceName).Key("filter.#").Exists(),
+				check.That(data.ResourceName).Key("filter.0.tag.0.name").HasValue("foo"),
+				check.That(data.ResourceName).Key("filter.0.tag.0.values.#").Exists(),
+				check.That(data.ResourceName).Key("filter.0.dimension.0.name").HasValue("ResourceGroupName"),
+				check.That(data.ResourceName).Key("filter.0.dimension.1.name").HasValue("ResourceId"),
+				check.That(data.ResourceName).Key("filter.0.not.0.tag.0.name").HasValue("zip"),
+				check.That(data.ResourceName).Key("notification.#").Exists(),
+				check.That(data.ResourceName).Key("notification.0.threshold").HasValue("90"),
+				check.That(data.ResourceName).Key("notification.0.operator").HasValue("EqualTo"),
+				check.That(data.ResourceName).Key("notification.0.enabled").Exists(),
+				check.That(data.ResourceName).Key("notification.0.contact_emails.0").HasValue("[email protected]"),
+				check.That(data.ResourceName).Key("notification.0.contact_emails.1").HasValue("[email protected]"),
+				check.That(data.ResourceName).Key("notification.0.contact_groups.#").Exists(),
+				check.That(data.ResourceName).Key("notification.0.contact_roles.0").HasValue("Owner"),
+			),
+		},
+	})
+}
+
+func (d ConsumptionBudgetSubscriptionDataSource) basic(data acceptance.TestData) string {
+	config := ConsumptionBudgetSubscriptionResource{}.complete(data)
+	return fmt.Sprintf(`
+  %s
+
+data "azurerm_consumption_budget_subscription" "test" {
+  name            = azurerm_consumption_budget_subscription.test.name
+  subscription_id = azurerm_consumption_budget_subscription.test.subscription_id
+}
+`, config)
+}

internal/services/consumption/registration.go

@@ -9,8 +9,9 @@ const (
 	// as the core logic for the Consumption Budget resources is generic and has been
 	// extracted out of the specific Consumption Budget resources. These constants are
 	// used when the generic Consumption Budget functions require a resource name.
-	consumptionBudgetResourceGroupName = "azurerm_consumption_budget_resource_group"
-	consumptionBudgetSubscriptionName  = "azurerm_consumption_budget_subscription"
+	consumptionBudgetResourceGroupName          = "azurerm_consumption_budget_resource_group"
+	consumptionBudgetSubscriptionName           = "azurerm_consumption_budget_subscription"
+	consumptionBudgetSubscriptionDataSourceName = "azurerm_consumption_budget_subscription"
 )
 
 type Registration struct{}
@@ -29,7 +30,9 @@ func (r Registration) WebsiteCategories() []string {
 
 // SupportedDataSources returns the supported Data Sources supported by this Service
 func (r Registration) SupportedDataSources() map[string]*pluginsdk.Resource {
-	return map[string]*pluginsdk.Resource{}
+	return map[string]*pluginsdk.Resource{
+		consumptionBudgetSubscriptionDataSourceName: resourceArmConsumptionBudgetSubscriptionDataSource(),
+	}
 }
 
 // SupportedResources returns the supported Resources supported by this Service