"""
Unit tests for the ``sys_session_send`` tool schema.

These cover the ``file_ids`` field added to the OBJECT form of the
``args`` parameter (U1). The schema is validated with ``jsonschema``
against the ``args`` sub-schema so the back-compat contract (plain
string form, object form without ``file_ids``) is exercised directly.
"""

from __future__ import annotations

import jsonschema
import pytest

from omnigent.spec.types import AgentSpec
from omnigent.tools.builtins.spawn import _build_sys_session_send_schema


def _args_schema() -> dict:
    """:returns: The ``anyOf`` schema for the ``args`` parameter."""
    schema = _build_sys_session_send_schema({})
    return schema["function"]["parameters"]["properties"]["args"]


def _schema_with_subagent() -> dict:
    """:returns: The schema for a tool with one named sub-agent."""
    return _build_sys_session_send_schema(
        {"researcher": AgentSpec(spec_version=1, name="researcher", description="d")}
    )


def _object_branch() -> dict:
    """:returns: The object branch of the ``args`` ``anyOf``."""
    for branch in _args_schema()["anyOf"]:
        if branch.get("type") == "object":
            return branch
    raise AssertionError("no object branch in args anyOf")


def _validate(args: object) -> None:
    """Validate ``args`` against the ``args`` parameter schema."""
    jsonschema.validate(instance=args, schema=_args_schema())


# ── Schema structure ──────────────────────────────────


def test_file_ids_present_and_optional() -> None:
    branch = _object_branch()
    assert branch["properties"]["file_ids"] == {
        "type": "array",
        "items": {"type": "string", "minLength": 1},
        "minItems": 1,
        "uniqueItems": True,
        "description": branch["properties"]["file_ids"]["description"],
    }
    assert "file_ids" not in branch["required"]
    assert branch["required"] == ["input"]
    assert branch["additionalProperties"] is False


def test_description_mentions_file_ids() -> None:
    schema = _schema_with_subagent()
    assert "file_ids" in schema["function"]["description"]


def test_file_ids_description_mentions_fresh_named_spawn_only() -> None:
    schema = _schema_with_subagent()
    description = schema["function"]["description"]
    file_ids_description = _object_branch()["properties"]["file_ids"]["description"]

    assert "first named" in description
    assert "session_id" in description
    assert "continuing an existing named session" in description
    assert "first named" in file_ids_description
    assert "session_id" in file_ids_description
    assert "continuing an existing named session" in file_ids_description


# ── Happy paths ───────────────────────────────────────


def test_object_args_with_file_ids_validate() -> None:
    _validate({"input": "go", "file_ids": ["file_abc"]})


def test_object_args_without_file_ids_validate() -> None:
    _validate({"input": "go"})


def test_plain_string_args_validate() -> None:
    _validate("just a message")


# ── Errors ────────────────────────────────────────────


def test_file_ids_string_rejected() -> None:
    with pytest.raises(jsonschema.ValidationError):
        _validate({"input": "go", "file_ids": "file_abc"})


def test_empty_file_ids_rejected() -> None:
    with pytest.raises(jsonschema.ValidationError):
        _validate({"input": "go", "file_ids": []})


def test_duplicate_file_ids_rejected() -> None:
    with pytest.raises(jsonschema.ValidationError):
        _validate({"input": "go", "file_ids": ["file_abc", "file_abc"]})


def test_empty_file_id_rejected() -> None:
    with pytest.raises(jsonschema.ValidationError):
        _validate({"input": "go", "file_ids": [""]})


def test_file_ids_non_string_item_rejected() -> None:
    with pytest.raises(jsonschema.ValidationError):
        _validate({"input": "go", "file_ids": [123]})