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

test(ui): keep session log pagination coverage

Browse source
This commit is contained in:
Yufeng He 2026-05-29 15:13:00 +08:00 • committed by Ceder
parent 1cc359f711
commit 0acf4812d9
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
tests/test_litellm/proxy/spend_tracking/test_spend_management_endpoints.py
View file

@ -1313,9 +1313,9 @@ async def test_ui_view_session_spend_logs_pagination(client, monkeypatch):
# Endpoint uses raw SQL for pagination - verify params
assert session_id == "session-123"
assert page_size == 1
assert skip == 0 # page=1, page_size=1
assert skip == 1 # page=2, page_size=1
assert 'ORDER BY "startTime" DESC' in sql_query
return [mock_spend_logs[1]]
return [mock_spend_logs[0]]
class MockPrismaClient:
def __init__(self):
@ -1327,18 +1327,18 @@ async def test_ui_view_session_spend_logs_pagination(client, monkeypatch):
response = client.get(
"/spend/logs/session/ui",
params={"session_id": "session-123", "page": 1, "page_size": 1},
params={"session_id": "session-123", "page": 2, "page_size": 1},
headers={"Authorization": "Bearer sk-test"},
)
assert response.status_code == 200
data = response.json()
assert data["total"] == 2
assert data["page"] == 1
assert data["page"] == 2
assert data["page_size"] == 1
assert data["total_pages"] == 2
assert len(data["data"]) == 1
assert data["data"][0]["request_id"] == "req2"
assert data["data"][0]["request_id"] == "req1"
@pytest.mark.asyncio