fix(test): mock execute_raw in test_create_team_member_add_team_admin

The test needs to mock the new atomic execute_raw call introduced by the
race condition fix, since it runs without a connected Prisma client

Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com>
This commit is contained in:
Krrish Dholakia 2026-07-03 01:17:38 +00:00
parent b6514c25fc
commit d383149ab0

View file

@ -1420,6 +1420,10 @@ async def test_create_team_member_add_team_admin(
"litellm.proxy.proxy_server.prisma_client.get_data",
new=AsyncMock(return_value=[]),
) as mock_get_data,
patch(
"litellm.proxy.proxy_server.prisma_client.db.execute_raw",
new_callable=AsyncMock,
),
):
mock_client = AsyncMock(
return_value=LiteLLM_UserTable(