from __future__ import annotations

from acp.schema import ToolCallStart

from vibe.acp.tools.session_update import tool_call_session_update
from vibe.core.tools.builtins.read_file import ReadFile, ReadFileArgs
from vibe.core.types import ToolCallEvent


class TestToolCallSessionUpdate:
    def _create_event(self) -> ToolCallEvent:
        return ToolCallEvent(
            tool_name="read_file",
            tool_call_id="test_call_123",
            args=ReadFileArgs(path="/tmp/test.txt"),
            tool_class=ReadFile,
        )

    def test_returns_tool_call_start(self) -> None:
        event = self._create_event()

        update = tool_call_session_update(event)

        assert update is not None
        assert isinstance(update, ToolCallStart)
        assert update.session_update == "tool_call"
        assert update.tool_call_id == "test_call_123"

    def test_returns_tool_call_start_for_streaming_event(self) -> None:
        event = ToolCallEvent(
            tool_name="read_file",
            tool_call_id="test_call_123",
            tool_class=ReadFile,
            args=None,
        )

        update = tool_call_session_update(event)

        assert update is not None
        assert isinstance(update, ToolCallStart)
        assert update.session_update == "tool_call"
        assert update.tool_call_id == "test_call_123"
        assert update.kind == "read"
        assert update.raw_input is None