From f41fb256bad6d05eafbde2510c8d4e58fa8bc417 Mon Sep 17 00:00:00 2001 From: Joshua Valluru <326636767+joshua-berri@users.noreply.github.com> Date: Sat, 26 Sep 2026 17:51:43 -0700 Subject: [PATCH] test(mcp): isolate reload cooldown from shared discovery --- .../_experimental/mcp_server/test_mcp_server.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py index 79f401cc1bb..03bb07d374f 100644 --- a/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py +++ b/tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_server.py @@ -6393,16 +6393,13 @@ class TestEnsureUpstreamInitializeInstructionsCached: @pytest.mark.asyncio async def test_reload_resets_probe_cooldown(self): """load_servers_from_config clears the negative-cache map so reloads re-probe.""" - from litellm.proxy._experimental.mcp_server.mcp_server_manager import ( - global_mcp_server_manager, - ) + from litellm.proxy._experimental.mcp_server.mcp_server_manager import MCPServerManager + + manager: Final = MCPServerManager() + manager._upstream_initialize_instructions_probed_at["reload-target"] = 1.0 + await manager.load_servers_from_config({}) + assert "reload-target" not in manager._upstream_initialize_instructions_probed_at - global_mcp_server_manager._upstream_initialize_instructions_probed_at["reload-target"] = 1.0 - try: - await global_mcp_server_manager.load_servers_from_config({}) - assert "reload-target" not in global_mcp_server_manager._upstream_initialize_instructions_probed_at - finally: - global_mcp_server_manager._upstream_initialize_instructions_probed_at.pop("reload-target", None) class TestGatewayCreateInitializationOptions: