BE-468: Add test

Author: chirag-sumologic

sumologic/resource_sumologic_scan_budget.go

@@ -1,3 +1,14 @@
+// ----------------------------------------------------------------------------
+//
+//	***     AUTO GENERATED CODE    ***    AUTO GENERATED CODE     ***
+//
+// ----------------------------------------------------------------------------
+//
+//	This file is automatically generated by Sumo Logic and manual
+//	changes will be clobbered when the file is regenerated. Do not submit
+//	changes to this file.
+//
+// ----------------------------------------------------------------------------
 package sumologic
 
 import (
@@ -24,12 +35,6 @@ func resourceSumologicScanBudget() *schema.Resource {
 				ForceNew: false,
 			},
 
-			"orgId": {
-				Type:     schema.TypeString,
-				Required: true,
-				ForceNew: false,
-			},
-
 			"capacity": {
 				Type:     schema.TypeInt,
 				Required: true,
@@ -42,7 +47,7 @@ func resourceSumologicScanBudget() *schema.Resource {
 				ForceNew: false,
 			},
 
-			"budgetType": {
+			"budget_type": {
 				Type:     schema.TypeString,
 				Required: true,
 				ForceNew: false,
@@ -54,24 +59,54 @@ func resourceSumologicScanBudget() *schema.Resource {
 				ForceNew: false,
 			},
 
-			"groupBy": {
+			"group_by": {
 				Type:     schema.TypeString,
 				Required: true,
 				ForceNew: false,
 			},
 
-			"applicableOn": {
+			"applicable_on": {
 				Type:     schema.TypeString,
 				Required: true,
 				ForceNew: false,
 			},
 
 			"scope": {
-				Type:     schema.TypeMap,
+				Type:     schema.TypeList,
 				Required: true,
 				ForceNew: false,
-				Elem: &schema.Schema{
-					Type: schema.TypeList,
+				MaxItems: 1,
+				Elem: &schema.Resource{
+					Schema: map[string]*schema.Schema{
+						"included_users": {
+							Type:     schema.TypeList,
+							Optional: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+						"excluded_users": {
+							Type:     schema.TypeList,
+							Optional: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+						"included_roles": {
+							Type:     schema.TypeList,
+							Optional: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+						"excluded_roles": {
+							Type:     schema.TypeList,
+							Optional: true,
+							Elem: &schema.Schema{
+								Type: schema.TypeString,
+							},
+						},
+					},
 				},
 			},
 
@@ -110,28 +145,27 @@ func resourceSumologicScanBudgetRead(d *schema.ResourceData, meta interface{}) e
 	c := meta.(*Client)
 
 	id := d.Id()
-	ScanBudget, err := c.GetScanBudget(id)
+	scanBudget, err := c.GetScanBudget(id)
 	if err != nil {
 		return err
 	}
 
-	if ScanBudget == nil {
+	if scanBudget == nil {
 		log.Printf("[WARN] ScanBudget not found, removing from state: %v - %v", id, err)
 		d.SetId("")
 		return nil
 	}
 
-	d.Set("name", ScanBudget.Name)
-	d.Set("orgId", ScanBudget.OrgID)
-	d.Set("capacity", ScanBudget.Capacity)
-	d.Set("unit", ScanBudget.Unit)
-	d.Set("budgetType", ScanBudget.BudgetType)
-	d.Set("window", ScanBudget.Window)
-	d.Set("applicableOn", ScanBudget.Grouping)
-	d.Set("groupBy", ScanBudget.GroupingEntity)
-	d.Set("action", ScanBudget.Action)
-	d.Set("scope", ScanBudget.Scope)
-	d.Set("status", ScanBudget.Status)
+	d.Set("name", scanBudget.Name)
+	d.Set("capacity", scanBudget.Capacity)
+	d.Set("unit", scanBudget.Unit)
+	d.Set("budget_type", scanBudget.BudgetType)
+	d.Set("window", scanBudget.Window)
+	d.Set("applicable_on", scanBudget.ApplicableOn)
+	d.Set("group_by", scanBudget.GroupBy)
+	d.Set("action", scanBudget.Action)
+	d.Set("scope", scanBudget.Scope)
+	d.Set("status", scanBudget.Status)
 
 	return nil
 }
@@ -145,8 +179,8 @@ func resourceSumologicScanBudgetDelete(d *schema.ResourceData, meta interface{})
 func resourceSumologicScanBudgetUpdate(d *schema.ResourceData, meta interface{}) error {
 	c := meta.(*Client)
 
-	ScanBudget := resourceToScanBudget(d)
-	err := c.UpdateScanBudget(ScanBudget)
+	scanBudget := resourceToScanBudget(d)
+	err := c.UpdateScanBudget(scanBudget)
 	if err != nil {
 		return err
 	}
@@ -156,17 +190,53 @@ func resourceSumologicScanBudgetUpdate(d *schema.ResourceData, meta interface{})
 
 func resourceToScanBudget(d *schema.ResourceData) ScanBudget {
 	return ScanBudget{
-		ID:             d.Id(),
-		OrgID:          d.Get("orgId").(string),
-		Name:           d.Get("name").(string),
-		Capacity:       d.Get("capacity").(int),
-		Unit:           d.Get("unit").(string),
-		BudgetType:     d.Get("budgetType").(string),
-		Window:         d.Get("window").(string),
-		Grouping:       d.Get("applicableOn").(string),
-		GroupingEntity: d.Get("groupBy").(string),
-		Action:         d.Get("action").(string),
-		Scope:          d.Get("scope").(map[string]interface{}),
-		Status:         d.Get("status").(string),
+		ID:           d.Id(),
+		Name:         d.Get("name").(string),
+		Capacity:     d.Get("capacity").(int),
+		Unit:         d.Get("unit").(string),
+		BudgetType:   d.Get("budget_type").(string),
+		Window:       d.Get("window").(string),
+		ApplicableOn: d.Get("applicable_on").(string),
+		GroupBy:      d.Get("group_by").(string),
+		Action:       d.Get("action").(string),
+		Scope:        resourceToScanBudgetScope(d.Get("scope")),
+		Status:       d.Get("status").(string),
+	}
+}
+
+func resourceToScanBudgetScope(data interface{}) ScanBudgetScope {
+	scopeMap, ok := data.([]interface{})[0].(map[string]interface{})
+	if !ok {
+		return ScanBudgetScope{}
+	}
+
+	scanBudgetScope := ScanBudgetScope{
+		IncludedUsers: convertToStringSlice(scopeMap["included_users"]),
+		ExcludedUsers: convertToStringSlice(scopeMap["excluded_users"]),
+		IncludedRoles: convertToStringSlice(scopeMap["included_roles"]),
+		ExcludedRoles: convertToStringSlice(scopeMap["excluded_roles"]),
+	}
+
+	return scanBudgetScope
+}
+
+func convertToStringSlice(data interface{}) []string {
+	if data == nil {
+		return nil
 	}
+
+	interfaceSlice, ok := data.([]interface{})
+	if !ok {
+		return nil
+	}
+
+	stringSlice := make([]string, len(interfaceSlice))
+	for i, v := range interfaceSlice {
+		str, ok := v.(string)
+		if ok {
+			stringSlice[i] = str
+		}
+	}
+
+	return stringSlice
 }