From 2fb575b630baa1a60c78911c0f9fe0725d9b1c32 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Sat, 8 Mar 2025 17:21:32 -0800 Subject: [PATCH] fix test_list_key_helper_team_filtering --- tests/proxy_admin_ui_tests/test_key_management.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/proxy_admin_ui_tests/test_key_management.py b/tests/proxy_admin_ui_tests/test_key_management.py index 9d6c24db0eb..dfb7fbba088 100644 --- a/tests/proxy_admin_ui_tests/test_key_management.py +++ b/tests/proxy_admin_ui_tests/test_key_management.py @@ -1029,7 +1029,9 @@ async def test_list_key_helper_team_filtering(prisma_client): # Test 1: Get all keys with pagination (exclude litellm-dashboard) all_keys = [] page = 1 - while True: + max_pages_to_check = 3 # Only check the first 3 pages + + while page <= max_pages_to_check: result = await _list_key_helper( prisma_client=prisma_client, size=100, @@ -1043,7 +1045,7 @@ async def test_list_key_helper_team_filtering(prisma_client): all_keys.extend(result["keys"]) - if page >= result["total_pages"]: + if page >= result["total_pages"] or page >= max_pages_to_check: break page += 1