mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 00:28:31 +00:00
115 lines
3.2 KiB
Python
115 lines
3.2 KiB
Python
from datetime import datetime
|
|
from unittest.mock import patch
|
|
|
|
from crewai.agent import Agent
|
|
from crewai.task import Task
|
|
|
|
|
|
def test_agent_inject_date():
|
|
"""Test that the inject_date flag injects the current date into the task.
|
|
|
|
Tests that when inject_date=True, the current date is added to the task description.
|
|
"""
|
|
agent = Agent(
|
|
role="test_agent",
|
|
goal="test_goal",
|
|
backstory="test_backstory",
|
|
inject_date=True,
|
|
)
|
|
|
|
task = Task(
|
|
description="Test task",
|
|
expected_output="Test output",
|
|
agent=agent,
|
|
)
|
|
|
|
with patch.object(Agent, 'execute_task', return_value="Task executed") as mock_execute:
|
|
agent.execute_task(task)
|
|
|
|
called_task = mock_execute.call_args[0][0]
|
|
|
|
current_date = datetime.now().strftime("%Y-%m-%d")
|
|
assert f"Current Date: {current_date}" in called_task.description
|
|
|
|
|
|
def test_agent_without_inject_date():
|
|
"""Test that without inject_date flag, no date is injected.
|
|
|
|
Tests that when inject_date=False (default), no date is added to the task description.
|
|
"""
|
|
agent = Agent(
|
|
role="test_agent",
|
|
goal="test_goal",
|
|
backstory="test_backstory",
|
|
)
|
|
|
|
task = Task(
|
|
description="Test task",
|
|
expected_output="Test output",
|
|
agent=agent,
|
|
)
|
|
|
|
original_description = task.description
|
|
|
|
with patch.object(Agent, 'execute_task', return_value="Task executed") as mock_execute:
|
|
agent.execute_task(task)
|
|
|
|
called_task = mock_execute.call_args[0][0]
|
|
|
|
assert "Current Date:" not in called_task.description
|
|
assert called_task.description == original_description
|
|
|
|
|
|
def test_agent_inject_date_custom_format():
|
|
"""Test that the inject_date flag with custom date_format works correctly.
|
|
|
|
Tests that when inject_date=True with a custom date_format, the date is formatted correctly.
|
|
"""
|
|
agent = Agent(
|
|
role="test_agent",
|
|
goal="test_goal",
|
|
backstory="test_backstory",
|
|
inject_date=True,
|
|
date_format="%d/%m/%Y",
|
|
)
|
|
|
|
task = Task(
|
|
description="Test task",
|
|
expected_output="Test output",
|
|
agent=agent,
|
|
)
|
|
|
|
with patch.object(Agent, 'execute_task', return_value="Task executed") as mock_execute:
|
|
agent.execute_task(task)
|
|
|
|
called_task = mock_execute.call_args[0][0]
|
|
|
|
current_date = datetime.now().strftime("%d/%m/%Y")
|
|
assert f"Current Date: {current_date}" in called_task.description
|
|
|
|
|
|
def test_agent_inject_date_invalid_format():
|
|
"""Test error handling with invalid date format.
|
|
|
|
Tests that when an invalid date_format is provided, the task description remains unchanged.
|
|
"""
|
|
agent = Agent(
|
|
role="test_agent",
|
|
goal="test_goal",
|
|
backstory="test_backstory",
|
|
inject_date=True,
|
|
date_format="invalid",
|
|
)
|
|
|
|
task = Task(
|
|
description="Test task",
|
|
expected_output="Test output",
|
|
agent=agent,
|
|
)
|
|
|
|
original_description = task.description
|
|
|
|
agent.execute_task(task)
|
|
|
|
assert task.description == original_description
|