mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-03 02:22:24 +00:00
39e31958f8
* test(proxy): move auth, hooks, policy_engine and client tests into tests/unit/proxy Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): stub HIBP through respx by disabling the aiohttp transport Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): share the httpx transport fixture across proxy unit tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): restore proxy globals without a missing-value sentinel Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): package moved dirs and stub the login breach check at the HTTP boundary Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(proxy): isolate the mcp server manager per test Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- Co-authored-by: yuneng <yuneng@berri.ai> Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
95 lines
2.5 KiB
Python
95 lines
2.5 KiB
Python
from __future__ import annotations
|
|
|
|
import pytest
|
|
from pydantic import ValidationError
|
|
|
|
from litellm.proxy.auth.auth_method import AuthMethod
|
|
from litellm.proxy.auth.resolvers.models import (
|
|
EndUserIdentity,
|
|
Principal,
|
|
PrincipalType,
|
|
ProjectIdentity,
|
|
TeamIdentity,
|
|
UserIdentity,
|
|
)
|
|
from litellm.proxy.auth.roles import Role, TeamRole
|
|
|
|
|
|
def _principal() -> Principal:
|
|
return Principal(
|
|
principal_type=PrincipalType.HUMAN,
|
|
subject="u1",
|
|
auth_method=AuthMethod.OIDC,
|
|
)
|
|
|
|
|
|
def test_principal_is_frozen():
|
|
principal = _principal()
|
|
with pytest.raises(ValidationError):
|
|
principal.subject = "mutated"
|
|
|
|
|
|
def test_principal_defaults_are_independent_instances():
|
|
a = _principal()
|
|
b = _principal()
|
|
assert a.teams == [] and a.scopes == [] and a.audience == []
|
|
assert a.teams is not b.teams
|
|
|
|
|
|
def test_principal_requires_identity_core_fields():
|
|
with pytest.raises(ValidationError):
|
|
Principal(subject="u1") # missing principal_type + auth_method
|
|
|
|
|
|
def test_principal_roles_are_validated_against_role_enum():
|
|
principal = Principal(
|
|
principal_type=PrincipalType.HUMAN,
|
|
subject="u1",
|
|
auth_method=AuthMethod.OIDC,
|
|
roles=["org_admin"],
|
|
)
|
|
assert principal.roles == [Role.ORG_ADMIN]
|
|
assert isinstance(principal.roles[0], Role)
|
|
|
|
with pytest.raises(ValidationError):
|
|
Principal(
|
|
principal_type=PrincipalType.HUMAN,
|
|
subject="u1",
|
|
auth_method=AuthMethod.OIDC,
|
|
roles=["not_a_real_role"],
|
|
)
|
|
|
|
|
|
def test_principal_default_network_and_collections():
|
|
principal = Principal(
|
|
principal_type=PrincipalType.SERVICE_ACCOUNT,
|
|
subject="svc",
|
|
auth_method=AuthMethod.MUTUAL_TLS,
|
|
)
|
|
assert principal.teams == []
|
|
assert principal.scopes == []
|
|
assert principal.project is None
|
|
assert principal.end_user is None
|
|
assert principal.network.client_ip is None
|
|
assert principal.network.via_trusted_proxy is False
|
|
|
|
|
|
def test_team_identity_defaults_to_member_role():
|
|
team = TeamIdentity(id="g1")
|
|
assert team.role == TeamRole.MEMBER
|
|
|
|
|
|
def test_user_identity_optional_fields_default_none():
|
|
user = UserIdentity(id="u1")
|
|
assert user.email is None
|
|
assert user.external_id is None
|
|
|
|
|
|
def test_project_identity_name_is_optional():
|
|
assert ProjectIdentity(id="p1").name is None
|
|
assert ProjectIdentity(id="p1", name="Acme").name == "Acme"
|
|
|
|
|
|
def test_end_user_identity_requires_id():
|
|
with pytest.raises(ValidationError):
|
|
EndUserIdentity()
|