mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-08 03:08:45 +00:00
fix(tests): use AsyncMock for prisma find_unique in agent get-by-id test (#23122)
This commit is contained in:
parent
ca4d4a0188
commit
0bc1bd6871
1 changed files with 3 additions and 0 deletions
|
|
@ -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"}
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue