From dbe1879707f6df3b3aefe8cc2bb3cca0500f83d2 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 3 Sep 2024 15:33:22 -0700 Subject: [PATCH] reset general settings post test --- litellm/tests/test_key_generate_prisma.py | 4 ++++ litellm/tests/test_user_api_key_auth.py | 4 ++++ 2 files changed, 8 insertions(+) diff --git a/litellm/tests/test_key_generate_prisma.py b/litellm/tests/test_key_generate_prisma.py index cd4def7657f..708025d1d6d 100644 --- a/litellm/tests/test_key_generate_prisma.py +++ b/litellm/tests/test_key_generate_prisma.py @@ -3111,6 +3111,8 @@ async def test_admin_only_routes(prisma_client): } from litellm.proxy import proxy_server + initial_general_settings = getattr(proxy_server, "general_settings") + setattr(proxy_server, "general_settings", general_settings) admin_user = await new_user( @@ -3156,3 +3158,5 @@ async def test_admin_only_routes(prisma_client): error_str = str(e.message) assert "Route" in error_str and "admin only route" in error_str pass + + setattr(proxy_server, "general_settings", initial_general_settings) diff --git a/litellm/tests/test_user_api_key_auth.py b/litellm/tests/test_user_api_key_auth.py index 5f94c5d238e..e7b01aa3fd7 100644 --- a/litellm/tests/test_user_api_key_auth.py +++ b/litellm/tests/test_user_api_key_auth.py @@ -268,6 +268,8 @@ async def test_auth_with_allowed_routes(route, should_raise_error): from litellm.proxy import proxy_server + initial_general_settings = getattr(proxy_server, "general_settings") + setattr(proxy_server, "master_key", "sk-1234") setattr(proxy_server, "general_settings", general_settings) @@ -285,3 +287,5 @@ async def test_auth_with_allowed_routes(route, should_raise_error): pass else: await user_api_key_auth(request=request, api_key="Bearer " + user_key) + + setattr(proxy_server, "general_settings", initial_general_settings)