scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-15 09:12:18 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 311

test: preserve captured MCP toolset update fields

Browse source
This commit is contained in:
Yuneng Jiang 2026-09-08 18:03:58 -07:00
parent 3dac89d2d0
commit 21fc69503a
No known key found for this signature in database
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/test_litellm/proxy/_experimental/mcp_server/test_mcp_partial_update.py
View file

@ -878,8 +878,8 @@ async def _run_toolset_update(payload: dict) -> dict:
mock_prisma = _mock_toolset_prisma()
await update_mcp_toolset(mock_prisma, UpdateMCPToolsetRequest.model_validate(payload), "test-user")
written = dict(mock_prisma.db.litellm_mcptoolsettable.update.call_args[1]["data"])
assert written.pop("updated_by") == "test-user"
return written
assert written["updated_by"] == "test-user"
return {name: value for name, value in written.items() if name != "updated_by"}
@pytest.mark.asyncio