agent-lab/tests/test_event_agent.py

import json

import pytest

from agent_lab.application.event_agent import EventAgent
from agent_lab.domain.events import ToolCallEvent


@pytest.mark.asyncio
async def test_event_agent_executes_enabled_tool_as_tool_reply_message():
    agent = EventAgent(enabled_tools=["handoff_note"])
    event = ToolCallEvent(
        id="call_1",
        name="handoff_note",
        arguments={"message": "inspect this event"},
        raw_arguments='{"message":"inspect this event"}',
    )

    reply = await agent.handle(event)

    assert reply.role == "tool"
    assert reply.tool_call_id == "call_1"
    assert reply.name == "handoff_note"
    assert json.loads(reply.content) == {
        "tool": "handoff_note",
        "message": "inspect this event",
    }