diff --git a/tests/proxy_unit_tests/test_jwt.py b/tests/proxy_unit_tests/test_jwt.py index f64ab888e2..964d3f60a0 100644 --- a/tests/proxy_unit_tests/test_jwt.py +++ b/tests/proxy_unit_tests/test_jwt.py @@ -418,6 +418,7 @@ async def test_team_token_output(prisma_client, audience, monkeypatch): ## 1. INITIAL TEAM CALL - should fail # use generated key to auth in + setattr(litellm.proxy.proxy_server, "premium_user", True) setattr( litellm.proxy.proxy_server, "general_settings", @@ -840,6 +841,7 @@ async def test_allowed_routes_admin( ## 1. INITIAL TEAM CALL - should fail # use generated key to auth in + setattr(litellm.proxy.proxy_server, "premium_user", True) setattr( litellm.proxy.proxy_server, "general_settings", @@ -1014,6 +1016,7 @@ async def test_allow_access_by_email( ## 1. INITIAL TEAM CALL - should fail # use generated key to auth in + setattr(litellm.proxy.proxy_server, "premium_user", True) setattr( litellm.proxy.proxy_server, "general_settings", @@ -1175,6 +1178,7 @@ async def test_end_user_jwt_auth(monkeypatch): ), ) + setattr(litellm.proxy.proxy_server, "premium_user", True) setattr( litellm.proxy.proxy_server, "general_settings",