feat: projects and worktree management (#62)

tests/test_worktrees.py

@@ -0,0 +1,56 @@
+from pathlib import Path
+from types import SimpleNamespace
+
+import pytest
+
+from takopi.config import ProjectConfig, ProjectsConfig
+from takopi.context import RunContext
+from takopi.worktrees import WorktreeError, ensure_worktree, resolve_run_cwd
+
+
+def _projects_config(path: Path) -> ProjectsConfig:
+    return ProjectsConfig(
+        projects={
+            "z80": ProjectConfig(
+                alias="z80",
+                path=path,
+                worktrees_dir=Path(".worktrees"),
+            )
+        },
+        default_project=None,
+    )
+
+
+def test_resolve_run_cwd_uses_project_root(tmp_path: Path) -> None:
+    projects = _projects_config(tmp_path)
+    ctx = RunContext(project="z80")
+    assert resolve_run_cwd(ctx, projects=projects) == tmp_path
+
+
+def test_resolve_run_cwd_rejects_invalid_branch(tmp_path: Path) -> None:
+    projects = _projects_config(tmp_path)
+    ctx = RunContext(project="z80", branch="../oops")
+    with pytest.raises(WorktreeError, match="branch name"):
+        resolve_run_cwd(ctx, projects=projects)
+
+
+def test_ensure_worktree_creates_from_base(monkeypatch, tmp_path: Path) -> None:
+    project = ProjectConfig(
+        alias="z80",
+        path=tmp_path,
+        worktrees_dir=Path(".worktrees"),
+    )
+    calls: list[list[str]] = []
+
+    monkeypatch.setattr("takopi.worktrees.git_ok", lambda *args, **kwargs: False)
+    monkeypatch.setattr("takopi.worktrees.resolve_default_base", lambda *_: "main")
+
+    def _fake_git_run(args, cwd):
+        calls.append(list(args))
+        return SimpleNamespace(returncode=0, stdout="", stderr="")
+
+    monkeypatch.setattr("takopi.worktrees.git_run", _fake_git_run)
+
+    worktree_path = ensure_worktree(project, "feat/name")
+    assert worktree_path == tmp_path / ".worktrees" / "feat" / "name"
+    assert calls == [["worktree", "add", "-b", "feat/name", str(worktree_path), "main"]]