feat(core): reconcile durable workflow instances

2026-07-30 03:09:58 +02:00
parent cf7afe9dda
commit e7c84e3227
3 changed files with 179 additions and 0 deletions
--- a/tests/test_workflow_runtime_worker.py
+++ b/tests/test_workflow_runtime_worker.py
@@ -0,0 +1,87 @@
+from __future__ import annotations
+
+import unittest
+from unittest.mock import MagicMock, patch
+
+from govoplan_core.celery_app import (
+    celery,
+    reconcile_workflow_instances,
+)
+from govoplan_core.core.modules import ModuleContext, ModuleManifest
+from govoplan_core.core.registry import PlatformRegistry
+from govoplan_core.core.workflows import (
+    CAPABILITY_WORKFLOW_RUNTIME_WORKER,
+    WorkflowRuntimeWorker,
+    workflow_runtime_worker,
+)
+
+
+class _Worker:
+    def reconcile_pending(self, session, *, now=None, limit=50):
+        return {"inspected": 0, "advanced": 0}
+
+
+class WorkflowRuntimeWorkerTests(unittest.TestCase):
+    def test_contract_is_runtime_checkable_and_resolved(self) -> None:
+        worker = _Worker()
+        self.assertIsInstance(worker, WorkflowRuntimeWorker)
+        registry = PlatformRegistry()
+        registry.register(
+            ModuleManifest(
+                id="workflow_runtime_test",
+                name="Workflow runtime test",
+                version="test",
+                capability_factories={
+                    CAPABILITY_WORKFLOW_RUNTIME_WORKER: (
+                        lambda _context: worker
+                    ),
+                },
+            )
+        )
+        registry.configure_capability_context(
+            ModuleContext(registry=registry, settings=object())
+        )
+
+        self.assertIs(worker, workflow_runtime_worker(registry))
+        self.assertIsNone(workflow_runtime_worker(PlatformRegistry()))
+
+    def test_celery_task_commits_reconciliation(self) -> None:
+        session = MagicMock()
+        database = MagicMock()
+        database.SessionLocal.return_value.__enter__.return_value = session
+        worker = MagicMock()
+        worker.reconcile_pending.return_value = {
+            "inspected": 2,
+            "advanced": 1,
+        }
+
+        with (
+            patch(
+                "govoplan_core.celery_app._workflow_runtime_worker",
+                return_value=worker,
+            ),
+            patch(
+                "govoplan_core.db.session.get_database",
+                return_value=database,
+            ),
+        ):
+            result = reconcile_workflow_instances.run(25)
+
+        worker.reconcile_pending.assert_called_once_with(session, limit=25)
+        session.commit.assert_called_once_with()
+        self.assertEqual(1, result["advanced"])
+
+    def test_route_and_periodic_job_are_registered(self) -> None:
+        self.assertEqual(
+            {"queue": "workflow"},
+            celery.conf.task_routes["govoplan.workflow.reconcile"],
+        )
+        schedule = celery.conf.beat_schedule[
+            "workflow-reconcile-every-five-seconds"
+        ]
+        self.assertEqual("govoplan.workflow.reconcile", schedule["task"])
+        self.assertEqual(5.0, schedule["schedule"])
+
+
+if __name__ == "__main__":
+    unittest.main()