Refactor blocklist import registration to async startup flow

backend/app/startup.py

@@ -123,7 +123,7 @@ async def startup_shared_resources(
         scheduler.start()
 
         health_check.register(app)
-        blocklist_import.register(app)
+        await blocklist_import.register(app)
         geo_cache_flush.register(app)
         geo_re_resolve.register(app)
         history_sync.register(app)

backend/app/tasks/blocklist_import.py

@@ -13,7 +13,6 @@ existing entry without creating duplicates.
 
 from __future__ import annotations
 
-import inspect
 from typing import TYPE_CHECKING, Any
 
 import structlog
@@ -76,7 +75,7 @@ async def _run_import(app: Any) -> None:
             await db.close()
 
 
-def register(app: FastAPI) -> None:
+async def register(app: FastAPI) -> None:
     """Add (or replace) the blocklist import job in the application scheduler.
 
     Reads the persisted :class:`~app.models.blocklist.ScheduleConfig` from
@@ -89,30 +88,14 @@ def register(app: FastAPI) -> None:
         app: The :class:`fastapi.FastAPI` application instance whose
             ``app.state.scheduler`` will receive the job.
     """
-    import asyncio  # noqa: PLC0415
-
-    async def _do_register() -> None:
-        db, close_db = await _get_db(app)
-        try:
-            config = await blocklist_service.get_schedule(db)
-        finally:
-            if close_db:
-                await db.close()
-        _apply_schedule(app, config)
-
-    # APScheduler is synchronous at registration time; use asyncio to read
-    # the stored schedule from the DB before registering.
-    coro = None
+    db, close_db = await _get_db(app)
     try:
-        loop = asyncio.get_event_loop()
-        coro = _do_register()
-        loop.run_until_complete(coro)
-    except RuntimeError:
-        # If the current thread already has a running loop (uvicorn), schedule
-        # the registration as a coroutine.
-        if coro is not None and inspect.getcoroutinestate(coro) != inspect.CORO_CLOSED:
-            coro.close()
-        asyncio.ensure_future(_do_register())
+        config = await blocklist_service.get_schedule(db)
+    finally:
+        if close_db:
+            await db.close()
+
+    _apply_schedule(app, config)
 
 
 def reschedule(app: FastAPI) -> None:

backend/tests/test_tasks/test_blocklist_import.py

@@ -300,10 +300,9 @@ class TestApplySchedule:
 class TestRegister:
     """Tests for :func:`~app.tasks.blocklist_import.register`."""
 
-    def test_register_calls_apply_schedule_via_event_loop(self) -> None:
+    @pytest.mark.asyncio
+    async def test_register_calls_apply_schedule(self) -> None:
         """``register`` must call ``_apply_schedule`` after reading the stored config."""
-        import asyncio
-
         from app.tasks.blocklist_import import register
 
         app = MagicMock()
@@ -324,52 +323,10 @@ class TestRegister:
             new_callable=AsyncMock,
             return_value=config,
         ), patch("app.tasks.blocklist_import._apply_schedule") as mock_apply:
-            # Use a fresh event loop to avoid interference from pytest-asyncio.
-            loop = asyncio.new_event_loop()
-            try:
-                with patch("asyncio.get_event_loop", return_value=loop):
-                    register(app)
-            finally:
-                loop.close()
+            await register(app)
 
         mock_apply.assert_called_once_with(app, config)
 
-    def test_register_falls_back_to_ensure_future_on_runtime_error(self) -> None:
-        """When ``run_until_complete`` raises ``RuntimeError``, ``ensure_future`` is used."""
-        from app.tasks.blocklist_import import register
-
-        app = MagicMock()
-        app.state.db = MagicMock()
-        app.state.db.close = AsyncMock()
-        app.state.settings = MagicMock(database_path="/tmp/fake.db")
-        app.state.scheduler = MagicMock()
-
-        config = ScheduleConfig(frequency=ScheduleFrequency.daily)
-
-        mock_loop = MagicMock()
-        mock_loop.run_until_complete.side_effect = RuntimeError("already running")
-
-        def _close_coro(coro: Any) -> None:
-            coro.close()
-
-        with (
-            patch(
-                "app.tasks.blocklist_import.open_db",
-                new_callable=AsyncMock,
-                return_value=app.state.db,
-            ),
-            patch(
-                "app.tasks.blocklist_import.blocklist_service.get_schedule",
-                new_callable=AsyncMock,
-                return_value=config,
-            ),
-            patch("asyncio.get_event_loop", return_value=mock_loop),
-            patch("asyncio.ensure_future", side_effect=_close_coro) as mock_ensure_future,
-        ):
-            register(app)
-
-        mock_ensure_future.assert_called_once()
-
 
 class TestReschedule:
     """Tests for :func:`~app.tasks.blocklist_import.reschedule`."""