scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-08 03:08:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 115

Assert reservation payload in unpriced-settle tests.

Browse source 
The new tests only inspected mocks, which the test-quality gate treats as mock-echo.
This commit is contained in:
Gyanu 2026-08-23 14:41:39 +05:30
parent 3122e13e6c
commit 44b378c416
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/test_litellm/proxy/hooks/test_proxy_track_cost_callback.py
View file

@ -438,6 +438,9 @@ async def test_track_cost_callback_settles_budget_reservation_when_response_cost
)
mock_update_database.assert_not_called()
mock_proxy_logging.failed_tracking_alert.assert_called()
# The hold is converted, not dropped: later traffic still sees the reserved cost.
assert budget_reservation["reserved_cost"] == 0.5
assert budget_reservation.get("finalized") is not True
@pytest.mark.asyncio
@ -497,6 +500,8 @@ async def test_track_cost_callback_settles_async_stream_when_response_cost_missi
)
mock_update_database.assert_not_called()
mock_proxy_logging.failed_tracking_alert.assert_called()
assert budget_reservation["reserved_cost"] == 0.5
assert budget_reservation.get("finalized") is not True
@pytest.mark.asyncio
@ -592,6 +597,9 @@ async def test_track_cost_callback_releases_budget_reservation_for_non_model_cal
mock_release_budget_reservation.assert_awaited_once_with(
budget_reservation=budget_reservation,
)
# Health checks refund the hold; they must not stamp it settled.
assert budget_reservation.get("finalized") is not True
assert budget_reservation["reserved_cost"] == 0.5
def test_get_budget_reservation_from_metadata_handles_dict_auth_object():