66 lines
2.3 KiB
Python
66 lines
2.3 KiB
Python
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
|