[go sdk] [budgets] Fix billing test for budget configuration update

databricks · Aug 15, 2024 · 17fa38b · 17fa38b
1 parent 27a5055
commit 17fa38b
Showing 1 changed file with 8 additions and 11 deletions.
diff --git a/internal/billing_test.go b/internal/billing_test.go
@@ -123,7 +123,8 @@ func TestMwsAccBudgets(t *testing.T) {
 
 		BudgetId: created.Budget.BudgetConfigurationId,
 		Budget: billing.UpdateBudgetConfigurationBudget{
-			DisplayName: RandomName("go-sdk-updated-"),
+			BudgetConfigurationId: created.Budget.BudgetConfigurationId,
+			DisplayName:           RandomName("go-sdk-updated-"),
 			Filter: &billing.BudgetConfigurationFilter{
 				Tags: []billing.BudgetConfigurationFilterTagClause{
 					{
@@ -136,16 +137,12 @@ func TestMwsAccBudgets(t *testing.T) {
 				}},
 			AlertConfigurations: []billing.AlertConfiguration{
 				{
-					TimePeriod:        billing.AlertConfigurationTimePeriodMonth,
-					QuantityType:      billing.AlertConfigurationQuantityTypeListPriceDollarsUsd,
-					TriggerType:       billing.AlertConfigurationTriggerTypeCumulativeSpendingExceeded,
-					QuantityThreshold: "50",
-					ActionConfigurations: []billing.ActionConfiguration{
-						{
-							ActionType: billing.ActionConfigurationTypeEmailNotification,
-							Target:     "admin@example.com",
-						},
-					},
+					AlertConfigurationId: created.Budget.AlertConfigurations[0].AlertConfigurationId,
+					TimePeriod:           billing.AlertConfigurationTimePeriodMonth,
+					QuantityType:         billing.AlertConfigurationQuantityTypeListPriceDollarsUsd,
+					TriggerType:          billing.AlertConfigurationTriggerTypeCumulativeSpendingExceeded,
+					QuantityThreshold:    "50",
+					ActionConfigurations: created.Budget.AlertConfigurations[0].ActionConfigurations,
 				},
 			},
 		},