scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

test(proxy): mock _cached_get_model_group_info in team model listing tests

Browse source
This commit is contained in:
Devin AI 2026-07-17 04:17:09 +00:00
parent 85520c1b29
commit 87143286f4
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/proxy_server/test_team_model_name_translation.py
View file

@ -802,7 +802,7 @@ async def test_v1_models_translates_team_model_with_metadata(monkeypatch):
router.get_fully_blocked_model_names.return_value = set()
router.model_list = [team_dep]
router.get_model_list.return_value = [team_dep]
router.get_model_group_info.return_value = None
router._cached_get_model_group_info.return_value = None
monkeypatch.setattr(ps, "llm_router", router)
monkeypatch.setattr(ps, "user_model", None)
@ -849,7 +849,7 @@ async def test_v1_models_metadata_fallbacks_use_internal_routing_key(monkeypatch
router.get_model_list.return_value = [team_dep]
# Fallbacks are keyed on the internal routing name, as the router stores them.
router.fallbacks = [{"model_name_teamX_uuid9": ["gpt-4o-backup"]}]
router.get_model_group_info.return_value = None
router._cached_get_model_group_info.return_value = None
monkeypatch.setattr(ps, "llm_router", router)
monkeypatch.setattr(ps, "user_model", None)
@ -907,7 +907,7 @@ async def test_v1_models_metadata_does_not_leak_other_team_fallbacks(monkeypatch
{"model_name_teamX_uuid9": ["teamX-backup"]},
{"model_name_teamY_uuidZ": ["teamY-backup"]},
]
router.get_model_group_info.return_value = None
router._cached_get_model_group_info.return_value = None
monkeypatch.setattr(ps, "llm_router", router)
monkeypatch.setattr(ps, "user_model", None)