phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-24 18:24:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix Unit tests for test_mcp_utils.py

Browse source
This commit is contained in:
vinodjayachandran 2025-04-24 12:44:28 +05:30
parent c2b21703d4
commit f2d42566fd
1 changed files with 5 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/litellm/experimental_mcp_client/test_mcp_utils.py
View file

@ -141,7 +141,11 @@ def test_update_mcp_servers_file_success(mock_fetch, mock_json_dump, mock_file):
update_mcp_servers_file("test_output.json")
mock_file.assert_called_once_with("test_output.json", 'w')
# Check if the file was opened with a filename containing "test_output.json"
mock_file.assert_called()
args, _ = mock_file.call_args
assert "test_output.json" in args[0] # Ensure the filename contains the substring
mock_json_dump.assert_called_once_with(mock_servers, mock_file(), indent=2)
@patch('litellm.experimental_mcp_client.mcp_utils.fetch_mcp_servers')