reset general settings post test

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

View file

@ -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)