scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

test: suppress TQ008 on new CLI session tests that stub proxy_server globals

Browse source 
Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
yassin 2026-08-25 20:13:13 +00:00
parent 9c031d208e
commit e57250f5f8
3 changed files with 21 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
tests/test_litellm/proxy/auth/test_user_api_key_auth.py
View file

@ -6780,9 +6780,9 @@ async def _auth_with_cli_token(cli_token: str, cache: DualCache):
mock_request.query_params = {}
with (
patch("litellm.proxy.proxy_server.master_key", "sk-master"),
patch("litellm.proxy.proxy_server.prisma_client", MagicMock()),
patch("litellm.proxy.proxy_server.user_api_key_cache", cache),
patch("litellm.proxy.proxy_server.master_key", "sk-master"), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.prisma_client", MagicMock()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.user_api_key_cache", cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
):
return await user_api_key_auth(request=mock_request, api_key=f"Bearer {cli_token}")

4
tests/test_litellm/proxy/management_endpoints/test_cli_session_endpoints.py
View file

@ -36,8 +36,8 @@ def _caller(role: LitellmUserRoles, user_id: str = "admin-1") -> UserAPIKeyAuth:
def _proxy(table: FakeCLISessionTable):
return (
patch("litellm.proxy.proxy_server.prisma_client", FakePrismaClient(table)),
patch("litellm.proxy.proxy_server.user_api_key_cache", DualCache()),
patch("litellm.proxy.proxy_server.prisma_client", FakePrismaClient(table)), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.user_api_key_cache", DualCache()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
)

32
tests/test_litellm/proxy/management_endpoints/test_ui_sso.py
View file

@ -3524,7 +3524,7 @@ class TestCLIKeyRegenerationFlow:
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value=mock_jwt_token,
@ -3677,7 +3677,7 @@ class TestCLIKeyRegenerationFlow:
with (
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value="minted-token",
@ -3742,7 +3742,7 @@ class TestCLIKeyRegenerationFlow:
with (
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value="minted-token",
@ -3785,7 +3785,7 @@ class TestCLIKeyRegenerationFlow:
with (
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value="minted-token",
@ -3839,7 +3839,7 @@ class TestCLIKeyRegenerationFlow:
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value=mock_jwt_token,
@ -3896,7 +3896,7 @@ class TestCLIKeyRegenerationFlow:
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value=mock_jwt_token,
@ -8126,7 +8126,7 @@ async def test_cli_poll_key_tolerates_missing_user_row():
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()),
patch("litellm.proxy.proxy_server.prisma_client", _cli_session_registry_prisma()), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch(
"litellm.proxy.auth.auth_checks.ExperimentalUIJWTToken.get_cli_jwt_auth_token",
return_value=mock_jwt_token,
@ -8690,9 +8690,9 @@ async def test_cli_poll_key_registers_the_session_it_mints(monkeypatch):
prisma = _cli_session_registry_prisma()
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", prisma),
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.prisma_client", prisma), # test-quality-ok: auth reads proxy_server module globals, no injection seam
):
result = await cli_poll_key(
key_id="cli-registered-session",
@ -8730,9 +8730,9 @@ async def test_cli_poll_key_withholds_the_credential_when_registration_fails(mon
prisma.db.litellm_clisessiontable.create = AsyncMock(side_effect=Exception("registry write failed"))
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", prisma),
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.prisma_client", prisma), # test-quality-ok: auth reads proxy_server module globals, no injection seam
):
with pytest.raises(HTTPException) as exc_info:
await cli_poll_key(
@ -8762,9 +8762,9 @@ async def test_cli_poll_key_registers_a_distinct_session_per_login(monkeypatch):
prisma = _cli_session_registry_prisma()
with (
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache),
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache),
patch("litellm.proxy.proxy_server.prisma_client", prisma),
patch("litellm.proxy.proxy_server.user_api_key_cache", mock_cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.cli_sso_session_cache", mock_cache), # test-quality-ok: auth reads proxy_server module globals, no injection seam
patch("litellm.proxy.proxy_server.prisma_client", prisma), # test-quality-ok: auth reads proxy_server module globals, no injection seam
):
for key_id in ("cli-login-session-alpha", "cli-login-session-bravo"):
await cli_poll_key(key_id=key_id, team_id=None, x_litellm_cli_poll_secret="poll-secret")