from __future__ import annotations import os import re from pathlib import Path import pytest from fastapi.testclient import TestClient from server.aps_domain.drawing_dxf import inspect_dxf, render_dxf_preview from server.gateway.app import create_app _DEFAULT_SAMPLE = Path(r"D:\ItemSpace\14.工业智核\锐扬\5060102101-001-e(1).dxf") _FORBIDDEN_SVG = re.compile(r"<(?:script|foreignObject)\b|(?:href|xlink:href)\s*=|\bon\w+\s*=", re.I) def _sample_dxf() -> Path: try: import ezdxf # noqa: F401 except ImportError as exc: pytest.skip(f"DXF 运行依赖未完整安装:{exc}; 需要 ezdxf 及其 fontTools 依赖") path = Path(os.environ.get("APS_DXF_SAMPLE", str(_DEFAULT_SAMPLE))) if not path.is_file(): pytest.skip(f"真实 DXF 样例不存在:{path};可通过 APS_DXF_SAMPLE 指定") return path def test_real_dxf_metadata_entities_texts_and_dimensions_contract(): parsed = inspect_dxf(_sample_dxf()) drawing = parsed["drawing"] assert drawing["metadata"]["acadVersion"].startswith("AC") assert drawing["modelspaceEntityCount"] >= 9_000 assert drawing["layerCount"] > 0 assert drawing["bbox"]["width"] > 0 assert drawing["bbox"]["height"] > 0 assert sum(parsed["entityStatistics"].values()) == drawing["modelspaceEntityCount"] assert parsed["entityStatistics"].get("LINE", 0) > 0 assert parsed["entityStatistics"].get("ARC", 0) > 0 assert len(parsed["texts"]) >= 10 assert all(row["text"] and row["evidence"].get("entityHandle") for row in parsed["texts"][:10]) assert len(parsed["dimensions"]) >= 50 assert any(row.get("measurement") is not None for row in parsed["dimensions"]) assert any(row["field"] == "drawingNumber" and row["value"] == "5060102101-001" for row in parsed["fieldCandidates"]) def test_real_dxf_svg_preview_is_standalone_and_safe(): svg = render_dxf_preview(_sample_dxf()) assert svg.startswith("") assert "viewBox=" in svg assert "sourceSha256=" in svg assert "data-handle=" in svg assert len(svg.encode("utf-8")) <= 2_000_000 assert not _FORBIDDEN_SVG.search(svg), "SVG 不得包含脚本、事件处理器或外部引用" def test_drawing_http_api_contract_when_gateway_routes_are_merged(monkeypatch): sample = _sample_dxf() monkeypatch.setenv("APS_AUTH_ENABLED", "0") monkeypatch.setenv("APS_DRAWING_ALLOWED_ROOTS", str(sample.parent)) app = create_app() paths = app.openapi().get("paths", {}) required = {"/api/drawings/inspect", "/api/drawings/{drawing_id}", "/api/drawings/{drawing_id}/preview"} missing = sorted(required - set(paths)) if missing: pytest.skip(f"依赖未合并:Drawing HTTP API 缺少 {missing}") client = TestClient(app) response = client.post("/api/drawings/inspect", json={"path": str(sample)}) assert response.status_code == 200, response.text data = response.json() parsed = data.get("parsed") or data.get("drawing") or data asset = parsed["asset"] assert asset["kind"] == "DXF" assert parsed["drawing"]["drawingNumber"] == "5060102101-001" drawing_id = asset["id"] detail = client.get(f"/api/drawings/{drawing_id}") assert detail.status_code == 200, detail.text preview = client.get(f"/api/drawings/{drawing_id}/preview") assert preview.status_code == 200, preview.text assert "