"""MCP tool-kind wiring tests (monkeypatched server - no live MCP connection)."""

from __future__ import annotations

import pytest
from langchain_core.tools import StructuredTool

from forge.db.base import SessionLocal
from forge.models import McpClient
from forge.services.runtime import make_runtime_ctx
from forge.tools import mcp as mcp_mod
from forge.tools.mcp import McpUnavailable, load_mcp_tool


def _fake_remote_tool(name="search"):
    async def _run(q: str) -> str:
        return f"results for {q}"
    return StructuredTool.from_function(coroutine=_run, name=name, description="remote search")


async def _make_client(tenant="t_mcp", project="p_mcp") -> str:
    async with SessionLocal() as s:
        row = McpClient(tenant_id=tenant, project_id=project, name="demo", transport="streamable_http", url="https://mcp.example/sse")
        s.add(row)
        await s.commit()
        await s.refresh(row)
        return row.id


async def test_adapters_import_available():
    # The optional extra is installed in this env; the loader resolves the client class.
    assert mcp_mod._require_adapters() is not None


async def test_load_mcp_tool_finds_remote_tool(monkeypatch):
    cid = await _make_client()

    async def fake_client_and_tools(row, tenant_id, project_id):
        return object(), [_fake_remote_tool("search")]

    monkeypatch.setattr(mcp_mod, "_client_and_tools", fake_client_and_tools)
    ctx = make_runtime_ctx("t_mcp", "p_mcp")
    tool = await load_mcp_tool({"mcp_client_id": cid, "remote_tool_name": "search"}, ctx)
    assert tool.name == "search"
    assert await tool.ainvoke({"q": "hello"}) == "results for hello"


async def test_load_mcp_tool_unknown_remote(monkeypatch):
    cid = await _make_client()
    monkeypatch.setattr(mcp_mod, "_client_and_tools", lambda *a, **k: _noop([]))
    ctx = make_runtime_ctx("t_mcp", "p_mcp")
    with pytest.raises(McpUnavailable):
        await load_mcp_tool({"mcp_client_id": cid, "remote_tool_name": "nope"}, ctx)


async def test_load_mcp_tool_missing_client():
    ctx = make_runtime_ctx("t_mcp", "p_mcp")
    with pytest.raises(McpUnavailable):
        await load_mcp_tool({"mcp_client_id": "does-not-exist", "remote_tool_name": "x"}, ctx)


async def _noop(tools):
    return object(), tools