roboco/tests/unit/agent_sdk/test_opencode_session.py

"""normalize_opencode_message maps an opencode message reply to panel chunks.

The OpencodeServeSession transport (subprocess + HTTP) is exercised live against
a real `opencode serve`; the deterministic message→chunk mapping, the
turn-level error surfacing, and session-id extraction are covered here.
"""

from __future__ import annotations

from typing import TYPE_CHECKING

from roboco.agent_sdk.opencode_session import (
    _extract_session_id,
    _message_error,
    normalize_opencode_message,
)

if TYPE_CHECKING:
    from roboco.agent_sdk.intake_driver import StreamChunk


def _kinds(chunks: list[StreamChunk]) -> list[str]:
    return [c.kind for c in chunks]


def test_text_part_emits_text_then_turn_end() -> None:
    chunks = normalize_opencode_message({"parts": [{"type": "text", "text": "Hi"}]})
    assert _kinds(chunks) == ["text", "turn_end"]
    assert chunks[0].text == "Hi"


def test_reasoning_part_maps_to_thinking() -> None:
    chunks = normalize_opencode_message({"parts": [{"type": "reasoning", "text": "x"}]})
    assert chunks[0].kind == "thinking"
    assert chunks[0].text == "x"


def test_tool_part_maps_to_tool_use() -> None:
    chunks = normalize_opencode_message(
        {"parts": [{"type": "tool", "tool": "read", "input": {"path": "x"}}]}
    )
    tool = next(c for c in chunks if c.kind == "tool_use")
    assert tool.tool == "read"
    assert tool.data == {"input": {"path": "x"}}


def test_fenced_draft_in_text_becomes_draft_chunk() -> None:
    fenced = '```roboco-draft\n{"title": "Add login"}\n```'
    chunks = normalize_opencode_message({"parts": [{"type": "text", "text": fenced}]})
    draft = next(c for c in chunks if c.kind == "draft")
    assert draft.data["title"] == "Add login"


def test_propose_draft_tool_part_becomes_draft_chunk() -> None:
    # The intake-tools.js propose_draft tool call (its input nested under
    # `draft`) is intercepted into a draft chunk — NOT rendered as a tool_use —
    # so the panel shows the draft card. This is the primary Grok-intake path.
    chunks = normalize_opencode_message(
        {
            "parts": [
                {
                    "type": "tool",
                    "tool": "propose_draft",
                    "input": {"draft": {"title": "Add login", "team": "backend"}},
                }
            ]
        }
    )
    assert "tool_use" not in _kinds(chunks)
    draft = next(c for c in chunks if c.kind == "draft")
    assert draft.data["title"] == "Add login"
    assert draft.data["team"] == "backend"


def test_unknown_part_skipped_but_turn_still_ends() -> None:
    chunks = normalize_opencode_message({"parts": [{"type": "mystery", "x": 1}]})
    assert _kinds(chunks) == ["turn_end"]


def test_empty_message_yields_only_turn_end() -> None:
    assert _kinds(normalize_opencode_message({"parts": []})) == ["turn_end"]


def test_turn_level_error_is_surfaced_not_blank() -> None:
    # A model failure lands in info.error with parts=[]; it must NOT render blank.
    msg = {
        "info": {
            "role": "assistant",
            "error": {
                "name": "APIError",
                "data": {"message": "Incorrect API key provided"},
            },
        },
        "parts": [],
    }
    chunks = normalize_opencode_message(msg)
    assert _kinds(chunks) == ["error", "turn_end"]
    assert "Incorrect API key" in chunks[0].text


def test_message_error_extraction() -> None:
    assert _message_error({"info": {"error": {"data": {"message": "boom"}}}}) == "boom"
    assert _message_error({"info": {"error": {"name": "APIError"}}}) == "APIError"
    assert _message_error({"info": {}}) is None
    assert _message_error({"parts": []}) is None


def test_extract_session_id_is_tolerant() -> None:
    assert _extract_session_id({"id": "s1"}) == "s1"
    assert _extract_session_id({"sessionID": "s2"}) == "s2"
    assert _extract_session_id({"info": {"id": "s3"}}) == "s3"
    assert _extract_session_id({}) is None
    assert _extract_session_id("nope") is None