Add daemon-selected container monitor delegation

2026-09-18 14:09:46 -04:00
parent 3937de82dc
commit cbb865729a
6 changed files with 411 additions and 0 deletions
@@ -0,0 +1,139 @@
+#!/usr/bin/env python3
+
+import importlib.util
+from pathlib import Path
+
+
+spec = importlib.util.spec_from_file_location(
+    "proxy_client",
+    Path(
+        "ocf/loxthelion/"
+        "managed-container.d/"
+        "proxy_client.py"
+    ),
+)
+
+proxy = importlib.util.module_from_spec(spec)
+spec.loader.exec_module(proxy)
+
+
+print("===== NORMAL API MONITOR RESULT =====")
+
+normal = {
+    "ha": {
+        "ocf_rc": 0,
+        "health": "healthy",
+        "ha_role": "unpromoted",
+        "state": "streaming",
+        "promotion_eligible": True,
+    },
+    "role": "standby",
+    "streaming": True,
+    "wal_receiver": {
+        "slot_name": "ha_pg_node1",
+        "sender_host": "192.0.2.10",
+    },
+}
+
+formatted = proxy.format_monitor_result(normal)
+
+expected = (
+    "0|healthy|unpromoted|streaming|true|"
+    "standby|true|ha_pg_node1|192.0.2.10"
+)
+
+assert formatted == expected, formatted
+
+print("api_monitor_format       accepted")
+
+
+print()
+print("===== PODMAN DELEGATION =====")
+
+delegated = {
+    "delegated": True,
+    "backend": "container",
+    "container_runtime": "podman",
+}
+
+formatted = proxy.format_monitor_result(delegated)
+
+assert formatted == "delegate|container|podman", formatted
+
+print("podman_delegation        accepted")
+
+
+print()
+print("===== DOCKER DELEGATION =====")
+
+delegated = {
+    "delegated": True,
+    "backend": "container",
+    "container_runtime": "docker",
+}
+
+formatted = proxy.format_monitor_result(delegated)
+
+assert formatted == "delegate|container|docker", formatted
+
+print("docker_delegation        accepted")
+
+
+print()
+print("===== INVALID DELEGATED BACKEND =====")
+
+bad = {
+    "delegated": True,
+    "backend": "api",
+    "container_runtime": "podman",
+}
+
+try:
+    proxy.format_monitor_result(bad)
+except proxy.ProtocolFailure:
+    print("invalid_backend          rejected")
+else:
+    raise AssertionError(
+        "delegation with non-container backend was accepted"
+    )
+
+
+print()
+print("===== INVALID DELEGATED RUNTIME =====")
+
+bad = {
+    "delegated": True,
+    "backend": "container",
+    "container_runtime": "lxc",
+}
+
+try:
+    proxy.format_monitor_result(bad)
+except proxy.ProtocolFailure:
+    print("invalid_runtime          rejected")
+else:
+    raise AssertionError(
+        "delegation with invalid runtime was accepted"
+    )
+
+
+print()
+print("===== MALFORMED DELEGATION =====")
+
+bad = {
+    "delegated": True,
+    "backend": "container",
+}
+
+try:
+    proxy.format_monitor_result(bad)
+except proxy.ProtocolFailure:
+    print("missing_runtime          rejected")
+else:
+    raise AssertionError(
+        "delegation without runtime was accepted"
+    )
+
+
+print()
+print("PROXY_MONITOR_DELEGATION_TEST_PASS=1")