import re
import git
import pytest
from helpers import (
EmptyDir,
TempGitFileRemote,
TempGitRepositoryWorktree,
funcname,
grm,
shell,
)
def test_worktree_fetch():
with TempGitRepositoryWorktree.get(funcname()) as (base_dir, root_commit):
with TempGitFileRemote() as (remote_path, _remote_sha):
shell(
f"""
cd {base_dir}
git --git-dir .git-main-working-tree remote add upstream file://{remote_path}
git --git-dir .git-main-working-tree push --force upstream master:master
"""
)
cmd = grm(["wt", "fetch"], cwd=base_dir)
assert cmd.returncode == 0
repo = git.Repo(f"{base_dir}/.git-main-working-tree")
assert repo.commit("master").hexsha == repo.commit("origin/master").hexsha
assert repo.commit("master").hexsha == repo.commit("upstream/master").hexsha
with EmptyDir() as tmp:
shell(
f"""
cd {tmp}
git clone {remote_path} tmp
cd tmp
echo change > mychange-remote
git add mychange-remote
git commit -m "change-remote"
git push origin HEAD:master
"""
)
remote_commit = git.Repo(f"{tmp}/tmp").commit("master").hexsha
assert repo.commit("master").hexsha == repo.commit("origin/master").hexsha
assert repo.commit("master").hexsha == repo.commit("upstream/master").hexsha
cmd = grm(["wt", "fetch"], cwd=base_dir)
assert cmd.returncode == 0
assert repo.commit("master").hexsha == repo.commit("origin/master").hexsha
assert repo.commit("master").hexsha == root_commit
assert repo.commit("upstream/master").hexsha == remote_commit
@pytest.mark.parametrize("rebase", [True, False])
@pytest.mark.parametrize("ffable", [True, False])
@pytest.mark.parametrize("has_changes", [True, False])
@pytest.mark.parametrize("stash", [True, False])
def test_worktree_pull(rebase, ffable, has_changes, stash):
with TempGitRepositoryWorktree.get(funcname()) as (base_dir, root_commit):
with TempGitFileRemote() as (remote_path, _remote_sha):
shell(
f"""
cd {base_dir}
git --git-dir .git-main-working-tree remote add upstream file://{remote_path}
git --git-dir .git-main-working-tree push --force upstream master:master
"""
)
repo = git.Repo(f"{base_dir}/.git-main-working-tree")
assert repo.commit("origin/master").hexsha == repo.commit("master").hexsha
assert repo.commit("upstream/master").hexsha == repo.commit("master").hexsha
with EmptyDir() as tmp:
shell(
f"""
cd {tmp}
git clone {remote_path} tmp
cd tmp
git checkout origin/master
echo change > mychange-remote
git add mychange-remote
git commit -m "change-remote"
git push origin HEAD:master
"""
)
remote_commit = git.Repo(f"{tmp}/tmp").commit("HEAD").hexsha
grm(["wt", "add", "master", "--track", "upstream/master"], cwd=base_dir)
repo = git.Repo(f"{base_dir}/master")
if not ffable:
shell(
f"""
cd {base_dir}/master
echo change > mychange
git add mychange
git commit -m "local-commit-in-master"
"""
)
if has_changes:
shell(
f"""
cd {base_dir}/master
echo change >> root-commit-in-worktree-1
echo uncommitedchange > uncommitedchange
"""
)
args = ["wt", "pull"]
if rebase:
args += ["--rebase"]
if stash:
args += ["--stash"]
cmd = grm(args, cwd=base_dir)
if has_changes and not stash:
assert cmd.returncode != 0
assert re.match(r".*master.*contains changes.*", cmd.stderr)
else:
assert repo.commit("upstream/master").hexsha == remote_commit
assert repo.commit("origin/master").hexsha == root_commit
assert (
repo.commit("master").hexsha
!= repo.commit("origin/master").hexsha
)
if has_changes:
assert ["uncommitedchange"] == repo.untracked_files
assert repo.is_dirty()
else:
assert not repo.is_dirty()
if not rebase:
if ffable:
assert cmd.returncode == 0
assert (
repo.commit("master").hexsha
!= repo.commit("origin/master").hexsha
)
assert (
repo.commit("master").hexsha
== repo.commit("upstream/master").hexsha
)
assert (
repo.commit("upstream/master").hexsha == remote_commit
)
else:
assert cmd.returncode != 0
assert "cannot be fast forwarded" in cmd.stderr
assert (
repo.commit("master").hexsha
!= repo.commit("origin/master").hexsha
)
assert repo.commit("master").hexsha != remote_commit
assert (
repo.commit("upstream/master").hexsha == remote_commit
)
else:
assert cmd.returncode == 0
if ffable:
assert (
repo.commit("master").hexsha
!= repo.commit("origin/master").hexsha
)
assert (
repo.commit("master").hexsha
== repo.commit("upstream/master").hexsha
)
assert (
repo.commit("upstream/master").hexsha == remote_commit
)
else:
assert (
repo.commit("master").message.strip()
== "local-commit-in-master"
)
assert repo.commit("master~1").hexsha == remote_commit