feat: support autonomous campaign schedule results

2026-08-20 21:10:04 +02:00
parent 4f4007aff1
commit 4b0737e1cd
3 changed files with 118 additions and 2 deletions
@@ -0,0 +1,106 @@
+from __future__ import annotations
+
+import unittest
+from contextlib import contextmanager
+from types import SimpleNamespace
+from unittest.mock import MagicMock, patch
+
+from govoplan_core.celery_app import celery, dispatch_campaign_schedules
+from tests.worker_test_support import allowed_worker_admissions
+
+
+class CampaignScheduleWorkerTests(unittest.TestCase):
+    def test_dispatch_preserves_autonomous_recovery_result(self) -> None:
+        session = MagicMock()
+
+        @contextmanager
+        def session_scope():
+            yield session
+
+        provider = MagicMock()
+        provider.dispatch_due.return_value = {
+            "selected": 1,
+            "prepared": 0,
+            "autonomous_prepared": 1,
+            "failed": 0,
+            "completed": 0,
+            "coalesced": 0,
+            "duplicates": 0,
+            "deferred": 0,
+            "campaign_ids": ["campaign-1"],
+            "operator_actions": [],
+            "refreshed": {
+                "checked": 1,
+                "accepted": 1,
+                "uncertain": 0,
+                "failed": 0,
+                "skipped": 0,
+            },
+        }
+        registry = MagicMock()
+        registry.has_capability.return_value = True
+        database = SimpleNamespace(SessionLocal=session_scope)
+
+        with (
+            patch("govoplan_core.celery_app._platform_registry", return_value=registry),
+            patch("govoplan_core.celery_app._campaign_schedules", return_value=provider),
+            patch(
+                "govoplan_core.celery_app._worker_admissions",
+                side_effect=allowed_worker_admissions,
+            ),
+            patch("govoplan_core.db.session.get_database", return_value=database),
+        ):
+            result = dispatch_campaign_schedules.run("tenant-1", 17)
+
+        provider.dispatch_due.assert_called_once_with(
+            session,
+            tenant_id="tenant-1",
+            limit=17,
+        )
+        session.commit.assert_called_once_with()
+        self.assertEqual(result["autonomous_prepared"], 1)
+        self.assertEqual(result["refreshed"]["accepted"], 1)
+
+    def test_missing_provider_returns_complete_autonomous_defaults(self) -> None:
+        session = MagicMock()
+
+        @contextmanager
+        def session_scope():
+            yield session
+
+        registry = MagicMock()
+        registry.has_capability.return_value = False
+        database = SimpleNamespace(SessionLocal=session_scope)
+
+        with (
+            patch("govoplan_core.celery_app._platform_registry", return_value=registry),
+            patch("govoplan_core.db.session.get_database", return_value=database),
+        ):
+            result = dispatch_campaign_schedules.run("tenant-1", 17)
+
+        self.assertEqual(result["autonomous_prepared"], 0)
+        self.assertEqual(result["duplicates"], 0)
+        self.assertEqual(result["deferred"], 0)
+        self.assertEqual(
+            result["refreshed"],
+            {
+                "checked": 0,
+                "accepted": 0,
+                "uncertain": 0,
+                "failed": 0,
+                "skipped": 0,
+            },
+        )
+
+    def test_worker_route_and_periodic_dispatch_are_registered(self) -> None:
+        self.assertEqual(
+            celery.conf.task_routes["govoplan.campaigns.dispatch_schedules"],
+            {"queue": "default"},
+        )
+        schedule = celery.conf.beat_schedule["campaign-schedules-every-minute"]
+        self.assertEqual(schedule["task"], "govoplan.campaigns.dispatch_schedules")
+        self.assertEqual(schedule["schedule"], 60.0)
+
+
+if __name__ == "__main__":
+    unittest.main()