From 3978de9e757397b25f044aa98601f78db3119d6c Mon Sep 17 00:00:00 2001
From: Emerson Hardisty <emerson@hardisty.uk>
Date: Mon, 23 Sep 2024 09:19:38 +0100
Subject: [PATCH] Iterate through list of supplied kap controls (#273)

---
 aquasec/resource_image_assurance_policy.go | 27 ++++++++++++----------
 1 file changed, 15 insertions(+), 12 deletions(-)

diff --git a/aquasec/resource_image_assurance_policy.go b/aquasec/resource_image_assurance_policy.go
index 58f1fcc..4679253 100644
--- a/aquasec/resource_image_assurance_policy.go
+++ b/aquasec/resource_image_assurance_policy.go
@@ -2,9 +2,10 @@ package aquasec
 
 import (
 	"fmt"
+	"strings"
+
 	"github.com/aquasecurity/terraform-provider-aquasec/client"
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
-	"strings"
 )
 
 func resourceImageAssurancePolicy() *schema.Resource {
@@ -1752,17 +1753,19 @@ func expandAssurancePolicy(d *schema.ResourceData, a_type string) *client.Assura
 	if ok {
 		controlsList := kubernetesControlsList.([]interface{})
 		if len(controlsList) > 0 {
-			v := controlsList[0].(map[string]interface{})
-			iap.KubernetesControls = append(iap.KubernetesControls, client.KubernetesControls{
-				ScriptID:    int(v["script_id"].(int)),
-				Name:        v["name"].(string),
-				Description: v["description"].(string),
-				Enabled:     v["enabled"].(bool),
-				Severity:    v["severity"].(string),
-				Kind:        v["kind"].(string),
-				OOTB:        v["ootb"].(bool),
-				AvdID:       v["avd_id"].(string),
-			})
+			for _, control := range controlsList {
+				v := control.(map[string]interface{})
+				iap.KubernetesControls = append(iap.KubernetesControls, client.KubernetesControls{
+					ScriptID:    int(v["script_id"].(int)),
+					Name:        v["name"].(string),
+					Description: v["description"].(string),
+					Enabled:     v["enabled"].(bool),
+					Severity:    v["severity"].(string),
+					Kind:        v["kind"].(string),
+					OOTB:        v["ootb"].(bool),
+					AvdID:       v["avd_id"].(string),
+				})
+			}
 		}
 	}