scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-26 06:03:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 35

style: ruff-format the model-management tests

Browse source 
Generated-by: termag
This commit is contained in:
JackDanger 2026-10-18 08:39:35 +00:00 • committed by Jack Danger
parent a73e813496
commit 88fb59746e
2 changed files with 865 additions and 337 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1190
tests/unit/proxy/management_endpoints/test_model_management_endpoints.py
View file

File diff suppressed because it is too large Load diff

12
tests/unit/proxy/management_endpoints/test_r2_9_api_base_team_gate.py
View file

@ -44,9 +44,7 @@ async def test_team_admin_bring_your_own_key_move_allowed():
from litellm.proxy._types import LitellmUserRoles, UserAPIKeyAuth
stored = GenericLiteLLMParams(api_key="server-owned-key")
incoming = GenericLiteLLMParams(
api_base="https://my-team-proxy.example", api_key="my-team-key"
)
incoming = GenericLiteLLMParams(api_base="https://my-team-proxy.example", api_key="my-team-key")
model_params = MagicMock()
model_params.litellm_params = stored
@ -80,9 +78,7 @@ async def test_team_admin_wif_association_still_guarded():
model_params.litellm_params = stored
with (
patch.object(
mme, "effective_server_owned_wif_fields", AsyncMock(return_value=("oidc_token_url",))
),
patch.object(mme, "effective_server_owned_wif_fields", AsyncMock(return_value=("oidc_token_url",))),
):
try:
await ModelManagementAuthChecks._reject_non_admin_wif_write(
@ -96,8 +92,6 @@ async def test_team_admin_wif_association_still_guarded():
except Exception as guard_error:
# On this stack both the static-key harvest gate and the WIF
# rule can fire for the same shape; accept either interdiction.
assert ("api_key stays server-owned" in str(guard_error)) or (
"workload identity" in str(guard_error)
)
assert ("api_key stays server-owned" in str(guard_error)) or ("workload identity" in str(guard_error))
return
raise AssertionError("expected a server-owned-key interdiction to fire")