123 lines
4.1 KiB
Python
123 lines
4.1 KiB
Python
import subprocess
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
SCRIPT = ROOT / "scripts" / "plan_progress.py"
|
|
|
|
|
|
def run_cli(*args, cwd=None):
|
|
return subprocess.run(
|
|
[sys.executable, str(SCRIPT), *args],
|
|
capture_output=True,
|
|
text=True,
|
|
cwd=cwd,
|
|
)
|
|
|
|
|
|
class PlanProgressCliTests(unittest.TestCase):
|
|
def test_select_seeds_progress_when_missing(self):
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
root = Path(tmp_dir)
|
|
plans_dir = root / "docs" / "plans"
|
|
plans_dir.mkdir(parents=True)
|
|
(plans_dir / "2026-01-01-old.md").write_text("old", encoding="utf-8")
|
|
(plans_dir / "2026-01-02-new.md").write_text("new", encoding="utf-8")
|
|
|
|
result = run_cli(
|
|
"select",
|
|
"-plans",
|
|
"docs/plans",
|
|
"-progress",
|
|
"memory-bank/progress.md",
|
|
cwd=root,
|
|
)
|
|
|
|
self.assertEqual(result.returncode, 0, msg=result.stderr)
|
|
self.assertEqual(result.stdout.strip(), "docs/plans/2026-01-01-old.md")
|
|
|
|
progress = root / "memory-bank" / "progress.md"
|
|
text = progress.read_text(encoding="utf-8")
|
|
self.assertIn("<!-- plan-status:start -->", text)
|
|
self.assertIn("<!-- plan-status:end -->", text)
|
|
self.assertIn("`2026-01-01-old.md` pending", text)
|
|
self.assertIn("`2026-01-02-new.md` pending", text)
|
|
|
|
def test_select_returns_first_pending_in_order(self):
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
root = Path(tmp_dir)
|
|
plans_dir = root / "docs" / "plans"
|
|
plans_dir.mkdir(parents=True)
|
|
(plans_dir / "2026-01-01-a.md").write_text("a", encoding="utf-8")
|
|
(plans_dir / "2026-01-02-b.md").write_text("b", encoding="utf-8")
|
|
|
|
progress = root / "memory-bank" / "progress.md"
|
|
progress.parent.mkdir(parents=True)
|
|
progress.write_text(
|
|
"\n".join(
|
|
[
|
|
"# Plan 状态",
|
|
"",
|
|
"<!-- plan-status:start -->",
|
|
"- [ ] `2026-01-02-b.md` pending",
|
|
"- [ ] `2026-01-01-a.md` pending",
|
|
"<!-- plan-status:end -->",
|
|
"",
|
|
]
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
result = run_cli(
|
|
"select",
|
|
"-plans",
|
|
"docs/plans",
|
|
"-progress",
|
|
"memory-bank/progress.md",
|
|
cwd=root,
|
|
)
|
|
|
|
self.assertEqual(result.returncode, 0, msg=result.stderr)
|
|
self.assertEqual(result.stdout.strip(), "docs/plans/2026-01-02-b.md")
|
|
|
|
def test_record_updates_line(self):
|
|
with tempfile.TemporaryDirectory() as tmp_dir:
|
|
root = Path(tmp_dir)
|
|
progress = root / "memory-bank" / "progress.md"
|
|
progress.parent.mkdir(parents=True)
|
|
progress.write_text(
|
|
"\n".join(
|
|
[
|
|
"# Plan 状态",
|
|
"",
|
|
"<!-- plan-status:start -->",
|
|
"- [ ] `2026-01-03-demo.md` pending",
|
|
"<!-- plan-status:end -->",
|
|
"",
|
|
]
|
|
),
|
|
encoding="utf-8",
|
|
)
|
|
|
|
result = run_cli(
|
|
"record",
|
|
"-plan",
|
|
"docs/plans/2026-01-03-demo.md",
|
|
"-status",
|
|
"done",
|
|
"-progress",
|
|
"memory-bank/progress.md",
|
|
cwd=root,
|
|
)
|
|
|
|
self.assertEqual(result.returncode, 0, msg=result.stderr)
|
|
text = progress.read_text(encoding="utf-8")
|
|
self.assertIn("- [x] `2026-01-03-demo.md` done", text)
|
|
self.assertEqual(text.count("2026-01-03-demo.md"), 1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|