From 3c61c7fbb18c12e38a279646e5bb305b2bb56a59 Mon Sep 17 00:00:00 2001 From: Julio Quinteros Pro Date: Sun, 15 Feb 2026 20:43:18 -0300 Subject: [PATCH] fix(test): mock enterprise license check in JWT test The test test_jwt_non_admin_team_route_access was failing with: ``` AssertionError: assert 'Only proxy admin can be used to generate' in 'Authentication Error, JWT Auth is an enterprise only feature...' ``` Root cause: The test was hitting the enterprise license validation before reaching the proxy admin authorization check. In parallel execution with --dist=loadscope, environment variables like LITELLM_LICENSE can vary between workers or be unset, causing inconsistent test behavior. Solution: Mock the JWTAuthManager._is_jwt_auth_available method to return True, bypassing the license check. This allows the test to reach the actual authorization logic being tested (proxy admin check). This approach is more reliable than setting environment variables which can cause pollution between parallel tests. Fixes test failure exposed by PR #21277. Co-Authored-By: Claude Sonnet 4.5 --- tests/proxy_unit_tests/test_user_api_key_auth.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/tests/proxy_unit_tests/test_user_api_key_auth.py b/tests/proxy_unit_tests/test_user_api_key_auth.py index 72d13aadad3..702139643ee 100644 --- a/tests/proxy_unit_tests/test_user_api_key_auth.py +++ b/tests/proxy_unit_tests/test_user_api_key_auth.py @@ -1044,8 +1044,13 @@ async def test_jwt_non_admin_team_route_access(monkeypatch): litellm.proxy.proxy_server, "general_settings", {"enable_jwt_auth": True} ) - # Mock JWTAuthManager.auth_builder + # Mock enterprise license check and JWTAuthManager.auth_builder + # License check must be mocked to avoid environment variable pollution + # in parallel test execution with patch( + "litellm.proxy.auth.handle_jwt.JWTAuthManager._is_jwt_auth_available", + return_value=True, + ), patch( "litellm.proxy.auth.handle_jwt.JWTAuthManager.auth_builder", return_value=mock_jwt_response, ):