from __future__ import annotations

import pytest
from httpx import AsyncClient

pytestmark = pytest.mark.smoke


class TestMcpHttpEndpoints:
    """Covers smoke-level HTTP checks for mounted MCP runtime endpoints."""

    class TestHealthz:
        """Covers health endpoint smoke behavior."""

        @pytest.mark.asyncio
        async def test_returns_ok_payload(self, client: AsyncClient) -> None:
            """Ensures GET /healthz responds with a healthy status payload."""
            response = await client.get("/healthz")

            assert response.status_code == 200
            assert response.json() == {"status": "ok"}

    class TestDocsRoute:
        """Covers static docs route smoke behavior."""

        @pytest.mark.asyncio
        async def test_serves_docs_entrypoint(self, client: AsyncClient) -> None:
            """Ensures GET /docs returns the docs site entrypoint response."""
            response = await client.get("/docs", follow_redirects=True)

            assert response.status_code == 200
            assert "text/html" in response.headers["content-type"]

    class TestMcpRoute:
        """Covers MCP transport endpoint smoke behavior."""

        @pytest.mark.asyncio
        async def test_rejects_get_stream_without_support(
            self,
            client: AsyncClient,
            mcp_session_factory,
        ) -> None:
            """Ensures GET /mcp returns method not allowed for current transport mode."""
            response = await client.get(
                "/mcp",
                headers={"Accept": "text/event-stream"},
            )

            async with mcp_session_factory() as mcp_session:
                # Keep the SDK-backed session in use for this route smoke lane.
                await mcp_session.list_tools()

            assert response.status_code == 405

        @pytest.mark.asyncio
        async def test_accepts_initialize_jsonrpc_request(
            self,
            mcp_session_factory,
        ) -> None:
            """Ensures POST /mcp accepts an initialize JSON-RPC request."""
            async with mcp_session_factory(initialize=False) as mcp_session_uninitialized:
                initialize_result = await mcp_session_uninitialized.initialize()

            assert initialize_result.protocolVersion
            assert initialize_result.serverInfo.name