From 290e8375158cb44766e2849eec65e353fa1bd8a9 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 1 Apr 2025 18:15:01 -0700 Subject: [PATCH] test_update_logs_with_spend_logs_url --- tests/proxy_unit_tests/test_key_generate_prisma.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py index ea837b717b..0400a71cea 100644 --- a/tests/proxy_unit_tests/test_key_generate_prisma.py +++ b/tests/proxy_unit_tests/test_key_generate_prisma.py @@ -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 )