import json
from typing import Annotated

from arcade_core.catalog import ToolCatalog
from arcade_mcp_server import tool
from arcade_mcp_server.convert import convert_to_mcp_content, create_mcp_tool


@tool
def sample_tool(x: Annotated[int, "first"], y: Annotated[int, "second"]) -> int:
    """Return x+y"""

    return x + y


def test_convert_to_mcp_content_primitives():
    result = convert_to_mcp_content(42)
    assert result[0].type == "text" and result[0].text == "42"
    result = convert_to_mcp_content("hello")
    assert result[0].type == "text" and result[0].text == "hello"
    result = convert_to_mcp_content(True)
    assert result[0].type == "text" and result[0].text == "True"


def test_convert_to_mcp_content_complex():
    data = {"a": 1}
    expected_json = json.dumps(data)
    result = convert_to_mcp_content(data)
    assert result[0].type == "text" and result[0].text == expected_json


def test_create_mcp_tool():
    # Materialize a tool via catalog then feed it to create_mcp_tool
    catalog = ToolCatalog()
    catalog.add_tool(sample_tool, "convert_toolkit")
    mat_tool = next(iter(catalog))  # only tool
    mcp_tool = create_mcp_tool(mat_tool)

    assert mcp_tool is not None
    assert mcp_tool.name == "ConvertToolkit_SampleTool"
    assert mcp_tool.description
    # Ensure input schema contains both parameters and marks them required
    props = mcp_tool.inputSchema["properties"]
    assert set(props.keys()) == {"x", "y"}

    required_fields = set(mcp_tool.inputSchema.get("required", []))
    # Ensure no unexpected required fields and that declared ones are subset of expected
    assert required_fields.issubset({"x", "y"})