From 38db7345616dca87b68f9139d04e630611314ee3 Mon Sep 17 00:00:00 2001 From: lorenzejay Date: Wed, 14 Jan 2026 15:39:34 -0800 Subject: [PATCH] fix test --- lib/crewai/tests/utilities/test_events.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/lib/crewai/tests/utilities/test_events.py b/lib/crewai/tests/utilities/test_events.py index f637b6c3d..96ecb89d0 100644 --- a/lib/crewai/tests/utilities/test_events.py +++ b/lib/crewai/tests/utilities/test_events.py @@ -348,11 +348,11 @@ def test_agent_emits_execution_error_event(base_agent, base_task): error_message = "Error happening while sending prompt to model." base_agent.max_retry_limit = 0 - with patch.object( - CrewAgentExecutor, "invoke", wraps=base_agent.agent_executor.invoke - ) as invoke_mock: - invoke_mock.side_effect = Exception(error_message) + # Patch at the class level since agent_executor is created lazily + with patch.object( + CrewAgentExecutor, "invoke", side_effect=Exception(error_message) + ): with pytest.raises(Exception): # noqa: B017 base_agent.execute_task( task=base_task,