From 59ea2a809e5ad4db2d08d38e029e8484ae9d48c1 Mon Sep 17 00:00:00 2001
From: Dillen Padhiar <38965141+dpadhiar@users.noreply.github.com>
Date: Thu, 31 Oct 2024 10:53:41 -0700
Subject: [PATCH] feat: support Suspended state for MonoVertexRollout (#20583)

Signed-off-by: Dillen Padhiar <dillen_padhiar@intuit.com>
---
 .../MonoVertexRollout/health.lua              |  8 +++
 .../MonoVertexRollout/health_test.yaml        |  4 ++
 .../testdata/MonoVertexRollout/paused.yaml    | 55 +++++++++++++++++++
 3 files changed, 67 insertions(+)
 create mode 100644 resource_customizations/numaplane.numaproj.io/MonoVertexRollout/testdata/MonoVertexRollout/paused.yaml

diff --git a/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health.lua b/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health.lua
index 8ae7650e5cf2b..e91efc6ac68ab 100644
--- a/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health.lua
+++ b/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health.lua
@@ -1,5 +1,6 @@
 local hs = {}
 local healthyCondition = {}
+local monoVertexPaused = {}
 
 -- check for certain cases of "Progressing"
 
@@ -35,6 +36,9 @@ if obj.status.conditions ~= nil then
     if condition.type == "ChildResourcesHealthy" then
       healthyCondition = condition
     end
+    if condition.type == "MonoVertexPausingOrPaused" then
+      monoVertexPaused = condition
+    end
   end
 end
 
@@ -51,6 +55,10 @@ elseif healthyCondition ~= {} and healthyCondition.status == "False" and healthy
   hs.status = "Progressing"
   hs.message = healthyCondition.message
   return hs
+elseif (monoVertexPaused ~= {} and monoVertexPaused.status == "True") then
+  hs.status = "Suspended"
+  hs.message = monoVertexPaused.message
+  return hs
 elseif healthyCondition ~= {} and healthyCondition.status == "True" and obj.status.phase == "Deployed" then
   hs.status = "Healthy"
   hs.message = healthyCondition.message
diff --git a/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health_test.yaml b/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health_test.yaml
index 82a52ec141c23..b86d285232d9a 100644
--- a/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health_test.yaml
+++ b/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/health_test.yaml
@@ -7,6 +7,10 @@ tests:
     status: Healthy
     message: "Successful"
   inputPath: testdata/MonoVertexRollout/healthy.yaml
+- healthStatus:
+    status: Suspended
+    message: "MonoVertex paused"
+  inputPath: testdata/MonoVertexRollout/paused.yaml
 - healthStatus:
     status: Degraded
     message: "MonoVertex Failed"
diff --git a/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/testdata/MonoVertexRollout/paused.yaml b/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/testdata/MonoVertexRollout/paused.yaml
new file mode 100644
index 0000000000000..0aec918a6b723
--- /dev/null
+++ b/resource_customizations/numaplane.numaproj.io/MonoVertexRollout/testdata/MonoVertexRollout/paused.yaml
@@ -0,0 +1,55 @@
+apiVersion: numaplane.numaproj.io/v1alpha1
+kind: MonoVertexRollout
+metadata:
+  annotations:
+    kubectl.kubernetes.io/last-applied-configuration: |
+      {"apiVersion":"numaplane.numaproj.io/v1alpha1","kind":"MonoVertexRollout","metadata":{"annotations":{},"name":"my-monovertex","namespace":"example-namespace"},"spec":{"monoVertex":{"spec":{"lifecycle":{"desiredPhase":"Paused"},"sink":{"udsink":{"container":{"image":"quay.io/numaio/numaflow-java/simple-sink:stable"}}},"source":{"transformer":{"container":{"image":"quay.io/numaio/numaflow-rs/source-transformer-now:stable"}},"udsource":{"container":{"image":"quay.io/numaio/numaflow-java/source-simple-source:stable"}}}}}}}
+  creationTimestamp: "2024-10-29T20:44:33Z"
+  finalizers:
+  - numaplane.numaproj.io/numaplane-controller
+  generation: 3
+  name: my-monovertex
+  namespace: example-namespace
+  resourceVersion: "295152"
+  uid: b1bc7ad6-4ffe-44e5-ad1f-57455f96921d
+spec:
+  monoVertex:
+    metadata: {}
+    spec:
+      lifecycle:
+        desiredPhase: Paused
+      sink:
+        udsink:
+          container:
+            image: quay.io/numaio/numaflow-java/simple-sink:stable
+      source:
+        transformer:
+          container:
+            image: quay.io/numaio/numaflow-rs/source-transformer-now:stable
+        udsource:
+          container:
+            image: quay.io/numaio/numaflow-java/source-simple-source:stable
+status:
+  conditions:
+  - lastTransitionTime: "2024-10-29T20:44:33Z"
+    message: Successful
+    observedGeneration: 3
+    reason: Successful
+    status: "True"
+    type: ChildResourceDeployed
+  - lastTransitionTime: "2024-10-29T20:44:59Z"
+    message: MonoVertex Pausing - health unknown
+    observedGeneration: 3
+    reason: MonoVertexUnknown
+    status: Unknown
+    type: ChildResourcesHealthy
+  - lastTransitionTime: "2024-10-29T20:44:58Z"
+    message: MonoVertex paused
+    observedGeneration: 3
+    reason: MonoVertexPaused
+    status: "True"
+    type: MonoVertexPausingOrPaused
+  message: Deployed
+  nameCount: 1
+  observedGeneration: 3
+  phase: Deployed
\ No newline at end of file