scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-12 08:42:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 207

test(proxy): keep only the raising call inside pytest.raises

Browse source
This commit is contained in:
yucheng 2026-11-02 14:44:12 -08:00
parent a8dc662b93
commit 4ec3750fb9
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
tests/unit/proxy/pass_through_endpoints/test_pass_through_endpoints.py
View file

@ -532,15 +532,15 @@ async def test_guardrail_proxy_exception_carries_the_guardrails_that_ran():
patch( # test-quality-ok: proxy_logging_obj is a proxy_server module global read inside pass_through_request
"litellm.proxy.proxy_server.proxy_logging_obj", ProxyLogging(user_api_key_cache=DualCache())
),
pytest.raises(ProxyException) as exc,
):
litellm_patch.setattr(litellm, "callbacks", [guardrail])
await pass_through_request(
request=mock_request,
target="http://test.com/v1/chat/completions",
custom_headers={},
user_api_key_dict=UserAPIKeyAuth(api_key="sk-test"),
)
with pytest.raises(ProxyException) as exc:
await pass_through_request(
request=mock_request,
target="http://test.com/v1/chat/completions",
custom_headers={},
user_api_key_dict=UserAPIKeyAuth(api_key="sk-test"),
)
assert exc.value.headers["x-litellm-applied-guardrails"] == "passed-guard"