From b6c8b9123c018bcb855cdad3febf8b5008bb6dd2 Mon Sep 17 00:00:00 2001 From: Yassin Kortam Date: Mon, 8 Jun 2026 15:46:36 -0700 Subject: [PATCH] test(identity): assert full UserAPIKeyAuth parity across resolve_identity wiring --- .../auth/test_user_api_key_auth_identity_wiring.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/tests/test_litellm/proxy/auth/test_user_api_key_auth_identity_wiring.py b/tests/test_litellm/proxy/auth/test_user_api_key_auth_identity_wiring.py index ac8362709d4..02a281caa0a 100644 --- a/tests/test_litellm/proxy/auth/test_user_api_key_auth_identity_wiring.py +++ b/tests/test_litellm/proxy/auth/test_user_api_key_auth_identity_wiring.py @@ -160,10 +160,8 @@ async def test_parity_uak_for_api_key_request_matches_pre_integration(): legacy = await _run_master_builder(request_legacy, request_data) assert request_legacy.state.identity_context is None - assert wired.user_role == legacy.user_role == LitellmUserRoles.PROXY_ADMIN - assert wired.user_id == legacy.user_id - assert wired.api_key == legacy.api_key - assert wired.end_user_id == legacy.end_user_id + assert wired.user_role == LitellmUserRoles.PROXY_ADMIN + assert wired.model_dump() == legacy.model_dump() def _jwt_result(): @@ -240,6 +238,6 @@ async def test_parity_uak_for_jwt_request_matches_pre_integration(): assert request_wired.state.identity_context.principal.kind == "jwt" assert request_legacy.state.identity_context is None - assert wired.user_id == legacy.user_id == "jwt-human-user" - assert wired.team_id == legacy.team_id == "jwt-team" - assert wired.user_role == legacy.user_role == LitellmUserRoles.PROXY_ADMIN + assert wired.user_id == "jwt-human-user" + assert wired.team_id == "jwt-team" + assert wired.model_dump() == legacy.model_dump()