62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
import importlib.util
|
||
import tempfile
|
||
import unittest
|
||
from pathlib import Path
|
||
|
||
ROOT = Path(__file__).resolve().parents[1]
|
||
SCRIPT = ROOT / "scripts" / "playbook.py"
|
||
|
||
|
||
def load_playbook_module():
|
||
spec = importlib.util.spec_from_file_location("playbook_script", SCRIPT)
|
||
module = importlib.util.module_from_spec(spec)
|
||
assert spec.loader is not None
|
||
spec.loader.exec_module(module)
|
||
return module
|
||
|
||
|
||
class PlaybookDocsIndexTests(unittest.TestCase):
|
||
def test_build_docs_index_lines_uses_canonical_sections(self):
|
||
playbook = load_playbook_module()
|
||
|
||
with tempfile.TemporaryDirectory() as tmp_dir:
|
||
source = Path(tmp_dir) / "index.md"
|
||
source.write_text(
|
||
"\n".join(
|
||
[
|
||
"# 文档导航（Docs Index）",
|
||
"",
|
||
"仓库级说明。",
|
||
"",
|
||
"## 跨语言（common）",
|
||
"",
|
||
"- 公共入口：`common/commit_message.md`",
|
||
"",
|
||
"## TSL（tsl/tsf）",
|
||
"",
|
||
"- 自定义 TSL 入口：`tsl/custom.md`",
|
||
"",
|
||
"## Python（python）",
|
||
"",
|
||
"- Python 入口：`python/style_guide.md`",
|
||
"",
|
||
]
|
||
)
|
||
+ "\n",
|
||
encoding="utf-8",
|
||
)
|
||
|
||
lines = playbook.build_docs_index_lines(["tsl"], source)
|
||
|
||
self.assertEqual(lines[0], "# 文档导航（Docs Index）")
|
||
self.assertEqual(lines[2], "本快照为裁剪版 Playbook（langs: tsl）。")
|
||
self.assertIn("## 跨语言（common）", lines)
|
||
self.assertIn("- 公共入口：`common/commit_message.md`", lines)
|
||
self.assertIn("## TSL（tsl/tsf）", lines)
|
||
self.assertIn("- 自定义 TSL 入口：`tsl/custom.md`", lines)
|
||
self.assertNotIn("## Python（python）", lines)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|