tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-02 18:21:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

test_update_logs_with_spend_logs_url

Browse Source
This commit is contained in:
Ishaan Jaff
2025-04-01 18:15:01 -07:00
parent 07fc5a72f1
commit 290e837515
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/proxy_unit_tests/test_key_generate_prisma.py
+5 -4
View File
@@ -2280,15 +2280,16 @@ def test_get_bearer_token():
result = _get_bearer_token(api_key)
assert result == "sk-1234", f"Expected 'valid_token', got '{result}'"
def test_update_logs_with_spend_logs_url(prisma_client):
@pytest.mark.asyncio
async def test_update_logs_with_spend_logs_url(prisma_client):
"""
Unit test for making sure spend logs list is still updated when url passed in
"""
from litellm.proxy.db.db_spend_update_writer import DBSpendUpdateWriter
db_spend_update_writer = DBSpendUpdateWriter()
payload = {"startTime": datetime.now(), "endTime": datetime.now()}
DBSpendUpdateWriter._set_spend_logs_payload(payload=payload, prisma_client=prisma_client)
await db_spend_update_writer._set_spend_logs_payload(payload=payload, prisma_client=prisma_client)
assert len(prisma_client.spend_log_transactions) > 0
@@ -2296,7 +2297,7 @@ def test_update_logs_with_spend_logs_url(prisma_client):
spend_logs_url = ""
payload = {"startTime": datetime.now(), "endTime": datetime.now()}
DBSpendUpdateWriter._set_spend_logs_payload(
await db_spend_update_writer._set_spend_logs_payload(
payload=payload, spend_logs_url=spend_logs_url, prisma_client=prisma_client
)