feat(deployment): add sequencer readiness check stage to system test workflow

Author: nadin-Starkware

.github/workflows/consolidated_system_test.yaml

@@ -114,6 +114,9 @@ jobs:
       - name: Set default namespace
         run: kubectl config set-context --current --namespace ${{ env.namespace }}
 
+      - name: Run readiness check
+        run: pipenv run python ./scripts/system_tests/readiness_check.py --deployment_config_path ${{ env.deployment_config_path }} --namespace ${{ env.namespace }}
+
       - name: Get container logs
         if: always()
         run: |

scripts/system_tests/readiness_check.py

@@ -0,0 +1,138 @@
+import argparse
+import os
+from kubernetes import client, config
+from kubernetes.client.rest import ApiException
+from pathlib import Path
+import json
+import sys
+import time
+
+
+def check_manifest_files(deployment_config_path: str, workspace: str):
+    with open(deployment_config_path, "r", encoding="utf-8") as f:
+        deployment_config = json.load(f)
+
+    services = deployment_config["services"]
+
+    for service in services:
+        service_name = service["name"]
+        controller = service["controller"]
+        service_name_lower = service_name.lower()
+        controller_lower = controller.lower()
+
+        manifest_path = (
+            Path(workspace)
+            / f"deployments/sequencer/dist/sequencer-{service_name_lower}/{controller}.sequencer-{service_name_lower}-{controller_lower}.k8s.yaml"
+        )
+
+        if not manifest_path.exists():
+            print(
+                f"❌ Manifest {manifest_path} for {service_name_lower} not found. Aborting..."
+            )
+            try:
+                dir_listing = list(manifest_path.parent.iterdir())
+                print(
+                    f"Contents of {manifest_path.parent}: {[str(f) for f in dir_listing]}"
+                )
+            except FileNotFoundError:
+                print(f"(Directory {manifest_path.parent} does not exist)")
+            sys.exit(1)
+
+
+def wait_for_services_ready(deployment_config_path: str, namespace: str):
+    config.load_kube_config()  # Or load_incluster_config() if inside a pod
+
+    with open(deployment_config_path, "r", encoding="utf-8") as f:
+        deployment_config = json.load(f)
+
+    services = deployment_config["services"]
+
+    apps_v1 = client.AppsV1Api()
+
+    for service in services:
+        service_name = service["name"]
+        controller = service["controller"]
+        service_name_lower = service_name.lower()
+        controller_lower = controller.lower()
+        resource_name = f"sequencer-{service_name_lower}-{controller_lower}"
+
+        print(f"🔍 Checking {controller_lower}: {resource_name}")
+
+        try:
+            if controller_lower == "statefulset":
+                obj = apps_v1.read_namespaced_stateful_set(
+                    resource_name, namespace=namespace
+                )
+            elif controller_lower == "deployment":
+                obj = apps_v1.read_namespaced_deployment(
+                    resource_name, namespace=namespace
+                )
+            else:
+                print(f"❌ Unknown controller: {controller}. Skipping...")
+                continue
+        except ApiException as e:
+            if e.status == 404:
+                print(f"❌ {controller} {resource_name} not found. Skipping...")
+                continue
+            else:
+                raise
+
+        # Describe & status info (light equivalent of kubectl describe)
+        print(
+            f"🔍 {controller} {resource_name} status: replicas={obj.status.replicas}, ready={obj.status.ready_replicas}"
+        )
+
+        print(f"⏳ Waiting for {controller_lower}/{resource_name} to become ready...")
+
+        timeout_seconds = 180
+        poll_interval = 5
+        elapsed = 0
+
+        while elapsed < timeout_seconds:
+            try:
+                if controller_lower == "statefulset":
+                    status = apps_v1.read_namespaced_stateful_set_status(
+                        resource_name, namespace
+                    ).status
+                    ready = status.ready_replicas or 0
+                    desired = status.replicas or 0
+                else:  # deployment
+                    status = apps_v1.read_namespaced_deployment_status(
+                        resource_name, namespace
+                    ).status
+                    ready = status.ready_replicas or 0
+                    desired = status.replicas or 0
+
+                if ready == desired and ready > 0:
+                    print(f"✅ {controller} {resource_name} is ready.")
+                    break
+            except ApiException as e:
+                print(f"❌ Error while checking status: {e}")
+
+            time.sleep(poll_interval)
+            elapsed += poll_interval
+        else:
+            print(
+                f"⚠️ Timeout waiting for {controller} {resource_name} to become ready."
+            )
+            sys.exit(1)
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(
+        description="Check manifest files and wait for K8s services to be ready."
+    )
+    parser.add_argument(
+        "--deployment_config_path", help="Path to the deployment config JSON file"
+    )
+    parser.add_argument(
+        "--namespace",
+        help="Kubernetes namespace",
+    )
+    args = parser.parse_args()
+
+    github_workspace = os.environ["GITHUB_WORKSPACE"]
+
+    check_manifest_files(args.deployment_config_path, github_workspace)
+    wait_for_services_ready(args.deployment_config_path, namespace=args.namespace)
+    print("✅ All sequencer services are ready.")