fix: remove unnecessary wraps parameter in test patch

This commit is contained in:
Greyson LaLonde
2025-09-04 23:15:20 -04:00
parent 4ef4632a8c
commit 929f9dadb4

View File

@@ -7,10 +7,8 @@ from pydantic import Field
from crewai.agent import Agent from crewai.agent import Agent
from crewai.agents.crew_agent_executor import CrewAgentExecutor from crewai.agents.crew_agent_executor import CrewAgentExecutor
from crewai.crew import Crew from crewai.crew import Crew
from crewai.flow.flow import Flow, listen, start from crewai.events.event_bus import crewai_event_bus
from crewai.llm import LLM from crewai.events.event_listener import EventListener
from crewai.task import Task
from crewai.tools.base_tool import BaseTool
from crewai.events.types.agent_events import ( from crewai.events.types.agent_events import (
AgentExecutionCompletedEvent, AgentExecutionCompletedEvent,
AgentExecutionErrorEvent, AgentExecutionErrorEvent,
@@ -24,9 +22,6 @@ from crewai.events.types.crew_events import (
CrewTestResultEvent, CrewTestResultEvent,
CrewTestStartedEvent, CrewTestStartedEvent,
) )
from crewai.events.event_bus import crewai_event_bus
from crewai.events.event_listener import EventListener
from crewai.events.types.tool_usage_events import ToolUsageFinishedEvent
from crewai.events.types.flow_events import ( from crewai.events.types.flow_events import (
FlowCreatedEvent, FlowCreatedEvent,
FlowFinishedEvent, FlowFinishedEvent,
@@ -47,7 +42,12 @@ from crewai.events.types.task_events import (
) )
from crewai.events.types.tool_usage_events import ( from crewai.events.types.tool_usage_events import (
ToolUsageErrorEvent, ToolUsageErrorEvent,
ToolUsageFinishedEvent,
) )
from crewai.flow.flow import Flow, listen, start
from crewai.llm import LLM
from crewai.task import Task
from crewai.tools.base_tool import BaseTool
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
@@ -327,9 +327,9 @@ def test_agent_emits_execution_error_event(base_agent, base_task):
error_message = "Error happening while sending prompt to model." error_message = "Error happening while sending prompt to model."
base_agent.max_retry_limit = 0 base_agent.max_retry_limit = 0
with patch.object(
CrewAgentExecutor, "invoke", wraps=base_agent.agent_executor.invoke # Patch the invoke method on the CrewAgentExecutor class directly
) as invoke_mock: with patch.object(CrewAgentExecutor, "invoke") as invoke_mock:
invoke_mock.side_effect = Exception(error_message) invoke_mock.side_effect = Exception(error_message)
with pytest.raises(Exception): with pytest.raises(Exception):