mirror of
https://github.com/openmlsys/openmlsys-zh.git
synced 2026-03-20 20:17:03 +08:00
bdb823c575
* feat: remove bilingual button on the front page * misc: clean repo * test: fix test suite for v1/v2 restructure and removed language switch
58 lines
2.2 KiB
Python
58 lines
2.2 KiB
Python
from __future__ import annotations
|
|
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from tools.ensure_book_resources import ensure_resource_views
|
|
|
|
|
|
class EnsureBookResourcesTests(unittest.TestCase):
|
|
def test_ensure_resource_views_creates_missing_symlinks(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
root = Path(tmpdir)
|
|
chapter_dir = root / "en_chapters"
|
|
chapter_dir.mkdir()
|
|
|
|
v1 = root / "v1"
|
|
for directory in ("img", "references", "static"):
|
|
(v1 / directory).mkdir(parents=True)
|
|
(v1 / "mlsys.bib").write_text("bib", encoding="utf-8")
|
|
|
|
ensure_resource_views(chapter_dir, root)
|
|
|
|
for name in ("img", "references", "static", "mlsys.bib"):
|
|
path = chapter_dir / name
|
|
self.assertTrue(path.is_symlink(), f"{name} should be a symlink")
|
|
self.assertEqual(path.resolve(), (v1 / name).resolve())
|
|
|
|
def test_ensure_resource_views_keeps_existing_non_symlink_paths(self) -> None:
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
root = Path(tmpdir)
|
|
chapter_dir = root / "en_chapters"
|
|
chapter_dir.mkdir()
|
|
|
|
v1 = root / "v1"
|
|
for directory in ("img", "references", "static"):
|
|
(v1 / directory).mkdir(parents=True)
|
|
(v1 / "mlsys.bib").write_text("root bib", encoding="utf-8")
|
|
|
|
local_bib = chapter_dir / "mlsys.bib"
|
|
local_bib.write_text("local bib", encoding="utf-8")
|
|
local_static = chapter_dir / "static"
|
|
local_static.mkdir()
|
|
(local_static / "frontpage.html").write_text("local static", encoding="utf-8")
|
|
|
|
ensure_resource_views(chapter_dir, root)
|
|
|
|
self.assertFalse(local_bib.is_symlink())
|
|
self.assertEqual(local_bib.read_text(encoding="utf-8"), "local bib")
|
|
self.assertFalse(local_static.is_symlink())
|
|
self.assertTrue((local_static / "frontpage.html").exists())
|
|
self.assertTrue((chapter_dir / "img").is_symlink())
|
|
self.assertTrue((chapter_dir / "references").is_symlink())
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|