149 lines
5.8 KiB
Python
149 lines
5.8 KiB
Python
import os
|
|
import subprocess
|
|
|
|
import pytest
|
|
|
|
from vdiff import git, parse
|
|
|
|
|
|
def sh(repo, *args):
|
|
subprocess.run(["git"] + list(args), cwd=repo, check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
|
|
|
|
def write(repo, name, text):
|
|
path = os.path.join(repo, name)
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
|
with open(path, "w") as f:
|
|
f.write(text)
|
|
|
|
|
|
@pytest.fixture
|
|
def repo(tmp_path):
|
|
repo = str(tmp_path)
|
|
sh(repo, "init", "-q", "-b", "main")
|
|
sh(repo, "config", "user.email", "t@example.com")
|
|
sh(repo, "config", "user.name", "Test")
|
|
write(repo, "a.py", "def f():\n return 1\n\n\ndef g():\n return 2\n")
|
|
write(repo, "b.txt", "keep\n")
|
|
write(repo, "gone.py", "bye\n")
|
|
sh(repo, "add", ".")
|
|
sh(repo, "commit", "-q", "-m", "initial")
|
|
return repo
|
|
|
|
|
|
def test_repo_root(repo):
|
|
assert git.repo_root(os.path.join(repo, "a.py")) == os.path.realpath(repo)
|
|
assert git.repo_root(repo) == os.path.realpath(repo)
|
|
assert git.repo_root("/") is None
|
|
|
|
|
|
def test_changed_files_all_kinds(repo):
|
|
write(repo, "a.py", "def f():\n return 10\n\n\ndef g():\n return 2\n") # unstaged M
|
|
write(repo, "b.txt", "keep\nmore\n")
|
|
sh(repo, "add", "b.txt") # staged M
|
|
os.remove(os.path.join(repo, "gone.py")) # unstaged D
|
|
write(repo, "sub/new.py", "x = 1\ny = 2") # untracked, no trailing newline
|
|
changes = {c.path: c for c in git.changed_files(repo)}
|
|
assert set(changes) == {"a.py", "b.txt", "gone.py", "sub/new.py"}
|
|
assert changes["a.py"] == git.FileChange("a.py", "M", False, 1, 1, None)
|
|
assert changes["b.txt"] == git.FileChange("b.txt", "M", True, 1, 0, None)
|
|
assert changes["gone.py"].status == "D" and changes["gone.py"].deletions == 1
|
|
assert changes["sub/new.py"] == git.FileChange("sub/new.py", "?", False, 2, 0, None)
|
|
|
|
|
|
def test_full_diff_working_tree(repo):
|
|
write(repo, "a.py", "def f():\n return 10\n\n\ndef g():\n return 2\n")
|
|
diff = parse.parse_unified(git.full_diff(repo, "a.py"))
|
|
assert [l.kind for l in diff.lines] == [" ", "-", "+", " ", " ", " ", " "]
|
|
assert diff.lines[2].text == " return 10"
|
|
write(repo, "n.py", "one\ntwo\n")
|
|
new = parse.parse_unified(git.full_diff(repo, "n.py", untracked=True))
|
|
assert [l.text for l in new.lines] == ["one", "two"] and all(l.kind == "+" for l in new.lines)
|
|
|
|
|
|
def test_stage_unstage_discard(repo):
|
|
write(repo, "a.py", "changed\n")
|
|
write(repo, "n.py", "new\n")
|
|
changes = {c.path: c for c in git.changed_files(repo)}
|
|
git.stage(repo, changes["a.py"])
|
|
changes = {c.path: c for c in git.changed_files(repo)}
|
|
assert changes["a.py"].staged
|
|
git.unstage(repo, changes["a.py"])
|
|
changes = {c.path: c for c in git.changed_files(repo)}
|
|
assert not changes["a.py"].staged
|
|
git.discard(repo, changes["a.py"])
|
|
git.discard(repo, git.FileChange("n.py", "?", False, 1, 0, None))
|
|
assert git.changed_files(repo) == []
|
|
assert not os.path.exists(os.path.join(repo, "n.py"))
|
|
# staged new file: discard removes it from index and disk
|
|
write(repo, "s.py", "s\n")
|
|
git.stage(repo, git.FileChange("s.py", "?", False, 1, 0, None))
|
|
change = git.changed_files(repo)[0]
|
|
assert change.status == "A"
|
|
git.discard(repo, change)
|
|
assert git.changed_files(repo) == []
|
|
|
|
|
|
def test_commits(repo):
|
|
write(repo, "a.py", "def f():\n return 10\n\n\ndef g():\n return 2\n")
|
|
write(repo, "c.py", "c\n")
|
|
os.remove(os.path.join(repo, "gone.py"))
|
|
sh(repo, "add", "-A")
|
|
sh(repo, "commit", "-q", "-m", "second\n\nbody text")
|
|
commits = git.recent_commits(repo, 10)
|
|
assert [c.subject for c in commits] == ["second", "initial"]
|
|
info, files = git.commit_with_files(repo, commits[0].sha)
|
|
assert info.subject == "second" and info.body.strip() == "body text" and info.author == "Test"
|
|
assert info.sha == commits[0].sha and len(info.date) == 16
|
|
files = {c.path: c for c in files}
|
|
assert files["a.py"] == git.FileChange("a.py", "M", False, 1, 1, None)
|
|
assert files["c.py"].status == "A" and files["c.py"].insertions == 1
|
|
assert files["gone.py"].status == "D" and files["gone.py"].deletions == 1
|
|
diff = parse.parse_unified(git.full_diff(repo, "a.py", sha=commits[0].sha))
|
|
assert [l.kind for l in diff.lines] == [" ", "-", "+", " ", " ", " ", " "]
|
|
# root commit diffs against the empty tree
|
|
_, root_files = git.commit_with_files(repo, commits[1].sha)
|
|
assert {c.path: c.status for c in root_files} == {"a.py": "A", "b.txt": "A", "gone.py": "A"}
|
|
# the short form resolves too (used by "review HEAD")
|
|
info, _ = git.commit_with_files(repo, "HEAD")
|
|
assert info.sha == commits[0].sha
|
|
|
|
|
|
def test_no_head_repo(tmp_path):
|
|
repo = str(tmp_path)
|
|
sh(repo, "init", "-q")
|
|
write(repo, "x.py", "x\n")
|
|
assert git.recent_commits(repo, 5) == []
|
|
changes = git.changed_files(repo)
|
|
assert changes[0].status == "?"
|
|
git.stage(repo, changes[0])
|
|
changes = git.changed_files(repo)
|
|
assert changes[0].status == "A" and changes[0].staged
|
|
diff = parse.parse_unified(git.full_diff(repo, "x.py"))
|
|
assert [l.kind for l in diff.lines] == ["+"]
|
|
git.unstage(repo, changes[0])
|
|
assert git.changed_files(repo)[0].status == "?"
|
|
|
|
|
|
def test_rename_in_commit(repo):
|
|
sh(repo, "mv", "b.txt", "moved.txt")
|
|
sh(repo, "commit", "-q", "-m", "rename")
|
|
_, files = git.commit_with_files(repo, "HEAD")
|
|
assert files == [git.FileChange("moved.txt", "R", False, 0, 0, "b.txt")]
|
|
|
|
|
|
def test_git_error_message(repo):
|
|
with pytest.raises(git.GitError) as info:
|
|
git.run(repo, ["rev-parse", "--verify", "nope"])
|
|
assert "\n" not in str(info.value) and info.value.stderr
|
|
|
|
|
|
def test_count_lines(tmp_path):
|
|
p = str(tmp_path / "f")
|
|
open(p, "wb").write(b"a\n" * 3000000)
|
|
assert git._count_lines(p) == 3000000
|
|
open(p, "wb").write(b"x\0y")
|
|
assert git._count_lines(p) is None
|
|
open(p, "wb").write(b"")
|
|
assert git._count_lines(p) == 0
|