fix(tests): use AsyncMock for prisma find_unique in agent get-by-id test (#23122)

This commit is contained in:
Joe Reyna 2026-03-09 07:13:50 -07:00 • committed by GitHub
parent ca4d4a0188
commit 0bc1bd6871
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -295,6 +295,9 @@ class TestAgentRBACInternalUser:
return_value=_sample_agent_response()
)
with patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma:
mock_prisma.db.litellm_agentstable.find_unique = AsyncMock(
return_value=None
)
resp = self.internal_client.get(
"/v1/agents/agent-123", headers={"Authorization": "Bearer k"}
)