From 4f840b1679883544acadb961fa15c81f561b9bb5 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 27 Nov 2024 10:06:30 -0800 Subject: [PATCH] test_generate_and_update_key --- .../test_key_generate_prisma.py | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py index 26e3f0646..8d264ea75 100644 --- a/tests/proxy_unit_tests/test_key_generate_prisma.py +++ b/tests/proxy_unit_tests/test_key_generate_prisma.py @@ -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(