From fba56056c0ccbd7b402f6b227e5e0a1e7c55097e Mon Sep 17 00:00:00 2001
From: Greyson LaLonde <greyson@crewai.com>
Date: Sat, 4 Apr 2026 02:16:34 +0800
Subject: [PATCH] fix: use spec= on mocks for typed executor fields

---
 lib/crewai/tests/memory/test_memory_root_scope.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/lib/crewai/tests/memory/test_memory_root_scope.py b/lib/crewai/tests/memory/test_memory_root_scope.py
index 9fd5ffb70..ec81fa524 100644
--- a/lib/crewai/tests/memory/test_memory_root_scope.py
+++ b/lib/crewai/tests/memory/test_memory_root_scope.py
@@ -1047,23 +1047,24 @@ class TestAgentExecutorBackwardCompat:
 
     def test_agent_executor_no_root_scope_when_memory_has_none(self) -> None:
         """Agent executor doesn't inject root_scope when memory has none."""
+        from crewai.agents.agent_builder.base_agent import BaseAgent
         from crewai.agents.agent_builder.base_agent_executor import (
             BaseAgentExecutor,
         )
         from crewai.agents.parser import AgentFinish
-        from crewai.utilities.printer import Printer
+        from crewai.task import Task
 
         mock_memory = MagicMock()
         mock_memory.read_only = False
         mock_memory.root_scope = None  # No root_scope set
         mock_memory.extract_memories.return_value = ["Fact A"]
 
-        mock_agent = MagicMock()
+        mock_agent = MagicMock(spec=BaseAgent)
         mock_agent.memory = mock_memory
         mock_agent._logger = MagicMock()
         mock_agent.role = "Researcher"
 
-        mock_task = MagicMock()
+        mock_task = MagicMock(spec=Task)
         mock_task.description = "Task"
         mock_task.expected_output = "Output"
 
@@ -1082,23 +1083,24 @@ class TestAgentExecutorBackwardCompat:
 
     def test_agent_executor_extends_root_scope_when_memory_has_one(self) -> None:
         """Agent executor extends root_scope when memory has one."""
+        from crewai.agents.agent_builder.base_agent import BaseAgent
         from crewai.agents.agent_builder.base_agent_executor import (
             BaseAgentExecutor,
         )
         from crewai.agents.parser import AgentFinish
-        from crewai.utilities.printer import Printer
+        from crewai.task import Task
 
         mock_memory = MagicMock()
         mock_memory.read_only = False
         mock_memory.root_scope = "/crew/test"  # Has root_scope
         mock_memory.extract_memories.return_value = ["Fact A"]
 
-        mock_agent = MagicMock()
+        mock_agent = MagicMock(spec=BaseAgent)
         mock_agent.memory = mock_memory
         mock_agent._logger = MagicMock()
         mock_agent.role = "Researcher"
 
-        mock_task = MagicMock()
+        mock_task = MagicMock(spec=Task)
         mock_task.description = "Task"
         mock_task.expected_output = "Output"