"""Integration tests for the root mcp-brasil server.

Tests the fully composed server with all features mounted.
MCP_BRASIL_TOOL_SEARCH=none is set in conftest.py (before any import).
"""

import pytest
from fastmcp import Client

from mcp_brasil.server import mcp


class TestRootServerTools:
    @pytest.mark.asyncio
    async def test_listar_features_registered(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "listar_features" in names

    @pytest.mark.asyncio
    async def test_recomendar_tools_registered(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "recomendar_tools" in names

    @pytest.mark.asyncio
    async def test_planejar_consulta_registered(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "planejar_consulta" in names

    @pytest.mark.asyncio
    async def test_ibge_tools_namespaced(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "ibge_listar_estados" in names
            assert "ibge_buscar_municipios" in names

    @pytest.mark.asyncio
    async def test_bacen_tools_namespaced(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "bacen_consultar_serie" in names
            assert "bacen_buscar_serie" in names

    @pytest.mark.asyncio
    async def test_camara_tools_namespaced(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "camara_listar_deputados" in names
            assert "camara_buscar_proposicao" in names
            assert "camara_votos_nominais" in names

    @pytest.mark.asyncio
    async def test_senado_tools_namespaced(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "senado_listar_senadores" in names
            assert "senado_buscar_materia" in names
            assert "senado_listar_comissoes" in names
            assert "senado_agenda_plenario" in names

    @pytest.mark.asyncio
    async def test_listar_features_returns_summary(self) -> None:
        async with Client(mcp) as c:
            result = await c.call_tool("listar_features", {})
            assert "ibge" in result.data
            assert "bacen" in result.data
            assert "camara" in result.data
            assert "senado" in result.data


class TestRootServerResources:
    @pytest.mark.asyncio
    async def test_ibge_resources_namespaced(self) -> None:
        async with Client(mcp) as c:
            resources = await c.list_resources()
            uris = {str(r.uri) for r in resources}
            assert "data://ibge/estados" in uris
            assert "data://ibge/regioes" in uris
            assert "data://ibge/niveis-territoriais" in uris

    @pytest.mark.asyncio
    async def test_bacen_resources_namespaced(self) -> None:
        async with Client(mcp) as c:
            resources = await c.list_resources()
            uris = {str(r.uri) for r in resources}
            assert "data://bacen/catalogo" in uris
            assert "data://bacen/categorias" in uris
            assert "data://bacen/indicadores-chave" in uris

    @pytest.mark.asyncio
    async def test_camara_resources_namespaced(self) -> None:
        async with Client(mcp) as c:
            resources = await c.list_resources()
            uris = {str(r.uri) for r in resources}
            assert "data://camara/tipos-proposicao" in uris
            assert "data://camara/legislaturas" in uris
            assert "data://camara/info-api" in uris

    @pytest.mark.asyncio
    async def test_senado_resources_namespaced(self) -> None:
        async with Client(mcp) as c:
            resources = await c.list_resources()
            uris = {str(r.uri) for r in resources}
            assert "data://senado/tipos-materia" in uris
            assert "data://senado/info-api" in uris
            assert "data://senado/comissoes-permanentes" in uris


class TestRootServerPrompts:
    @pytest.mark.asyncio
    async def test_ibge_prompts_namespaced(self) -> None:
        async with Client(mcp) as c:
            prompts = await c.list_prompts()
            names = {p.name for p in prompts}
            assert "ibge_resumo_estado" in names
            assert "ibge_comparativo_regional" in names

    @pytest.mark.asyncio
    async def test_bacen_prompts_namespaced(self) -> None:
        async with Client(mcp) as c:
            prompts = await c.list_prompts()
            names = {p.name for p in prompts}
            assert "bacen_analise_economica" in names
            assert "bacen_comparar_indicadores" in names

    @pytest.mark.asyncio
    async def test_camara_prompts_namespaced(self) -> None:
        async with Client(mcp) as c:
            prompts = await c.list_prompts()
            names = {p.name for p in prompts}
            assert "camara_acompanhar_proposicao" in names
            assert "camara_perfil_deputado" in names
            assert "camara_analise_votacao" in names

    @pytest.mark.asyncio
    async def test_senado_prompts_namespaced(self) -> None:
        async with Client(mcp) as c:
            prompts = await c.list_prompts()
            names = {p.name for p in prompts}
            assert "senado_acompanhar_materia" in names
            assert "senado_perfil_senador" in names
            assert "senado_analise_votacao_senado" in names


class TestExecutarLote:
    @pytest.mark.asyncio
    async def test_registered(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            names = {t.name for t in tools}
            assert "executar_lote" in names

    @pytest.mark.asyncio
    async def test_has_docstring(self) -> None:
        async with Client(mcp) as c:
            tools = await c.list_tools()
            tool = next(t for t in tools if t.name == "executar_lote")
            assert tool.description
            assert "paralelo" in tool.description.lower()


class TestRootServerToolTags:
    @pytest.mark.asyncio
    async def test_tools_have_tags(self) -> None:
        """Verify that tools from features have tags after mounting."""
        async with Client(mcp) as c:
            tools = await c.list_tools()
            # Find a known tool
            ibge_tool = next((t for t in tools if t.name == "ibge_listar_estados"), None)
            assert ibge_tool is not None
            # Tags should propagate through mount
            # Note: tags are on the Tool object, accessible via annotations or the tool itself

    @pytest.mark.asyncio
    async def test_meta_tools_have_discovery_tag(self) -> None:
        """Meta tools (listar_features, recomendar_tools) should have discovery tags."""
        async with Client(mcp) as c:
            tools = await c.list_tools()
            listar = next((t for t in tools if t.name == "listar_features"), None)
            assert listar is not None
