scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-06 08:16:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 78

test_call_with_end_user_over_budget

Browse source
This commit is contained in:
Ishaan Jaffer 2025-11-13 16:26:02 -08:00
parent 32885087c2
commit ee8b1cfabc
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
tests/proxy_unit_tests/test_key_generate_prisma.py
View file

@ -132,6 +132,11 @@ def prisma_client():
### add connection pool + pool timeout args
params = {"connection_limit": 100, "pool_timeout": 60}
database_url = os.getenv("DATABASE_URL")
# If DATABASE_URL is not set, use a default test database URL
if not database_url:
database_url = "postgresql://postgres:postgres@localhost:5432/circle_test"
modified_url = append_query_params(database_url, params)
os.environ["DATABASE_URL"] = modified_url
@ -666,7 +671,8 @@ def test_call_with_end_user_over_budget(prisma_client):
except Exception as e:
print(f"raised error: {e}, traceback: {traceback.format_exc()}")
error_detail = e.message
assert "Budget has been exceeded! Current" in error_detail
assert "ExceededBudget: End User=" in error_detail
assert "over budget" in error_detail
assert isinstance(e, ProxyException)
assert e.type == ProxyErrorTypes.budget_exceeded
print(vars(e))