from mcp.client2 import MCPClient
from mcp.schemas import MCPInvokeRequest


def test_mcp_registry_and_invoke_mock():
    client = MCPClient()

    assert "postgres" in client.list_servers()
    result = client.invoke(MCPInvokeRequest(server_name="postgres", tool_name="run_query", arguments={"sql": "SELECT 1"}))

    assert result.success is True
    assert result.content["status"] == "mocked"