Add governed automation contracts

This commit is contained in:
2026-07-28 15:02:42 +02:00
parent baa2143a26
commit 26ae034153
10 changed files with 539 additions and 0 deletions

View File

@@ -0,0 +1,56 @@
from __future__ import annotations
import unittest
from unittest.mock import MagicMock, patch
from govoplan_core.celery_app import celery, dispatch_dataflow_triggers
class DataflowTriggerWorkerTests(unittest.TestCase):
def test_worker_commits_dispatch_outcome(self) -> None:
session = MagicMock()
database = MagicMock()
database.SessionLocal.return_value.__enter__.return_value = session
provider = MagicMock()
provider.dispatch_due.return_value = {
"queued": 1,
"processed": 1,
"succeeded": 1,
"failed": 0,
"blocked": 0,
"skipped": 0,
}
with (
patch(
"govoplan_core.celery_app._dataflow_trigger_dispatcher",
return_value=provider,
),
patch(
"govoplan_core.db.session.get_database",
return_value=database,
),
):
result = dispatch_dataflow_triggers.run(25)
provider.dispatch_due.assert_called_once_with(session, limit=25)
session.commit.assert_called_once_with()
self.assertEqual(result["succeeded"], 1)
def test_worker_route_and_periodic_dispatch_are_registered(self) -> None:
self.assertEqual(
celery.conf.task_routes["govoplan.dataflow.dispatch_triggers"],
{"queue": "dataflow"},
)
schedule = celery.conf.beat_schedule[
"dataflow-triggers-every-minute"
]
self.assertEqual(
schedule["task"],
"govoplan.dataflow.dispatch_triggers",
)
self.assertEqual(schedule["schedule"], 60.0)
if __name__ == "__main__":
unittest.main()