feat: rewrite controlplane integration tests with pytest

Author: Hweinstock

.github/workflows/integration-testing.yml

@@ -118,10 +118,15 @@ jobs:
             extra-deps: ""
             ignore: ""
           # TODO: expand to full tests_integ/memory once test stability is addressed
-          - group: memory
+          - group: memory-controlplane
+            path: tests_integ/memory/test_controlplane.py
+            timeout: 10
+            extra-deps: "pytest-order"
+            ignore: ""
+          - group: memory-client
             path: tests_integ/memory/test_memory_client.py
             timeout: 10
-            extra-deps: ""
+            extra-deps: "pytest-order"
             ignore: ""
           - group: evaluation
             path: tests_integ/evaluation

pyproject.toml

@@ -144,6 +144,7 @@ dev = [
     "pytest>=8.4.1",
     "pytest-asyncio>=0.24.0",
     "pytest-cov>=6.0.0",
+    "pytest-order>=1.3.0",
     "ruff>=0.12.0",
     "websockets>=14.1",
     "wheel>=0.45.1",

tests_integ/memory/helpers.py

@@ -0,0 +1,16 @@
+import time
+
+import pytest
+
+
+def poll_until(fn, predicate, max_wait=180, poll_interval=10, fail_if=None):
+    """Poll fn() until predicate(result) is True or max_wait exceeded."""
+    deadline = time.time() + max_wait
+    while time.time() < deadline:
+        result = fn()
+        if fail_if and fail_if(result):
+            pytest.fail(f"poll_until hit terminal state: {result}")
+        if predicate(result):
+            return result
+        time.sleep(poll_interval)
+    pytest.fail(f"poll_until timed out after {max_wait}s")