From 0b980db1e9d8ff5290401a44900d023a75a0a9a8 Mon Sep 17 00:00:00 2001 From: Greyson LaLonde Date: Fri, 3 Apr 2026 17:42:47 +0800 Subject: [PATCH] fix: use spec= on test mocks for typed executor fields --- lib/crewai/tests/agents/test_async_agent_executor.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/lib/crewai/tests/agents/test_async_agent_executor.py b/lib/crewai/tests/agents/test_async_agent_executor.py index 01297bdcc..4b065637a 100644 --- a/lib/crewai/tests/agents/test_async_agent_executor.py +++ b/lib/crewai/tests/agents/test_async_agent_executor.py @@ -8,13 +8,15 @@ import pytest from crewai.agents.crew_agent_executor import CrewAgentExecutor from crewai.agents.parser import AgentAction, AgentFinish +from crewai.agents.tools_handler import ToolsHandler +from crewai.llms.base_llm import BaseLLM from crewai.tools.tool_types import ToolResult @pytest.fixture def mock_llm() -> MagicMock: """Create a mock LLM for testing.""" - llm = MagicMock() + llm = MagicMock(spec=BaseLLM) llm.supports_stop_words.return_value = True llm.stop = [] return llm @@ -51,7 +53,7 @@ def mock_crew() -> MagicMock: @pytest.fixture def mock_tools_handler() -> MagicMock: """Create a mock tools handler.""" - return MagicMock() + return MagicMock(spec=ToolsHandler) @pytest.fixture