andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 08:12:39 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

fix: Correct test mocks for MCP progress and headers tests

Browse Source 
- Fix event bus API: use register_handler instead of register
- Fix mock import paths: patch mcp.client.streamable_http.streamablehttp_client and mcp.ClientSession at their actual import locations

Co-Authored-By: João <joao@crewai.com>
This commit is contained in:
Devin AI
2025-10-26 09:52:58 +00:00
parent 873d501401
commit 99418b1160
1 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
lib/crewai/tests/tools/test_mcp_tool_wrapper.py
View File

@@ -119,8 +119,8 @@ class TestMCPToolWrapperProgress:
mock_result = Mock() mock_result = Mock()
mock_result.content = [Mock(text="Test result")] mock_result.content = [Mock(text="Test result")]
with patch("crewai.tools.mcp_tool_wrapper.streamablehttp_client") as mock_client, \ with patch("mcp.client.streamable_http.streamablehttp_client") as mock_client, \
patch("crewai.tools.mcp_tool_wrapper.ClientSession") as mock_session_class: patch("mcp.ClientSession") as mock_session_class:
mock_session = AsyncMock() mock_session = AsyncMock()
mock_session.initialize = AsyncMock() mock_session.initialize = AsyncMock()
@@ -148,7 +148,7 @@ class TestMCPToolWrapperProgress:
if isinstance(event, MCPToolProgressEvent): if isinstance(event, MCPToolProgressEvent):
events_received.append(event) events_received.append(event)
crewai_event_bus.register(MCPToolProgressEvent, event_handler) crewai_event_bus.register_handler(MCPToolProgressEvent, event_handler)
try: try:
wrapper = MCPToolWrapper( wrapper = MCPToolWrapper(
@@ -243,8 +243,8 @@ class TestMCPToolWrapperHeaders:
mock_result = Mock() mock_result = Mock()
mock_result.content = [Mock(text="Test result")] mock_result.content = [Mock(text="Test result")]
with patch("crewai.tools.mcp_tool_wrapper.streamablehttp_client") as mock_client, \ with patch("mcp.client.streamable_http.streamablehttp_client") as mock_client, \
patch("crewai.tools.mcp_tool_wrapper.ClientSession") as mock_session_class: patch("mcp.ClientSession") as mock_session_class:
mock_session = AsyncMock() mock_session = AsyncMock()
mock_session.initialize = AsyncMock() mock_session.initialize = AsyncMock()
@@ -277,8 +277,8 @@ class TestMCPToolWrapperHeaders:
mock_result = Mock() mock_result = Mock()
mock_result.content = [Mock(text="Test result")] mock_result.content = [Mock(text="Test result")]
with patch("crewai.tools.mcp_tool_wrapper.streamablehttp_client") as mock_client, \ with patch("mcp.client.streamable_http.streamablehttp_client") as mock_client, \
patch("crewai.tools.mcp_tool_wrapper.ClientSession") as mock_session_class: patch("mcp.ClientSession") as mock_session_class:
mock_session = AsyncMock() mock_session = AsyncMock()
mock_session.initialize = AsyncMock() mock_session.initialize = AsyncMock()
@@ -326,8 +326,8 @@ class TestMCPToolWrapperIntegration:
mock_result = Mock() mock_result = Mock()
mock_result.content = [Mock(text="Test result")] mock_result.content = [Mock(text="Test result")]
with patch("crewai.tools.mcp_tool_wrapper.streamablehttp_client") as mock_client, \ with patch("mcp.client.streamable_http.streamablehttp_client") as mock_client, \
patch("crewai.tools.mcp_tool_wrapper.ClientSession") as mock_session_class: patch("mcp.ClientSession") as mock_session_class:
mock_session = AsyncMock() mock_session = AsyncMock()
mock_session.initialize = AsyncMock() mock_session.initialize = AsyncMock()