util: adding test cases

Signed-off-by: Annaraya Narasagond <[email protected]>
RamenDR · Dec 16, 2024 · 717f791 · 717f791
1 parent 240c77e
commit 717f791
Showing 1 changed file with 50 additions and 2 deletions.
diff --git a/internal/controller/util/json_util_test.go b/internal/controller/util/json_util_test.go
@@ -14,9 +14,10 @@ import (
 type testCases struct {
 	jsonPathExprs string
 	result        bool
+	jsonText      []byte
 }
 
-var jsonText1 = []byte(`{
+var jsonDeployment = []byte(`{
     "kind": "Deployment",
     "spec": {
         "progressDeadlineSeconds": 600,
@@ -38,18 +39,65 @@ var jsonText1 = []byte(`{
     }
     }`)
 
+var jsonPod = []byte(`{
+		"kind": "Pod",
+		"spec": {
+			"progressDeadlineSeconds": 600,
+			"replicas": 1,
+			"revisionHistoryLimit": 10
+		},
+		"status": {
+			"replicas": 1,
+			"conditions": [
+				{
+					"status": "True",
+					"type": "Progressing"
+				},
+				{
+					"status": "True",
+					"type": "Available"
+				}
+			]
+		}
+		}`)
+
+var jsonStatefulset = []byte(`{
+			"kind": "Statefulset",
+			"spec": {
+				"progressDeadlineSeconds": 600,
+				"replicas": 1,
+				"revisionHistoryLimit": 10
+			},
+			"status": {
+				"replicas": 1,
+				"conditions": [
+					{
+						"status": "True",
+						"type": "Progressing"
+					},
+					{
+						"status": "True",
+						"type": "Available"
+					}
+				]
+			}
+			}`)
+
 var testCasesData = []testCases{
 	{
 		jsonPathExprs: "{$.status.conditions[0].status} == True",
 		result:        true,
+		jsonText:      jsonDeployment,
 	},
 	{
 		jsonPathExprs: "{$.spec.replicas} == 1",
 		result:        true,
+		jsonText:      jsonPod,
 	},
 	{
 		jsonPathExprs: "{$.status.conditions[0].status} == {True}",
 		result:        false,
+		jsonText:      jsonStatefulset,
 	},
 }
 
@@ -60,7 +108,7 @@ func TestEvaluateCheckHookExp(t *testing.T) {
 		t.Run(strconv.Itoa(i), func(t *testing.T) {
 			var jsonData map[string]interface{}
 
-			err := json.Unmarshal(jsonText1, &jsonData)
+			err := json.Unmarshal(tt.jsonText, &jsonData)
 			if err != nil {
 				t.Error(err)
 			}