Extend wait_for with optional condition blocks

alekc · Aug 21, 2024 · 8cbcd6e · 8cbcd6e
1 parent abed1de
commit 8cbcd6e
Show file tree

Hide file tree

Showing 3 changed files with 147 additions and 6 deletions.
diff --git a/internal/types/wait_for.go b/internal/types/wait_for.go
@@ -2,9 +2,15 @@ package types
 
 type WaitFor struct {
 	Field []WaitForField
+	Condition []WaitForStatusCondition
 }
 type WaitForField struct {
 	Key       string
 	Value     string
 	ValueType string `mapstructure:"value_type"`
 }
+
+type WaitForStatusCondition struct {
+	Type   string
+	Status string
+}
diff --git a/kubernetes/resource_kubectl_manifest.go b/kubernetes/resource_kubectl_manifest.go
@@ -460,11 +460,31 @@ var (
 			MaxItems:    1,
 			Elem: &schema.Resource{
 				Schema: map[string]*schema.Schema{
+					"condition": {
+						Type:        schema.TypeList,
+						MinItems:    0,
+						Description: "Condition criteria for a Status Condition",
+						Optional:    true,
+						Elem: &schema.Resource{
+							Schema: map[string]*schema.Schema{
+								"type": {
+									Type:        schema.TypeString,
+									Description: "Type as expected from the resulting Condition object",
+									Required:    true,
+								},
+								"status": {
+									Type:        schema.TypeString,
+									Description: "Status to wait for in the resulting Condition object",
+									Required:    true,
+								},
+							},
+						},
+					},
 					"field": {
 						Type:        schema.TypeList,
-						MinItems:    1,
+						MinItems:    0,
 						Description: "Condition criteria for a field",
-						Required:    true,
+						Optional:    true,
 						Elem: &schema.Resource{
 							Schema: map[string]*schema.Schema{
 								"key": {
@@ -635,7 +655,7 @@ func resourceKubectlManifestApply(ctx context.Context, d *schema.ResourceData, m
 		}
 		log.Printf("[INFO] %v waiting for wait conditions for %vmin", manifest, timeout.Minutes())
 		err = resource.RetryContext(ctx, timeout,
-			waitForFields(ctx, restClient, waitFor.Field, manifest.GetNamespace(), manifest.GetName()))
+			waitForConditions(ctx, restClient, waitFor.Field, waitFor.Condition, manifest.GetNamespace(), manifest.GetName()))
 		if err != nil {
 			return err
 		}
@@ -940,7 +960,7 @@ func GetDeploymentCondition(status apps_v1.DeploymentStatus, condType apps_v1.De
 	return nil
 }
 
-func waitForFields(ctx context.Context, provider *RestClientResult, conditions []types.WaitForField, ns, name string) resource.RetryFunc {
+func waitForConditions(ctx context.Context, provider *RestClientResult, fields []types.WaitForField, conditions []types.WaitForStatusCondition, ns, name string) resource.RetryFunc {
 	return func() *resource.RetryError {
 		rawResponse, err := provider.ResourceInterface.Get(ctx, name, meta_v1.GetOptions{})
 		if err != nil {
@@ -953,7 +973,7 @@ func waitForFields(ctx context.Context, provider *RestClientResult, conditions [
 			return resource.NonRetryableError(err)
 		}
 		gq := gojsonq.New().FromString(string(yamlJson))
-		for _, c := range conditions {
+		for _, c := range fields {
 			//find the key
 			v := gq.Reset().Find(c.Key)
 			if v == nil {
@@ -978,6 +998,14 @@ func waitForFields(ctx context.Context, provider *RestClientResult, conditions [
 				}
 			}
 		}
+
+		for _, c := range conditions {
+			//find the conditions by status and type
+			v := gq.Reset().From("status.conditions").Where("type", "=", c.Type).Where("status", "=", c.Status)
+			if v == nil {
+				return resource.RetryableError(fmt.Errorf("key %s was not found in the resource %s", c.Status, name))
+			}
+		}
 		return nil
 	}
 }

diff --git a/kubernetes/resource_kubectl_manifest_test.go b/kubernetes/resource_kubectl_manifest_test.go
@@ -224,7 +224,7 @@ YAML
 	})
 }
 
-func TestAccKubectl_WaitFor(t *testing.T) {
+func TestAccKubectl_WaitForField(t *testing.T) {
 	//language=hcl
 	config := `
 resource "kubectl_manifest" "test" {
@@ -275,6 +275,113 @@ YAML
 	})
 }
 
+
+func TestAccKubectl_WaitForConditions(t *testing.T) {
+	//language=hcl
+	config := `
+resource "kubectl_manifest" "test" {
+	wait_for {
+		condition {
+			type = "ContainersReady"
+			status = "True"
+		}
+		condition {
+			type = "Ready"
+			status = "True"
+		}
+	}
+	yaml_body = <<YAML
+apiVersion: v1
+kind: Pod
+metadata:
+  name: nginx
+spec:
+  containers:
+  - name: nginx
+    image: nginx:1.14.2
+    readinessProbe:
+      httpGet:
+        path: "/"
+        port: 80			
+      initialDelaySeconds: 10
+YAML
+}
+`
+
+	//start := time.Now()
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckkubectlDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				//todo: improve checking
+			},
+		},
+	})
+}
+
+func TestAccKubectl_WaitForFieldAndCondition(t *testing.T) {
+	//language=hcl
+	config := `
+resource "kubectl_manifest" "test" {
+	wait_for {
+		condition {
+			type = "ContainersReady"
+			status = "True"
+		}
+		condition {
+			type = "Ready"
+			status = "True"
+		}
+		field {
+			key = "status.containerStatuses.[0].ready"
+			value = "true"
+		}
+		field {
+			key = "status.phase"
+			value = "Running"
+		}
+		field {
+			key = "status.podIP"
+			value = "^(\\d+(\\.|$)){4}"
+			value_type = "regex"
+		}
+	}
+	yaml_body = <<YAML
+apiVersion: v1
+kind: Pod
+metadata:
+  name: nginx
+spec:
+  containers:
+  - name: nginx
+    image: nginx:1.14.2
+    readinessProbe:
+      httpGet:
+        path: "/"
+        port: 80			
+      initialDelaySeconds: 10
+YAML
+}
+`
+
+	//start := time.Now()
+	resource.Test(t, resource.TestCase{
+		PreCheck:     func() { testAccPreCheck(t) },
+		Providers:    testAccProviders,
+		CheckDestroy: testAccCheckkubectlDestroy,
+		Steps: []resource.TestStep{
+			{
+				Config: config,
+				//todo: improve checking
+			},
+		},
+	})
+}
+
+
 //func TestAccKubect_Debug(t *testing.T) {
 //	//language=hcl
 //	config := `