phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 10:44:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test(test_users.py): add more reattempts and delays

Browse source
This commit is contained in:
Krrish Dholakia 2024-01-31 10:05:35 -08:00
parent eb9eca6f37
commit a0022410fa
1 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_users.py
View file

@ -123,11 +123,19 @@ async def test_users_budgets_reset():
session=session, get_user=get_user, call_user=key session=session, get_user=get_user, call_user=key
) )
reset_at_init_value = user_info["user_info"]["budget_reset_at"] reset_at_init_value = user_info["user_info"]["budget_reset_at"]
i = 0
reset_at_new_value = None
while i < 3:
await asyncio.sleep(15) await asyncio.sleep(15)
user_info = await get_user_info( user_info = await get_user_info(
session=session, get_user=get_user, call_user=key session=session, get_user=get_user, call_user=key
) )
reset_at_new_value = user_info["user_info"]["budget_reset_at"] reset_at_new_value = user_info["user_info"]["budget_reset_at"]
try:
assert reset_at_init_value != reset_at_new_value
break
except:
i + 1
assert reset_at_init_value != reset_at_new_value assert reset_at_init_value != reset_at_new_value