From 49eca129a6e972a8672d812ef75caf1466d27275 Mon Sep 17 00:00:00 2001
From: Raghav <raghav.gupta@harness.io>
Date: Thu, 23 May 2024 16:24:04 +0530
Subject: [PATCH] [CI-11703]: Fix image pull policy enums (#283)

---
 engine/spec/const.go      | 16 ++++++++--------
 engine/spec/const_test.go | 18 +++++++++---------
 2 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/engine/spec/const.go b/engine/spec/const.go
index 2bb9bcea..bf93e5b4 100644
--- a/engine/spec/const.go
+++ b/engine/spec/const.go
@@ -30,17 +30,17 @@ func (p PullPolicy) String() string {
 
 var pullPolicyID = map[PullPolicy]string{
 	PullDefault:     "default",
-	PullAlways:      "always",
-	PullIfNotExists: "if-not-exists",
-	PullNever:       "never",
+	PullAlways:      "Always",
+	PullIfNotExists: "IfNotPresent",
+	PullNever:       "Never",
 }
 
 var pullPolicyName = map[string]PullPolicy{
-	"":              PullDefault,
-	"default":       PullDefault,
-	"always":        PullAlways,
-	"if-not-exists": PullIfNotExists,
-	"never":         PullNever,
+	"":             PullDefault,
+	"default":      PullDefault,
+	"Always":       PullAlways,
+	"IfNotPresent": PullIfNotExists,
+	"Never":        PullNever,
 }
 
 // MarshalJSON marshals the string representation of the
diff --git a/engine/spec/const_test.go b/engine/spec/const_test.go
index cb4cbd43..6bd51c9b 100644
--- a/engine/spec/const_test.go
+++ b/engine/spec/const_test.go
@@ -21,7 +21,7 @@ func TestPullPolicy_Marshal(t *testing.T) {
 	}{
 		{
 			policy: PullAlways,
-			data:   `"always"`,
+			data:   `"Always"`,
 		},
 		{
 			policy: PullDefault,
@@ -29,11 +29,11 @@ func TestPullPolicy_Marshal(t *testing.T) {
 		},
 		{
 			policy: PullIfNotExists,
-			data:   `"if-not-exists"`,
+			data:   `"IfNotPresent"`,
 		},
 		{
 			policy: PullNever,
-			data:   `"never"`,
+			data:   `"Never"`,
 		},
 	}
 	for _, test := range tests {
@@ -55,7 +55,7 @@ func TestPullPolicy_Unmarshal(t *testing.T) {
 	}{
 		{
 			policy: PullAlways,
-			data:   `"always"`,
+			data:   `"Always"`,
 		},
 		{
 			policy: PullDefault,
@@ -63,11 +63,11 @@ func TestPullPolicy_Unmarshal(t *testing.T) {
 		},
 		{
 			policy: PullIfNotExists,
-			data:   `"if-not-exists"`,
+			data:   `"IfNotPresent"`,
 		},
 		{
 			policy: PullNever,
-			data:   `"never"`,
+			data:   `"Never"`,
 		},
 		{
 			// no policy should default to on-success
@@ -103,7 +103,7 @@ func TestPullPolicy_String(t *testing.T) {
 	}{
 		{
 			policy: PullAlways,
-			value:  "always",
+			value:  "Always",
 		},
 		{
 			policy: PullDefault,
@@ -111,11 +111,11 @@ func TestPullPolicy_String(t *testing.T) {
 		},
 		{
 			policy: PullIfNotExists,
-			value:  "if-not-exists",
+			value:  "IfNotPresent",
 		},
 		{
 			policy: PullNever,
-			value:  "never",
+			value:  "Never",
 		},
 	}
 	for _, test := range tests {