fix: await takeover polling jobs

digital-avatar-app/backend/tests/test_takeover_scheduler.py

@@ -14,7 +14,7 @@ def test_app_has_startup_event():
 @patch("services.takeover_service.TakeoverService")
 @patch("services.boxim_client.BoxIMClient")
 @patch("main.redis_lib.from_url")
-@patch("main.BackgroundScheduler")
+@patch("main.AsyncIOScheduler")
 def test_scheduler_initialized_with_redis(mock_scheduler_class, mock_redis_from_url, mock_boxim_cls, mock_takeover_cls):
     """Verify scheduler is initialized when Redis is available."""
     mock_redis = MagicMock()
@@ -25,6 +25,7 @@ def test_scheduler_initialized_with_redis(mock_scheduler_class, mock_redis_from_
     mock_boxim_cls.return_value = mock_boxim
 
     mock_takeover = MagicMock()
+    mock_takeover.poll_and_process_messages = AsyncMock()
     mock_takeover_cls.return_value = mock_takeover
 
     with patch("main.init_db"), patch("main.seed"), patch.dict("os.environ", {"REDIS_URL": "redis://localhost:6379"}):
@@ -32,19 +33,23 @@ def test_scheduler_initialized_with_redis(mock_scheduler_class, mock_redis_from_
         on_startup()
 
     mock_scheduler_class.return_value.add_job.assert_called_once()
+    scheduled_callable = mock_scheduler_class.return_value.add_job.call_args.args[0]
     call_kwargs = mock_scheduler_class.return_value.add_job.call_args[1]
+    assert scheduled_callable is mock_takeover.poll_and_process_messages
     assert call_kwargs["id"] == "takeover_message_poll"
+    mock_scheduler_class.return_value.start.assert_called_once_with()
 
 
 @patch("services.takeover_service.TakeoverService")
 @patch("services.boxim_client.BoxIMClient")
-@patch("main.BackgroundScheduler")
+@patch("main.AsyncIOScheduler")
 def test_scheduler_starts_without_redis(mock_scheduler_class, mock_boxim_cls, mock_takeover_cls):
     """App should start even when REDIS_URL is not set."""
     mock_boxim = MagicMock()
     mock_boxim_cls.return_value = mock_boxim
 
     mock_takeover = MagicMock()
+    mock_takeover.poll_and_process_messages = AsyncMock()
     mock_takeover_cls.return_value = mock_takeover
 
     with patch("main.init_db"), patch("main.seed"), patch.dict("os.environ", {"REDIS_URL": ""}, clear=False):
@@ -57,7 +62,7 @@ def test_scheduler_starts_without_redis(mock_scheduler_class, mock_boxim_cls, mo
 @patch("services.takeover_service.TakeoverService")
 @patch("services.boxim_client.BoxIMClient")
 @patch("main.redis_lib.from_url")
-@patch("main.BackgroundScheduler")
+@patch("main.AsyncIOScheduler")
 def test_scheduler_starts_when_redis_fails(mock_scheduler_class, mock_redis_from_url, mock_boxim_cls, mock_takeover_cls):
     """App should start even when Redis ping fails."""
     mock_redis_from_url.side_effect = ConnectionError("Connection refused")
@@ -66,6 +71,7 @@ def test_scheduler_starts_when_redis_fails(mock_scheduler_class, mock_redis_from
     mock_boxim_cls.return_value = mock_boxim
 
     mock_takeover = MagicMock()
+    mock_takeover.poll_and_process_messages = AsyncMock()
     mock_takeover_cls.return_value = mock_takeover
 
     with patch("main.init_db"), patch("main.seed"), patch.dict("os.environ", {"REDIS_URL": "redis://badhost:6379"}):
@@ -75,7 +81,7 @@ def test_scheduler_starts_when_redis_fails(mock_scheduler_class, mock_redis_from
     mock_scheduler_class.return_value.add_job.assert_called_once()
 
 
-@patch("main.BackgroundScheduler")
+@patch("main.AsyncIOScheduler")
 def test_scheduler_fails_gracefully(mock_scheduler_class):
     """If scheduler init raises, the app should still start (exception caught)."""
     mock_scheduler_class.side_effect = RuntimeError("Scheduler crash")
@@ -87,6 +93,24 @@ def test_scheduler_fails_gracefully(mock_scheduler_class):
     # No exception should propagate
 
 
+def test_scheduler_shutdown_releases_resources():
+    """Shutdown should stop polling and close its dedicated database session."""
+    import main
+
+    mock_scheduler = MagicMock()
+    mock_scheduler.running = True
+    mock_db = MagicMock()
+    main.takeover_scheduler = mock_scheduler
+    main.takeover_db = mock_db
+
+    main.on_shutdown()
+
+    mock_scheduler.shutdown.assert_called_once_with(wait=False)
+    mock_db.close.assert_called_once_with()
+    assert main.takeover_scheduler is None
+    assert main.takeover_db is None
+
+
 # --- poll_and_process_messages ---