diff --git a/tests/common.py b/tests/common.py index d92347b..6a30287 100644 --- a/tests/common.py +++ b/tests/common.py @@ -11,8 +11,8 @@ def sync_client() -> Client: return Client( token=get_env_var("TEST_TOKEN"), host=get_env_var("TEST_API_URL"), - # This will retry after [0.0, 0.5, 1.0, 2.0, 4.0] seconds - total_retries=5, + # This will retry after [0.0, 0.25, 0.5, 1.0, 2.0] seconds + total_retries=int(os.environ.get("TEST_API_RETRIES", "5")), ) @@ -21,8 +21,8 @@ async def async_client() -> AsyncIterable[AsyncClient]: async with AsyncClient( token=get_env_var("TEST_TOKEN"), host=get_env_var("TEST_API_URL"), - # This will retry after [0.0, 0.5, 1.0, 2.0, 4.0] seconds - total_retries=5, + # This will retry after [0.0, 0.25, 0.5, 1.0, 2.0] seconds + total_retries=int(os.environ.get("TEST_API_RETRIES", "5")), ) as client: yield client