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

test(router): cover fusion authorization helper

Browse source
This commit is contained in:
moe-berri 2026-09-03 14:49:58 -07:00
parent 462cdd6990
commit 079065bb39
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
tests/test_litellm/test_fusion_router.py
View file

@ -686,14 +686,14 @@ async def test_proxy_fusion_authorizes_every_hidden_model(monkeypatch: pytest.Mo
authorize = AsyncMock(return_value=None)
monkeypatch.setattr(auth_checks, "can_key_call_resolved_model", authorize)
response = await router.acompletion(
model="fusion/test",
messages=[{"role": "user", "content": "Answer"}],
metadata={"user_api_key_auth": UserAPIKeyAuth(models=["*"])},
proxy_server_request={"body": {"model": "fusion/test"}},
await router._authorize_fusion_dependencies( # pyright: ignore[reportPrivateUsage]
fusion_router=router.fusion_routers["fusion/test"],
request_kwargs={
"metadata": {"user_api_key_auth": UserAPIKeyAuth(models=["*"])},
"proxy_server_request": {"body": {"model": "fusion/test"}},
},
)
assert isinstance(response, ModelResponse)
assert [call.kwargs["model"] for call in authorize.await_args_list] == ["outer", "panel-a", "analyst"]