File size: 2,820 Bytes
310260a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 | """
Integration Tests for Agent Update Task (User Story 4)
Tests the end-to-end workflow of AI agent updating tasks via chat endpoint.
"""
import pytest
from tests.utils.agent_helpers import (
invoke_agent_with_message,
assert_tool_called,
extract_tool_calls
)
from tests.utils.task_helpers import create_test_task, get_task_by_id
@pytest.mark.integration
@pytest.mark.asyncio
async def test_agent_updates_task_via_chat_endpoint(test_user, test_session):
"""
Test: Agent updates task via chat endpoint
Verifies that AI agent can update a task through natural language.
"""
# Setup
task = create_test_task(test_session, test_user.id, title="Buy milk")
# Execute
response = await invoke_agent_with_message(
user_id=test_user.id,
message=f"Update task {task.id} to 'Buy organic milk'"
)
# Assert agent called update_task tool
assert_tool_called(response, "update_task")
# Verify task was updated
updated_task = get_task_by_id(test_session, task.id)
assert "organic milk" in updated_task.title.lower()
@pytest.mark.integration
@pytest.mark.asyncio
async def test_agent_updates_task_with_natural_language_change_x_to_y(test_user, test_session):
"""
Test: Agent updates task with natural language "change X to Y"
Verifies that agent recognizes "change X to Y" intent and updates task.
"""
# Setup
task = create_test_task(test_session, test_user.id, title="Buy milk")
# Execute
response = await invoke_agent_with_message(
user_id=test_user.id,
message="Change the milk task to 'Buy almond milk'"
)
# Agent should list tasks to find the task, then update it
tool_calls = extract_tool_calls(response)
tool_names = [tc["tool"] for tc in tool_calls]
# Verify agent attempted to update
assert "list_tasks" in tool_names or "update_task" in tool_names
@pytest.mark.integration
@pytest.mark.asyncio
async def test_agent_lists_tasks_then_updates_specific_task(test_user, test_session):
"""
Test: Agent lists tasks then updates specific task
Verifies multi-step workflow: list tasks, identify task, update it.
"""
# Setup
task = create_test_task(test_session, test_user.id, title="Review PR")
# Step 1: List tasks
response1 = await invoke_agent_with_message(
user_id=test_user.id,
message="Show my tasks"
)
assert_tool_called(response1, "list_tasks")
# Step 2: Update specific task
response2 = await invoke_agent_with_message(
user_id=test_user.id,
message=f"Update task {task.id} title to 'Review PR #123'"
)
assert_tool_called(response2, "update_task")
# Verify update
updated_task = get_task_by_id(test_session, task.id)
assert "123" in updated_task.title
|