scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

fix(test): make test_google_endpoint_routing resilient to module reloads (#24499)

Browse source 
Same root cause as test_end_user_jwt_auth: test_cors_config reloads
proxy_server module, so initialize() sets llm_router on the old module
while the test's patch target resolves to the new one where it's None.

Fix by syncing llm_router from the original module to the live
sys.modules entry after initialize().

Co-authored-by: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
Aarish Alam 2026-03-24 13:43:56 +05:30 • committed by GitHub
parent de7763fcfc
commit 92a6507d65
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
tests/proxy_unit_tests/test_google_endpoint_routing.py
View file

@ -81,9 +81,22 @@ async def test_google_generate_content_with_slashes_in_model_name(
try:
await initialize(config=config_fp)
with patch("litellm.proxy.proxy_server.llm_router.agenerate_content", new_callable=AsyncMock) as mock_agenerate_content:
# initialize() sets llm_router on the module object that existed at
# import time. Other tests (test_cors_config) do sys.modules.pop +
# re-import of litellm.proxy.proxy_server, creating a new module
# where llm_router is still None. google_generate_content does a
# lazy `from litellm.proxy.proxy_server import llm_router` which
# reads from the current sys.modules entry. Copy the router from
# the original module (where initialize set it) to the live module.
_init_globals = initialize.__globals__
_live_mod = sys.modules["litellm.proxy.proxy_server"]
if _live_mod.__dict__ is not _init_globals:
_live_mod.llm_router = _init_globals.get("llm_router")
router = _live_mod.llm_router
with patch.object(router, "agenerate_content", new_callable=AsyncMock) as mock_agenerate_content:
mock_agenerate_content.return_value = ModelResponse()
await google_generate_content(
request=mock_request,
model_name="bedrock/claude-sonnet-3.7",