test_chat_history.py

import pytest
from textual.app import App, ComposeResult

from agent_chat_cli.components.chat_history import ChatHistory
from agent_chat_cli.components.messages import (
    Message,
    SystemMessage,
    UserMessage,
    AgentMessage,
    ToolMessage,
)


class ChatHistoryApp(App):
    def compose(self) -> ComposeResult:
        yield ChatHistory()


class TestChatHistoryAddMessage:
    @pytest.fixture
    def app(self):
        return ChatHistoryApp()

    async def test_adds_system_message(self, app):
        async with app.run_test():
            chat_history = app.query_one(ChatHistory)
            chat_history.add_message(Message.system("System alert"))

            widgets = chat_history.query(SystemMessage)
            assert len(widgets) == 1
            assert widgets.first().message == "System alert"

    async def test_adds_user_message(self, app):
        async with app.run_test():
            chat_history = app.query_one(ChatHistory)
            chat_history.add_message(Message.user("Hello"))

            widgets = chat_history.query(UserMessage)
            assert len(widgets) == 1
            assert widgets.first().message == "Hello"

    async def test_adds_agent_message(self, app):
        async with app.run_test():
            chat_history = app.query_one(ChatHistory)
            chat_history.add_message(Message.agent("I can help"))

            widgets = chat_history.query(AgentMessage)
            assert len(widgets) == 1
            assert widgets.first().message == "I can help"

    async def test_adds_tool_message_with_json_content(self, app):
        async with app.run_test():
            chat_history = app.query_one(ChatHistory)
            chat_history.add_message(
                Message.tool("read_file", '{"path": "/tmp/test.txt"}')
            )

            widgets = chat_history.query(ToolMessage)
            assert len(widgets) == 1
            assert widgets.first().tool_name == "read_file"
            assert widgets.first().tool_input == {"path": "/tmp/test.txt"}

    async def test_tool_message_handles_invalid_json(self, app):
        async with app.run_test():
            chat_history = app.query_one(ChatHistory)
            chat_history.add_message(Message.tool("bash", "not valid json"))

            widgets = chat_history.query(ToolMessage)
            assert len(widgets) == 1
            assert widgets.first().tool_input == {"raw": "not valid json"}

    async def test_adds_multiple_messages(self, app):
        async with app.run_test():
            chat_history = app.query_one(ChatHistory)
            chat_history.add_message(Message.user("First"))
            chat_history.add_message(Message.agent("Second"))
            chat_history.add_message(Message.user("Third"))

            assert len(chat_history.children) == 3