feat: Implement MachineHealthCheck CRD health check

Signed-off-by: Jellyfrog <Jellyfrog@users.noreply.github.com>
argoproj · Oct 7, 2022 · 7b4d964 · 7b4d964
1 parent 22ba8f5
commit 7b4d964
Show file tree

Hide file tree

Showing 4 changed files with 89 additions and 0 deletions.
diff --git a/resource_customizations/cluster.x-k8s.io/MachineHealthCheck/health.lua b/resource_customizations/cluster.x-k8s.io/MachineHealthCheck/health.lua
@@ -0,0 +1,14 @@
+hs = {}
+
+hs.status = "Progressing"
+hs.message = ""
+
+if obj.status ~= nil and obj.status.currentHealthy ~= nil then
+    if obj.status.expectedMachines == obj.status.currentHealthy then
+        hs.status = "Healthy"
+    else
+        hs.status = "Degraded"
+    end
+end
+
+return hs
diff --git a/resource_customizations/cluster.x-k8s.io/MachineHealthCheck/health_test.yaml b/resource_customizations/cluster.x-k8s.io/MachineHealthCheck/health_test.yaml
@@ -0,0 +1,9 @@
+tests:
+- healthStatus:
+    status: Degraded
+    message: ''
+  inputPath: testdata/degraded_expectedMachines.yaml
+- healthStatus:
+    status: Healthy
+    message: ''
+  inputPath: testdata/healthy.yaml
diff --git a/...ustomizations/cluster.x-k8s.io/MachineHealthCheck/testdata/degraded_expectedMachines.yaml b/...ustomizations/cluster.x-k8s.io/MachineHealthCheck/testdata/degraded_expectedMachines.yaml
@@ -0,0 +1,33 @@
+apiVersion: cluster.x-k8s.io/v1beta1
+kind: MachineHealthCheck
+metadata:
+  labels:
+    cluster.x-k8s.io/cluster-name: test
+  name: test-node-unhealthy-5m
+spec:
+  clusterName: test
+  maxUnhealthy: 100%
+  nodeStartupTimeout: 10m0s
+  selector:
+    matchLabels:
+      cluster.x-k8s.io/deployment-name: test-md-workers-0
+  unhealthyConditions:
+  - status: Unknown
+    timeout: 5m
+    type: Ready
+  - status: "False"
+    timeout: 5m
+    type: Ready
+status:
+  conditions:
+  - lastTransitionTime: "2022-10-07T10:33:46Z"
+    status: "True"
+    type: RemediationAllowed
+  currentHealthy: 1
+  expectedMachines: 3
+  observedGeneration: 3
+  remediationsAllowed: 1
+  targets:
+  - test-md-workers-0-76f7db5786-8nl6m
+  - test-md-workers-0-76f7db5786-jjzvf
+  - test-md-workers-0-76f7db5786-l4vfb
diff --git a/resource_customizations/cluster.x-k8s.io/MachineHealthCheck/testdata/healthy.yaml b/resource_customizations/cluster.x-k8s.io/MachineHealthCheck/testdata/healthy.yaml
@@ -0,0 +1,33 @@
+apiVersion: cluster.x-k8s.io/v1beta1
+kind: MachineHealthCheck
+metadata:
+  labels:
+    cluster.x-k8s.io/cluster-name: test
+  name: test-node-unhealthy-5m
+spec:
+  clusterName: test
+  maxUnhealthy: 100%
+  nodeStartupTimeout: 10m0s
+  selector:
+    matchLabels:
+      cluster.x-k8s.io/deployment-name: test-md-workers-0
+  unhealthyConditions:
+  - status: Unknown
+    timeout: 5m
+    type: Ready
+  - status: "False"
+    timeout: 5m
+    type: Ready
+status:
+  conditions:
+  - lastTransitionTime: "2022-10-07T10:33:46Z"
+    status: "True"
+    type: RemediationAllowed
+  currentHealthy: 3
+  expectedMachines: 3
+  observedGeneration: 3
+  remediationsAllowed: 1
+  targets:
+  - test-md-workers-0-76f7db5786-8nl6m
+  - test-md-workers-0-76f7db5786-jjzvf
+  - test-md-workers-0-76f7db5786-l4vfb