新增完善测试模板

tests/_support.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+"""
+测试公共支撑。
+
+1. 将 scripts/ 加入 sys.path，便于 `from cli…` / `from db…` / `from util…` 等导入。
+2. `IsolatedDataRoot`：同时设置 CLAW_* 与 JIANGCHANG_* 数据根与用户 ID，
+   与 `jiangchang_skill_core.runtime_env` 的优先级一致（CLAW_* 优先），
+   避免只设 JIANGCHANG_* 时仍读到真实 CLAW_DATA_ROOT。
+   退出时四个变量均恢复；原不存在的键会删除；并 rmtree 临时目录。
+"""
+from __future__ import annotations
+
+import os
+import shutil
+import sys
+import tempfile
+from typing import Optional
+
+_TESTS_DIR = os.path.dirname(os.path.abspath(__file__))
+_SKILL_ROOT = os.path.abspath(os.path.join(_TESTS_DIR, ".."))
+_SCRIPTS_DIR = os.path.join(_SKILL_ROOT, "scripts")
+
+if _SCRIPTS_DIR not in sys.path:
+    sys.path.insert(0, _SCRIPTS_DIR)
+
+_ISOLATION_KEYS = (
+    "CLAW_DATA_ROOT",
+    "JIANGCHANG_DATA_ROOT",
+    "CLAW_USER_ID",
+    "JIANGCHANG_USER_ID",
+)
+
+
+def get_scripts_dir() -> str:
+    return _SCRIPTS_DIR
+
+
+def get_skill_root() -> str:
+    return _SKILL_ROOT
+
+
+class IsolatedDataRoot:
+    """
+    在独立临时目录下模拟数据根，避免污染本机默认数据目录。
+
+    进入上下文时（同一套隔离值写入 CLAW_* 与 JIANGCHANG_*）：
+      - CLAW_DATA_ROOT / JIANGCHANG_DATA_ROOT -> 同一 tempfile.mkdtemp(prefix='skill-test-')
+      - CLAW_USER_ID / JIANGCHANG_USER_ID     -> _test（与业务默认用户区分）
+
+    退出时：四个变量均恢复为进入前状态（若原先未设置则 pop），并删除临时目录。
+    """
+
+    def __init__(self, user_id: str = "_test") -> None:
+        self._tmp: Optional[str] = None
+        self._old: dict[str, Optional[str]] = {}
+        self._user_id = user_id
+
+    def __enter__(self) -> str:
+        self._tmp = tempfile.mkdtemp(prefix="skill-test-")
+        for k in _ISOLATION_KEYS:
+            self._old[k] = os.environ.get(k)
+        os.environ["CLAW_DATA_ROOT"] = self._tmp
+        os.environ["JIANGCHANG_DATA_ROOT"] = self._tmp
+        os.environ["CLAW_USER_ID"] = self._user_id
+        os.environ["JIANGCHANG_USER_ID"] = self._user_id
+        return self._tmp
+
+    def __exit__(self, *exc_info: object) -> None:
+        for k in _ISOLATION_KEYS:
+            v = self._old.get(k)
+            if v is None:
+                os.environ.pop(k, None)
+            else:
+                os.environ[k] = v
+        if self._tmp and os.path.isdir(self._tmp):
+            shutil.rmtree(self._tmp, ignore_errors=True)
+        self._tmp = None