wehub-resource-sync
84 lines
2.7 KiB
Python
84 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
from opensquilla.tools.candidate_patch_checkpoint import (
|
|
_git_show_head_path,
|
|
create_candidate_patch_checkpoint,
|
|
restore_candidate_patch_checkpoint,
|
|
)
|
|
|
|
|
|
def test_git_show_head_path_disambiguates_revision_like_paths(
|
|
monkeypatch,
|
|
tmp_path: Path,
|
|
) -> None:
|
|
recorded: list[str] = []
|
|
|
|
def fake_run(command: list[str], **kwargs):
|
|
recorded.extend(command)
|
|
return subprocess.CompletedProcess(command, 0, stdout=b"content")
|
|
|
|
monkeypatch.setattr(subprocess, "run", fake_run)
|
|
|
|
assert _git_show_head_path(tmp_path, "--help") == b"content"
|
|
assert recorded == ["git", "show", "--end-of-options", "HEAD:--help"]
|
|
|
|
|
|
def _init_git_workspace(path: Path) -> None:
|
|
subprocess.run(["git", "init"], cwd=path, check=True, stdout=subprocess.DEVNULL)
|
|
subprocess.run(
|
|
["git", "config", "user.email", "test@example.com"],
|
|
cwd=path,
|
|
check=True,
|
|
)
|
|
subprocess.run(["git", "config", "user.name", "Test User"], cwd=path, check=True)
|
|
|
|
|
|
def _commit_file(workspace: Path, relative_path: str, text: str) -> Path:
|
|
target = workspace / relative_path
|
|
target.parent.mkdir(parents=True, exist_ok=True)
|
|
target.write_text(text, encoding="utf-8")
|
|
subprocess.run(["git", "add", relative_path], cwd=workspace, check=True)
|
|
subprocess.run(["git", "commit", "-m", "initial"], cwd=workspace, check=True)
|
|
return target
|
|
|
|
|
|
def test_candidate_patch_checkpoint_restores_tracked_and_untracked_files(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
workspace = tmp_path / "workspace"
|
|
workspace.mkdir()
|
|
_init_git_workspace(workspace)
|
|
source = _commit_file(workspace, "src/app.py", "print('base')\n")
|
|
|
|
checkpoint = create_candidate_patch_checkpoint(workspace, label="before-candidate")
|
|
|
|
source.write_text("print('candidate')\n", encoding="utf-8")
|
|
scratch = workspace / "scratch.py"
|
|
scratch.write_text("print('debug')\n", encoding="utf-8")
|
|
|
|
restore_candidate_patch_checkpoint(checkpoint)
|
|
|
|
assert source.read_text(encoding="utf-8") == "print('base')\n"
|
|
assert not scratch.exists()
|
|
|
|
|
|
def test_candidate_patch_checkpoint_preserves_preexisting_dirty_state(
|
|
tmp_path: Path,
|
|
) -> None:
|
|
workspace = tmp_path / "workspace"
|
|
workspace.mkdir()
|
|
_init_git_workspace(workspace)
|
|
source = _commit_file(workspace, "src/app.py", "print('base')\n")
|
|
|
|
source.write_text("print('accepted')\n", encoding="utf-8")
|
|
checkpoint = create_candidate_patch_checkpoint(workspace, label="accepted")
|
|
|
|
source.write_text("print('failed-candidate')\n", encoding="utf-8")
|
|
|
|
restore_candidate_patch_checkpoint(checkpoint)
|
|
|
|
assert source.read_text(encoding="utf-8") == "print('accepted')\n"
|