phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

test_generate_and_update_key

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-27 10:06:30 -08:00
parent 5fcb4130a0
commit 4f840b1679
1 changed files with 16 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

17
tests/proxy_unit_tests/test_key_generate_prisma.py
View file

@ -1349,7 +1349,13 @@ def test_generate_and_update_key(prisma_client):
"days between now and budget_reset_at",
(budget_reset_at - current_time).days,
)
assert (budget_reset_at - current_time).days >= 29 # around 1 month
# assert budget_reset_at is 30 days from now
assert (
abs(
(budget_reset_at - current_time).total_seconds() - 30 * 24 * 60 * 60
)
<= 10
)
# cleanup - delete key
delete_key_request = KeyRequest(keys=[generated_key])
@ -2630,6 +2636,15 @@ async def test_create_update_team(prisma_client):
_updated_info["budget_reset_at"], datetime.datetime
)
# budget_reset_at should be 2 days from now
budget_reset_at = _updated_info["budget_reset_at"].replace(tzinfo=timezone.utc)
current_time = datetime.datetime.now(timezone.utc)
# assert budget_reset_at is 2 days from now
assert (
abs((budget_reset_at - current_time).total_seconds() - 2 * 24 * 60 * 60) <= 10
)
# now hit team_info
try:
response = await team_info(