新增 MCP 工具 set_drawing_area(mode="max"|"restore"):最大化主窗口 + CLEANSCREENON (隐藏功能区/命令行/面板)。实测绘图区 2110x462 -> 3310x1084(面积 3.68x), restore 精确回位(窗口矩形/最大化态/绘图区尺寸全部还原)。 坑:CLEANSCREENOFF 是异步命令,AutoCAD 会按「进入干净屏幕时的布局」重设窗口几何, 把同步的 SetWindowPlacement 覆盖掉(表现为窗口停在最大化)。解法是插件内 SetTimer 延迟 700ms 再压回;实测窗口在 0.5~1.0s 之间收敛,故调用方等待取 1.5s。 取证层: - 新增 maximized_drawing_area(ctx) 上下文管理器,__main__ 用它包住整个取证循环 (多图框只进出一次);受 DRAWING_AREA_MAXIMIZE 控制,工具缺失/调用失败静默降级, restore 失败不掩盖业务异常。 - 内容过滤统一到规划与回退两条路径,共用同一谓词;过滤后为空则退回原集合。 - 判定改用「实体 bbox」而非「占用格」:格子量化会把跨界内容算进相邻格,反而更粗 (实测实体 bbox 正好压象限边界时 4 个象限全判有内容)。heatmap.occupied_boxes 随之删除。 实测(真实图纸 A0/A1/A2 三图框):截图张数 12/8/4 -> 8/5/4;pytest 296 passed。 新增 verify_drawing_area.py 作为该工具的验收脚本。
140 lines
5.1 KiB
Python
140 lines
5.1 KiB
Python
"""`dwg_review.config` 单元测试(只验证纯函数与常量类型,不依赖真实环境变量)。"""
|
|
|
|
import os
|
|
|
|
from dwg_review import config
|
|
|
|
|
|
def test_resolve_path_relative_uses_project_root():
|
|
assert config._resolve_path("reports") == config.PROJECT_ROOT / "reports"
|
|
|
|
|
|
def test_resolve_path_absolute_unchanged(tmp_path):
|
|
assert config._resolve_path(str(tmp_path)) == tmp_path
|
|
|
|
|
|
def test_project_root_is_repo_root():
|
|
assert (config.PROJECT_ROOT / "dwg_review" / "config.py").is_file()
|
|
|
|
|
|
def test_read_env_prefers_environ(monkeypatch):
|
|
monkeypatch.setenv("CAD_TEST_ENV_PROBE", "hello")
|
|
assert config._read_env("CAD_TEST_ENV_PROBE") == "hello"
|
|
|
|
|
|
def test_read_env_missing_returns_empty(monkeypatch):
|
|
monkeypatch.delenv("CAD_TEST_DEFINITELY_MISSING_XYZ", raising=False)
|
|
assert config._read_env("CAD_TEST_DEFINITELY_MISSING_XYZ") == ""
|
|
|
|
|
|
def test_read_bool_defaults_when_unset(monkeypatch):
|
|
monkeypatch.delenv("CAD_TEST_BOOL_UNSET", raising=False)
|
|
assert config._read_bool("CAD_TEST_BOOL_UNSET", True) is True
|
|
assert config._read_bool("CAD_TEST_BOOL_UNSET", False) is False
|
|
|
|
|
|
def test_read_bool_empty_string_uses_default(monkeypatch):
|
|
monkeypatch.setenv("CAD_TEST_BOOL_EMPTY", " ")
|
|
assert config._read_bool("CAD_TEST_BOOL_EMPTY", True) is True
|
|
|
|
|
|
def test_read_bool_truthy_and_falsy(monkeypatch):
|
|
for raw in ("1", "true", "TRUE", "Yes", "on"):
|
|
monkeypatch.setenv("CAD_TEST_BOOL", raw)
|
|
assert config._read_bool("CAD_TEST_BOOL", False) is True
|
|
for raw in ("0", "false", "no", "off", "nope"):
|
|
monkeypatch.setenv("CAD_TEST_BOOL", raw)
|
|
assert config._read_bool("CAD_TEST_BOOL", True) is False
|
|
|
|
|
|
def test_has_api_key_reflects_module_value(monkeypatch):
|
|
monkeypatch.setattr(config, "API_KEY", "")
|
|
assert config.has_api_key() is False
|
|
monkeypatch.setattr(config, "API_KEY", "sk-x")
|
|
assert config.has_api_key() is True
|
|
|
|
|
|
def test_has_api_key_ignores_whitespace(monkeypatch):
|
|
monkeypatch.setattr(config, "API_KEY", " ")
|
|
assert config.has_api_key() is False
|
|
|
|
|
|
def test_constants_have_expected_types():
|
|
assert isinstance(config.HOST, str)
|
|
assert isinstance(config.PORT, int)
|
|
assert isinstance(config.CALL_TIMEOUT, float)
|
|
assert isinstance(config.CALL_RETRIES, int)
|
|
assert isinstance(config.MAX_FRAME_BYTES, int)
|
|
assert isinstance(config.IMAGE_MAX_EDGE, int)
|
|
assert isinstance(config.IMAGE_FORMAT, str)
|
|
assert isinstance(config.CHECKLIST_PATH, type(config.PROJECT_ROOT))
|
|
|
|
|
|
def test_credibility_constants_have_expected_types():
|
|
assert isinstance(config.DETERMINISTIC_CHECKS, bool)
|
|
assert isinstance(config.VALIDATE_HANDLES, bool)
|
|
assert isinstance(config.CHECK_UNCLOSED_POLYLINES, bool)
|
|
assert isinstance(config.ZERO_LENGTH_TOLERANCE, float)
|
|
assert isinstance(config.DUP_TOLERANCE, float)
|
|
assert isinstance(config.DUP_GROUP_LIMIT, int)
|
|
|
|
|
|
def test_tiling_constants_have_expected_types():
|
|
assert isinstance(config.TILING_ENABLED, bool)
|
|
assert isinstance(config.TEXT_HEIGHT_MM, float)
|
|
assert isinstance(config.TEXT_READABLE_PX, int)
|
|
assert isinstance(config.LINE_READABLE_PX, int)
|
|
assert isinstance(config.TILE_MAX_COUNT, int)
|
|
assert isinstance(config.TILE_OVERLAP, float)
|
|
assert isinstance(config.HEATMAP_CELL_DIV, int)
|
|
assert isinstance(config.HEATMAP_CLUSTER_DILATE, int)
|
|
assert isinstance(config.HEATMAP_MAX_AREA_FRAC, float)
|
|
|
|
|
|
def test_tiling_defaults_are_sane():
|
|
# 默认启用瓦片规划;可读下限为正;成本上界为正
|
|
assert config.TILING_ENABLED is True
|
|
assert config.TEXT_HEIGHT_MM > 0
|
|
assert config.TEXT_READABLE_PX > 0
|
|
assert config.LINE_READABLE_PX > 0
|
|
assert config.TILE_MAX_COUNT > 0
|
|
assert config.HEATMAP_CELL_DIV > 0
|
|
|
|
|
|
def test_drawing_area_constants_have_expected_types():
|
|
assert isinstance(config.DRAWING_AREA_MAXIMIZE, bool)
|
|
assert isinstance(config.DRAWING_AREA_ENTER_WAIT, float)
|
|
assert isinstance(config.DRAWING_AREA_RESTORE_WAIT, float)
|
|
|
|
|
|
def test_drawing_area_defaults_are_sane():
|
|
# 默认开启绘图区最大化;还原等待须盖过插件内 700ms 的延迟还原(实测 1.0s 才收敛)
|
|
assert config.DRAWING_AREA_MAXIMIZE is True
|
|
assert config.DRAWING_AREA_ENTER_WAIT >= 0
|
|
assert config.DRAWING_AREA_RESTORE_WAIT >= 1.2
|
|
|
|
|
|
def test_credibility_defaults_are_on():
|
|
# 默认开启,保证默认拿到可信度收益
|
|
assert config.DETERMINISTIC_CHECKS is True
|
|
assert config.VALIDATE_HANDLES is True
|
|
assert config.CHECK_UNCLOSED_POLYLINES is True
|
|
|
|
|
|
def test_checklist_path_resolves_relative_to_root():
|
|
assert config.CHECKLIST_PATH.name == "review_checklist.json"
|
|
assert config.CHECKLIST_PATH.parent == config.PROJECT_ROOT
|
|
|
|
|
|
def test_port_default_matches_baseline():
|
|
# 基座固定监听 127.0.0.1:8080;若此值被改动,请同步更新基座与文档。
|
|
assert config.HOST == "127.0.0.1"
|
|
assert config.PORT == 8080
|
|
|
|
|
|
def test_os_name_guard_in_read_env(monkeypatch):
|
|
# 非 Windows 平台不应触碰注册表
|
|
monkeypatch.setattr(os, "name", "posix")
|
|
monkeypatch.delenv("CAD_TEST_ENV_PROBE2", raising=False)
|
|
assert config._read_env("CAD_TEST_ENV_PROBE2") == ""
|