phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 19:54:13 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

reset general settings post test

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-03 15:33:22 -07:00
parent daaca2760e
commit 7a367bddc4
2 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_key_generate_prisma.py
View file

@ -3111,6 +3111,8 @@ async def test_admin_only_routes(prisma_client):
} }
from litellm.proxy import proxy_server from litellm.proxy import proxy_server
initial_general_settings = getattr(proxy_server, "general_settings")
setattr(proxy_server, "general_settings", general_settings) setattr(proxy_server, "general_settings", general_settings)
admin_user = await new_user( admin_user = await new_user(
@ -3156,3 +3158,5 @@ async def test_admin_only_routes(prisma_client):
error_str = str(e.message) error_str = str(e.message)
assert "Route" in error_str and "admin only route" in error_str assert "Route" in error_str and "admin only route" in error_str
pass pass
setattr(proxy_server, "general_settings", initial_general_settings)

4
litellm/tests/test_user_api_key_auth.py
View file

@ -268,6 +268,8 @@ async def test_auth_with_allowed_routes(route, should_raise_error):
from litellm.proxy import proxy_server from litellm.proxy import proxy_server
initial_general_settings = getattr(proxy_server, "general_settings")
setattr(proxy_server, "master_key", "sk-1234") setattr(proxy_server, "master_key", "sk-1234")
setattr(proxy_server, "general_settings", general_settings) setattr(proxy_server, "general_settings", general_settings)
@ -285,3 +287,5 @@ async def test_auth_with_allowed_routes(route, should_raise_error):
pass pass
else: else:
await user_api_key_auth(request=request, api_key="Bearer " + user_key) await user_api_key_auth(request=request, api_key="Bearer " + user_key)
setattr(proxy_server, "general_settings", initial_general_settings)