autocad-mcp-pro/tests/test_preflight.py

from __future__ import annotations

import pytest

from engineering.preflight import preflight_drawing


def _complete_requirements() -> dict:
    return {
        "units": "mm",
        "part_type": "bracket",
        "dimensions": {"width": 80.0, "height": 50.0, "thickness": 8.0},
        "tolerance_policy": "ISO 2768-m",
    }


def test_preflight_missing_production_fields_is_not_ready():
    result = preflight_drawing("L bracket", requirements={})

    assert result.ready is False
    assert {q.code for q in result.questions} == {
        "MISSING_UNITS",
        "MISSING_PART_TYPE",
        "MISSING_DIMENSIONS",
        "MISSING_TOLERANCE_POLICY",
    }
    assert result.spec_hash.startswith("sha256:")


def test_preflight_normalizes_complete_spec_and_hash_is_deterministic():
    first = preflight_drawing(
        "  L bracket  ",
        requirements=_complete_requirements(),
        sheet_size="a3",
        layer_set_id="MECH",
    )
    second = preflight_drawing(
        "L bracket",
        requirements={
            "tolerance_policy": "ISO 2768-m",
            "dimensions": {"thickness": 8, "height": 50, "width": 80},
            "part_type": "bracket",
            "units": "MM",
        },
        sheet_size="A3",
        layer_set_id="mech",
    )

    assert first.ready is True
    assert first.normalized_spec["requirements"]["units"] == "mm"
    assert first.spec_hash == second.spec_hash


def test_preflight_detects_conflicting_constraints():
    requirements = _complete_requirements()
    requirements["constraints"] = [
        {"field": "dimensions.width", "value": 80},
        {"field": "dimensions.width", "value": 90},
    ]

    result = preflight_drawing("L bracket", requirements=requirements)

    assert result.ready is False
    assert result.conflicts[0].code == "CONFLICTING_REQUIREMENT"
    assert result.conflicts[0].field == "dimensions.width"


@pytest.mark.parametrize(
    ("kwargs", "code"),
    [
        ({"sheet_size": "LETTER"}, "INVALID_SHEET_SIZE"),
        ({"scale": 0}, "INVALID_SCALE"),
        ({"view_count": 0}, "INVALID_VIEW_COUNT"),
        ({"dim_style": "guess"}, "INVALID_DIM_STYLE"),
        ({"layer_set_id": "custom"}, "INVALID_LAYER_SET"),
    ],
)
def test_preflight_rejects_invalid_planning_parameters(kwargs, code):
    result = preflight_drawing(
        "L bracket",
        requirements=_complete_requirements(),
        **kwargs,
    )

    assert result.ready is False
    assert code in {conflict.code for conflict in result.conflicts}


def test_preflight_does_not_invent_values_when_assumptions_disabled():
    result = preflight_drawing(
        "L bracket",
        requirements={"part_type": "bracket"},
        allow_assumptions=False,
    )

    assert result.assumptions == []
    assert "units" not in result.normalized_spec["requirements"]


@pytest.mark.asyncio
async def test_backend_plan_accepts_matching_ready_preflight(backend):
    preflight = await backend.drawing_preflight(
        "L bracket",
        requirements=_complete_requirements(),
    )

    plan = await backend.drawing_plan(
        "L bracket",
        requirements=preflight.normalized_spec["requirements"],
        spec_hash=preflight.spec_hash,
    )

    assert plan.preflight_status == "ready"
    assert plan.spec_hash == preflight.spec_hash
    assert plan.requirements["units"] == "mm"


@pytest.mark.asyncio
async def test_backend_plan_rejects_not_ready_preflight_hash(backend):
    preflight = await backend.drawing_preflight("L bracket", requirements={})

    with pytest.raises(RuntimeError, match="not ready"):
        await backend.drawing_plan("L bracket", spec_hash=preflight.spec_hash)


@pytest.mark.asyncio
async def test_backend_plan_rejects_fields_that_do_not_match_preflight_hash(backend):
    preflight = await backend.drawing_preflight(
        "L bracket",
        requirements=_complete_requirements(),
    )

    with pytest.raises(RuntimeError, match="do not match"):
        await backend.drawing_plan(
            "Different part",
            requirements=preflight.normalized_spec["requirements"],
            spec_hash=preflight.spec_hash,
        )


@pytest.mark.asyncio
async def test_legacy_plan_remains_compatible_and_marks_preflight_skipped(backend):
    plan = await backend.drawing_plan("Legacy drawing")

    assert plan.preflight_status == "skipped"
    assert plan.spec_hash is None
    assert plan.requirements == {}


@pytest.mark.asyncio
async def test_document_change_clears_plan_preflight_and_gdt_state(backend):
    preflight = await backend.drawing_preflight(
        "First drawing",
        requirements=_complete_requirements(),
    )
    await backend.drawing_plan(
        "First drawing",
        requirements=preflight.normalized_spec["requirements"],
        spec_hash=preflight.spec_hash,
    )
    backend._gdt_datums_defined = {"A"}
    backend._gdt_datums_referenced = {"A", "B"}

    await backend.drawing_new()

    assert backend.get_plan_spec() is None
    assert backend._preflight_result is None
    assert backend._gdt_datums_defined == set()
    assert backend._gdt_datums_referenced == set()