autocad-mcp-pro/tests/test_pid_resources_prompt.py

"""pid_tag_parse tool, the two P&ID resources, the rewritten prompt, the contact sheet."""

from __future__ import annotations

import json

import pytest
import pytest_asyncio
from fastmcp import Client

import server


@pytest_asyncio.fixture
async def client(monkeypatch):
    monkeypatch.setenv("AUTOCAD_MCP_BACKEND", "ezdxf")
    async with Client(server.mcp) as connected:
        yield connected


@pytest.mark.asyncio
async def test_tag_parse_tool(client):
    out = (await client.call_tool("pid_tag_parse", {"tag": "10-FIC-101A"})).structured_content
    assert out["valid"] is True and out["description"] == "Flow Indicating Controller"
    out = (
        await client.call_tool(
            "pid_tag_parse",
            {"tag": "Q-1", "kind": "equipment", "equipment_prefixes": {"Q": "quench"}},
        )
    ).structured_content
    assert out["equipment_kind"] == "quench"


@pytest.mark.asyncio
async def test_resources_serve_the_catalogue_and_the_letter_tables(client):
    uris = {str(r.uri) for r in await client.list_resources()}
    assert {"autocad://pid/symbols", "autocad://standards/isa51"} <= uris
    symbols = json.loads((await client.read_resource("autocad://pid/symbols"))[0].text)
    assert symbols["catalog_version"] == "1" and any(
        s["symbol"] == "gate" for s in symbols["symbols"]
    )
    isa = json.loads((await client.read_resource("autocad://standards/isa51"))[0].text)
    assert isa["first_letters"]["F"] == "Flow" and "rules" in isa


@pytest.mark.asyncio
async def test_prompt_walks_the_tool_workflow(client):
    prompt = await client.get_prompt(
        "prompt_pid_diagram", {"project_name": "Unit 7", "revision": "B"}
    )
    text = "\n".join(m.content.text for m in prompt.messages)
    for needle in (
        "drawing_plan",
        'layer_set_id="pid"',
        "pid_symbol_insert",
        "pid_line_draw",
        "pid_graph",
        "drawing_critique",
        "drawing_finalize",
        "pid_from_spec",
        "Unit 7",
        "Ø10",
    ):
        assert needle in text, needle
    assert "circle with triangle" not in text, "no hand-drawn symbols"


def test_catalog_contact_sheet_renders_every_symbol(tmp_path):
    pytest.importorskip("matplotlib")
    from scripts.render_pid_catalog import render

    result = render(tmp_path / "sheet.png")
    assert result["symbols"] >= 100 and result["bytes"] > 10_000