scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-25 01:02:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 80

test(vantage-export): cover upstream status resolution

Browse source 
Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
Andy Cheung 2026-05-18 08:00:32 -04:00
parent 4efa4817e9
commit 222b379207
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
tests/proxy_unit_tests/test_vantage_endpoints.py
View file

@ -10,6 +10,29 @@ from litellm.proxy.spend_tracking import vantage_endpoints
from litellm.types.proxy.vantage_endpoints import VantageExportRequest
@pytest.mark.parametrize(
("exception", "expected_status_code"),
[
(type("StatusCodeError", (Exception,), {"status_code": 422})(), 422),
(
type(
"ResponseStatusCodeError",
(Exception,),
{"response": httpx.Response(409)},
)(),
409,
),
(Exception("unexpected failure"), 500),
],
)
def test_get_upstream_status_code_uses_upstream_status_when_available(
exception, expected_status_code
):
assert (
vantage_endpoints._get_upstream_status_code(exception) == expected_status_code
)
@pytest.mark.asyncio
async def test_vantage_export_preserves_upstream_http_status(monkeypatch):
request = httpx.Request(