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

test(spend): scope unauth spend-row probe to the request under test

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
kerry 2026-10-02 20:11:43 +00:00
parent f08fba76bf
commit 7d2b062059
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
tests/integration/spend/test_spend_log_request_tags.py
View file

@ -701,6 +701,9 @@ def test_unauthenticated_request_writes_no_spend_row(gateway: Gateway, tmp_path:
candidate.scenario() as scenario,
):
before: Final = _spend_count()
anonymous_before: Final = len(
read_rows("SELECT request_tags FROM \"LiteLLM_SpendLogs\" WHERE api_key IS NULL OR api_key=''", ())
)
response: Final = candidate.client.post(
route,
json={
@ -713,13 +716,13 @@ def test_unauthenticated_request_writes_no_spend_row(gateway: Gateway, tmp_path:
assert response.status_code == 401, response.text
anonymous: Final = eventually(
lambda: read_rows(
"SELECT request_tags FROM \"LiteLLM_SpendLogs\" WHERE api_key IS NULL OR api_key=''",
'SELECT request_tags FROM "LiteLLM_SpendLogs" WHERE api_key IS NULL OR api_key=\'\' ORDER BY "startTime" DESC',
(),
),
lambda rows: len(rows) == 1,
lambda rows: len(rows) == anonymous_before + 1,
seconds=70,
)
assert [tags_of(row) for row in anonymous] == [[]]
assert tags_of(anonymous[0]) == []
key: Final = scenario.key()
control: Final = candidate.request(
"POST",