scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-10 22:41:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 82

fix: mock prisma_client in internal user get-agent-by-id test

Browse source 
Co-authored-by: Ishaan Jaff <ishaan-jaff@users.noreply.github.com>
This commit is contained in:
Cursor Agent 2026-02-27 18:53:28 +00:00
parent f0c22559bd
commit e0eda3d981
1 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_litellm/proxy/agent_endpoints/test_endpoints.py
View file

@ -294,9 +294,10 @@ class TestAgentRBACInternalUser:
self.mock_registry.get_agent_by_id = MagicMock(
return_value=_sample_agent_response()
)
resp = self.internal_client.get(
"/v1/agents/agent-123", headers={"Authorization": "Bearer k"}
)
with patch("litellm.proxy.proxy_server.prisma_client") as mock_prisma:
resp = self.internal_client.get(
"/v1/agents/agent-123", headers={"Authorization": "Bearer k"}
)
assert resp.status_code == 200
def test_should_block_internal_user_from_creating_agent(self):