From 1454dbf482cb55977a3b245ac61830c66ed4d34c Mon Sep 17 00:00:00 2001 From: Mrinal Chanshetty Date: Fri, 25 Sep 2026 19:57:44 -0400 Subject: [PATCH] test(jwt): assert on race-loser's returned key, not only mocks (TQ002) Co-Authored-By: Claude Opus 5.5 --- tests/test_litellm/proxy/auth/test_user_api_key_auth.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/test_litellm/proxy/auth/test_user_api_key_auth.py b/tests/test_litellm/proxy/auth/test_user_api_key_auth.py index 3628ea59b46..6ea4ad21584 100644 --- a/tests/test_litellm/proxy/auth/test_user_api_key_auth.py +++ b/tests/test_litellm/proxy/auth/test_user_api_key_auth.py @@ -2296,8 +2296,12 @@ async def test_auto_register_map_existing_key_race_loser_keeps_reused_key(): return_value="winner-hash", ), ): - await _auto_register_jwt_mapping(**_auto_register_kwargs(prisma_client, user_api_key_cache, jwt_handler)) + result = await _auto_register_jwt_mapping( + **_auto_register_kwargs(prisma_client, user_api_key_cache, jwt_handler) + ) + assert result is not None + assert result.org_id == "key-own-org" prisma_client.db.litellm_verificationtoken.delete.assert_not_awaited() assert user_api_key_cache.async_set_cache.await_args.kwargs["value"] == "winner-hash"