merge: testing → main (reconcile 2-week divergence)

tests/fleet/test_reconciler_worker.py

@@ -0,0 +1,72 @@
+"""Worker shutdown smoke test for fleet_reconciler_worker.
+
+The reconcile logic itself is exercised in test_reconciler.py.  This file
+just verifies the worker's lifecycle wrapper (control listener + heartbeat
++ tick loop) exits cleanly when the bus shutdown signal fires.
+"""
+from __future__ import annotations
+
+import asyncio
+from unittest.mock import AsyncMock, patch
+
+import pytest
+
+from decnet.fleet.reconciler_worker import fleet_reconciler_worker
+
+
+class _FakeRepo:
+    async def list_fleet_deckies(self, *, host_uuid=None):
+        return []
+    async def upsert_fleet_decky(self, data): pass
+    async def delete_fleet_decky(self, **kw): pass
+    async def update_fleet_decky_state(self, **kw): pass
+
+
+@pytest.mark.anyio
+async def test_worker_exits_on_shutdown_event(monkeypatch):
+    # Patch the bus + control listener so the worker doesn't try to bind
+    # to a real socket. The control_task will set `shutdown` once we fire it.
+    fake_bus = AsyncMock()
+    monkeypatch.setattr(
+        "decnet.fleet.reconciler_worker.get_bus",
+        lambda **kw: fake_bus,
+    )
+
+    captured: dict = {}
+
+    async def _capturing_control_listener(bus, name, shutdown_event):
+        captured["shutdown_event"] = shutdown_event
+        # Hold the event loop briefly so the worker enters its tick wait,
+        # then trigger shutdown.
+        await asyncio.sleep(0.05)
+        shutdown_event.set()
+
+    async def _noop_heartbeat(bus, name):
+        await asyncio.sleep(3600)  # never returns naturally
+
+    monkeypatch.setattr(
+        "decnet.fleet.reconciler_worker.run_control_listener",
+        _capturing_control_listener,
+    )
+    monkeypatch.setattr(
+        "decnet.fleet.reconciler_worker.run_health_heartbeat",
+        _noop_heartbeat,
+    )
+    # Skip docker observation entirely — we just need the loop to exit.
+    monkeypatch.setattr(
+        "decnet.fleet.reconciler._real_load_state",
+        lambda: None,
+    )
+    with patch("decnet.fleet.reconciler._collect_container_states",
+               return_value=None):
+        # interval=10 (long) so we exit via shutdown, not via tick completion
+        await asyncio.wait_for(
+            fleet_reconciler_worker(_FakeRepo(), interval=10),
+            timeout=2.0,
+        )
+    assert captured["shutdown_event"].is_set()
+
+
+@pytest.fixture
+def anyio_backend():
+    return "asyncio"