From a0022410fa8001671291363810133d5654dc15d6 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Wed, 31 Jan 2024 10:05:35 -0800 Subject: [PATCH] test(test_users.py): add more reattempts and delays --- tests/test_users.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/tests/test_users.py b/tests/test_users.py index d7e2b6a35..898274467 100644 --- a/tests/test_users.py +++ b/tests/test_users.py @@ -123,11 +123,19 @@ async def test_users_budgets_reset(): session=session, get_user=get_user, call_user=key ) reset_at_init_value = user_info["user_info"]["budget_reset_at"] - await asyncio.sleep(15) - user_info = await get_user_info( - session=session, get_user=get_user, call_user=key - ) - reset_at_new_value = user_info["user_info"]["budget_reset_at"] + i = 0 + reset_at_new_value = None + while i < 3: + await asyncio.sleep(15) + user_info = await get_user_info( + session=session, get_user=get_user, call_user=key + ) + 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