From 079065bb390048059fca396e64148f2f4c44b2bc Mon Sep 17 00:00:00 2001 From: moe-berri Date: Thu, 3 Sep 2026 14:49:58 -0700 Subject: [PATCH] test(router): cover fusion authorization helper --- tests/test_litellm/test_fusion_router.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/tests/test_litellm/test_fusion_router.py b/tests/test_litellm/test_fusion_router.py index 5c55be9420e..129bfc6ff8d 100644 --- a/tests/test_litellm/test_fusion_router.py +++ b/tests/test_litellm/test_fusion_router.py @@ -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"]