60 lines
1.9 KiB
Python
60 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
from govoplan_core.celery_app import celery, worker_acceptance_probe
|
|
from govoplan_core.settings import settings
|
|
|
|
|
|
class CeleryQueueContractTests(unittest.TestCase):
|
|
def test_default_worker_queues_cover_every_routed_task(self) -> None:
|
|
configured = {
|
|
value.strip()
|
|
for value in settings.celery_queues.split(",")
|
|
if value.strip()
|
|
}
|
|
routes = celery.conf.task_routes
|
|
self.assertIsInstance(routes, dict)
|
|
routed = {
|
|
str(route["queue"])
|
|
for route in routes.values()
|
|
if isinstance(route, dict) and route.get("queue")
|
|
}
|
|
|
|
self.assertEqual(set(), routed - configured)
|
|
|
|
def test_delivery_tasks_keep_worker_loss_protection(self) -> None:
|
|
self.assertTrue(celery.conf.task_acks_late)
|
|
self.assertTrue(celery.conf.task_reject_on_worker_lost)
|
|
self.assertTrue(celery.conf.task_track_started)
|
|
self.assertEqual(1, celery.conf.worker_prefetch_multiplier)
|
|
self.assertEqual(
|
|
settings.celery_visibility_timeout_seconds,
|
|
celery.conf.broker_transport_options["visibility_timeout"],
|
|
)
|
|
|
|
def test_worker_acceptance_probe_is_bounded_and_side_effect_free(self) -> None:
|
|
with patch.object(worker_acceptance_probe, "update_state") as update_state:
|
|
result = worker_acceptance_probe.run(
|
|
"probe-1",
|
|
mode="complete",
|
|
delay_seconds=0,
|
|
)
|
|
|
|
self.assertEqual(
|
|
{
|
|
"probe_id": "probe-1",
|
|
"mode": "complete",
|
|
"retries": 0,
|
|
"redelivered": False,
|
|
"delivery_count": None,
|
|
},
|
|
result,
|
|
)
|
|
update_state.assert_called_once()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|