test(proxy): assert the scheduled add_deployment job in the startup DB lookup tests

The three general_settings startup tests only inspected mocks, which the
TQ002 mock-echo gate rejects. Inject a real AsyncIOScheduler and assert
whether add_deployment_job was scheduled, which is what store_model_in_db
actually controls. The retry and string-normalization tests fail on the
pre-fix proxy_server.py.

tests/unit/proxy/test_proxy_server_endpoints_and_startup.py

@@ -1170,6 +1170,8 @@ async def test_initialize_scheduled_jobs_recovers_from_transient_read_error(monk
     """
     import types
 
+    from apscheduler.schedulers.asyncio import AsyncIOScheduler
+
     monkeypatch.delenv("DISABLE_PRISMA_SCHEMA_UPDATE", raising=False)
     monkeypatch.delenv("STORE_MODEL_IN_DB", raising=False)
     from litellm.proxy.proxy_server import ProxyStartupEvent
@@ -1187,23 +1189,29 @@ async def test_initialize_scheduled_jobs_recovers_from_transient_read_error(monk
     mock_proxy_logging.db_spend_update_writer = MagicMock()
     mock_proxy_config = _mock_scheduled_proxy_config()
 
-    with (
-        patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config),
-        patch("litellm.proxy.proxy_server.store_model_in_db", False),
-    ):
-        await ProxyStartupEvent.initialize_scheduled_background_jobs(
-            general_settings={},
-            prisma_client=mock_prisma_client,
-            proxy_budget_rescheduler_min_time=1,
-            proxy_budget_rescheduler_max_time=2,
-            proxy_batch_write_at=5,
-            proxy_logging_obj=mock_proxy_logging,
-        )
+    scheduler = AsyncIOScheduler()
+    try:
+        with (
+            patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config),
+            patch("litellm.proxy.proxy_server.store_model_in_db", False),
+            patch("litellm.proxy.proxy_server.AsyncIOScheduler", return_value=scheduler),
+        ):
+            await ProxyStartupEvent.initialize_scheduled_background_jobs(
+                general_settings={},
+                prisma_client=mock_prisma_client,
+                proxy_budget_rescheduler_min_time=1,
+                proxy_budget_rescheduler_max_time=2,
+                proxy_batch_write_at=5,
+                proxy_logging_obj=mock_proxy_logging,
+            )
 
-    mock_prisma_client.attempt_db_reconnect.assert_awaited_once()
-    assert find_first.await_count == 2
-    mock_proxy_config.add_deployment.assert_awaited()
-    mock_proxy_config.init_mcp_servers_from_db.assert_not_called()
+        assert scheduler.get_job("add_deployment_job") is not None
+        mock_prisma_client.attempt_db_reconnect.assert_awaited_once()
+        assert find_first.await_count == 2
+        mock_proxy_config.add_deployment.assert_awaited()
+        mock_proxy_config.init_mcp_servers_from_db.assert_not_called()
+    finally:
+        scheduler.shutdown(wait=False)
 
 
 @pytest.mark.asyncio
@@ -1217,6 +1225,8 @@ async def test_initialize_scheduled_jobs_normalizes_string_general_settings(monk
     """
     import types
 
+    from apscheduler.schedulers.asyncio import AsyncIOScheduler
+
     monkeypatch.delenv("DISABLE_PRISMA_SCHEMA_UPDATE", raising=False)
     monkeypatch.delenv("STORE_MODEL_IN_DB", raising=False)
     from litellm.proxy.proxy_server import ProxyStartupEvent
@@ -1234,23 +1244,29 @@ async def test_initialize_scheduled_jobs_normalizes_string_general_settings(monk
     mock_proxy_logging.db_spend_update_writer = MagicMock()
     mock_proxy_config = _mock_scheduled_proxy_config()
 
-    with (
-        patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config),
-        patch("litellm.proxy.proxy_server.store_model_in_db", False),
-    ):
-        await ProxyStartupEvent.initialize_scheduled_background_jobs(
-            general_settings={},
-            prisma_client=mock_prisma_client,
-            proxy_budget_rescheduler_min_time=1,
-            proxy_budget_rescheduler_max_time=2,
-            proxy_batch_write_at=5,
-            proxy_logging_obj=mock_proxy_logging,
-        )
+    scheduler = AsyncIOScheduler()
+    try:
+        with (
+            patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config),
+            patch("litellm.proxy.proxy_server.store_model_in_db", False),
+            patch("litellm.proxy.proxy_server.AsyncIOScheduler", return_value=scheduler),
+        ):
+            await ProxyStartupEvent.initialize_scheduled_background_jobs(
+                general_settings={},
+                prisma_client=mock_prisma_client,
+                proxy_budget_rescheduler_min_time=1,
+                proxy_budget_rescheduler_max_time=2,
+                proxy_batch_write_at=5,
+                proxy_logging_obj=mock_proxy_logging,
+            )
 
-    assert find_first.await_count == 1
-    mock_prisma_client.attempt_db_reconnect.assert_not_awaited()
-    mock_proxy_config.add_deployment.assert_awaited()
-    mock_proxy_config.init_mcp_servers_from_db.assert_not_called()
+        assert scheduler.get_job("add_deployment_job") is not None
+        assert find_first.await_count == 1
+        mock_prisma_client.attempt_db_reconnect.assert_not_awaited()
+        mock_proxy_config.add_deployment.assert_awaited()
+        mock_proxy_config.init_mcp_servers_from_db.assert_not_called()
+    finally:
+        scheduler.shutdown(wait=False)
 
 
 @pytest.mark.asyncio
@@ -1261,6 +1277,8 @@ async def test_initialize_scheduled_jobs_ignores_malformed_string_general_settin
     """
     import types
 
+    from apscheduler.schedulers.asyncio import AsyncIOScheduler
+
     monkeypatch.delenv("DISABLE_PRISMA_SCHEMA_UPDATE", raising=False)
     monkeypatch.delenv("STORE_MODEL_IN_DB", raising=False)
     from litellm.proxy.proxy_server import ProxyStartupEvent
@@ -1278,26 +1296,32 @@ async def test_initialize_scheduled_jobs_ignores_malformed_string_general_settin
     mock_proxy_logging.db_spend_update_writer = MagicMock()
     mock_proxy_config = _mock_scheduled_proxy_config()
 
-    with (
-        patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config),
-        patch("litellm.proxy.proxy_server.store_model_in_db", False),
-        patch("litellm.proxy.proxy_server.verbose_proxy_logger") as mock_logger,
-    ):
-        await ProxyStartupEvent.initialize_scheduled_background_jobs(
-            general_settings={},
-            prisma_client=mock_prisma_client,
-            proxy_budget_rescheduler_min_time=1,
-            proxy_budget_rescheduler_max_time=2,
-            proxy_batch_write_at=5,
-            proxy_logging_obj=mock_proxy_logging,
-        )
+    scheduler = AsyncIOScheduler()
+    try:
+        with (
+            patch("litellm.proxy.proxy_server.proxy_config", mock_proxy_config),
+            patch("litellm.proxy.proxy_server.store_model_in_db", False),
+            patch("litellm.proxy.proxy_server.AsyncIOScheduler", return_value=scheduler),
+            patch("litellm.proxy.proxy_server.verbose_proxy_logger") as mock_logger,
+        ):
+            await ProxyStartupEvent.initialize_scheduled_background_jobs(
+                general_settings={},
+                prisma_client=mock_prisma_client,
+                proxy_budget_rescheduler_min_time=1,
+                proxy_budget_rescheduler_max_time=2,
+                proxy_batch_write_at=5,
+                proxy_logging_obj=mock_proxy_logging,
+            )
 
-    assert find_first.await_count == 1
-    assert not any(
-        "Failed to check DB for store_model_in_db" in str(call.args[0]) for call in mock_logger.debug.call_args_list
-    )
-    mock_proxy_config.add_deployment.assert_not_called()
-    mock_proxy_config.init_mcp_servers_from_db.assert_awaited_once()
+        assert scheduler.get_job("add_deployment_job") is None
+        assert find_first.await_count == 1
+        assert not any(
+            "Failed to check DB for store_model_in_db" in str(call.args[0]) for call in mock_logger.debug.call_args_list
+        )
+        mock_proxy_config.add_deployment.assert_not_called()
+        mock_proxy_config.init_mcp_servers_from_db.assert_awaited_once()
+    finally:
+        scheduler.shutdown(wait=False)
 
 
 @pytest.mark.asyncio