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

test(proxy): 5xx (http + proxy) propagation

Browse source
This commit is contained in:
Ishaan Jaff 2026-06-24 13:40:27 -07:00
parent f13272b216
commit acbd5991c8
No known key found for this signature in database
1 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
tests/test_litellm/proxy/auth/test_internal_auth_endpoints.py
View file

@ -167,3 +167,38 @@ async def test_verify_key_401_on_http_exception(monkeypatch):
assert exc_info.value.status_code == 401
# Internals must not leak.
assert exc_info.value.detail == "invalid api key"
@pytest.mark.asyncio
async def test_verify_key_propagates_http_5xx(monkeypatch):
# A 5xx (e.g. DB outage) must NOT be masked as 401 — operators need the real error.
async def fake_user_api_key_auth(request, api_key):
raise HTTPException(status_code=503, detail="db unavailable")
monkeypatch.setattr(
"litellm.proxy.auth.user_api_key_auth.user_api_key_auth",
fake_user_api_key_auth,
)
body = VerifyKeyRequest(api_key="sk-key", route="/v1/realtime")
with pytest.raises(HTTPException) as exc_info:
await verify_key(body=body)
assert exc_info.value.status_code == 503
@pytest.mark.asyncio
async def test_verify_key_propagates_proxy_5xx(monkeypatch):
# A ProxyException carrying a 5xx code propagates too (not converted to 401).
async def fake_user_api_key_auth(request, api_key):
raise ProxyException(
message="internal", type="internal_error", param=None, code="500"
)
monkeypatch.setattr(
"litellm.proxy.auth.user_api_key_auth.user_api_key_auth",
fake_user_api_key_auth,
)
body = VerifyKeyRequest(api_key="sk-key", route="/v1/realtime")
with pytest.raises(ProxyException):
await verify_key(body=body)