289 lines
10 KiB
Python
289 lines
10 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import os
|
|
import re
|
|
import tempfile
|
|
|
|
import pytest
|
|
import toml
|
|
import yaml
|
|
from helpers import NonExistentPath, TempGitRepository, grm, shell
|
|
|
|
|
|
def test_repos_find_nonexistent():
|
|
with NonExistentPath() as nonexistent_dir:
|
|
cmd = grm(["repos", "find", "local", nonexistent_dir])
|
|
assert "does not exist" in cmd.stderr.lower()
|
|
assert cmd.returncode != 0
|
|
assert not os.path.exists(nonexistent_dir)
|
|
|
|
|
|
def test_repos_find_file():
|
|
with tempfile.NamedTemporaryFile() as tmpfile:
|
|
cmd = grm(["repos", "find", "local", tmpfile.name])
|
|
assert "not a directory" in cmd.stderr.lower()
|
|
assert cmd.returncode != 0
|
|
|
|
|
|
def test_repos_find_empty():
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
cmd = grm(["repos", "find", "local", tmpdir])
|
|
assert cmd.returncode == 0
|
|
assert len(cmd.stdout) == 0
|
|
assert len(cmd.stderr) != 0
|
|
|
|
|
|
def test_repos_find_invalid_format():
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
cmd = grm(
|
|
["repos", "find", "local", tmpdir, "--format", "invalidformat"],
|
|
is_invalid=True,
|
|
)
|
|
assert cmd.returncode != 0
|
|
assert len(cmd.stdout) == 0
|
|
assert "invalid value 'invalidformat'" in cmd.stderr
|
|
|
|
|
|
def test_repos_find_non_git_repos():
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
shell(
|
|
f"""
|
|
cd {tmpdir}
|
|
mkdir non_git
|
|
(
|
|
cd ./non_git
|
|
echo test > test
|
|
)
|
|
"""
|
|
)
|
|
|
|
cmd = grm(["repos", "find", "local", tmpdir])
|
|
|
|
assert cmd.returncode == 0
|
|
assert len(cmd.stdout) == 0
|
|
assert len(cmd.stderr) != 0
|
|
|
|
|
|
@pytest.mark.parametrize("default_format", [True, False])
|
|
@pytest.mark.parametrize("configtype", ["toml", "yaml"])
|
|
@pytest.mark.parametrize("exclude", [None, "^.*/repo2$", "^not_matching$"])
|
|
def test_repos_find(configtype, exclude, default_format):
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
shell(
|
|
f"""
|
|
cd {tmpdir}
|
|
mkdir repo1
|
|
(
|
|
cd ./repo1
|
|
git -c init.defaultBranch=master init
|
|
echo test > test
|
|
git add test
|
|
git commit -m "commit1"
|
|
git remote add origin https://example.com/repo2.git
|
|
git remote add someremote ssh://example.com/repo2.git
|
|
)
|
|
mkdir repo2
|
|
(
|
|
cd ./repo2
|
|
git -c init.defaultBranch=master init
|
|
git checkout -b main
|
|
echo test > test
|
|
git add test
|
|
git commit -m "commit1"
|
|
git remote add origin https://example.com/repo2.git
|
|
)
|
|
mkdir non_git
|
|
(
|
|
cd non_git
|
|
echo test > test
|
|
)
|
|
"""
|
|
)
|
|
|
|
args = ["repos", "find", "local", tmpdir]
|
|
if not default_format:
|
|
args += ["--format", configtype]
|
|
if exclude:
|
|
args += ["--exclude", exclude]
|
|
cmd = grm(args)
|
|
assert cmd.returncode == 0
|
|
if exclude == "^.*/repo2$":
|
|
assert re.match(r"^.*\[skipped\] .*\/repo2$", cmd.stderr.lower())
|
|
assert "repo2" in cmd.stderr.lower()
|
|
else:
|
|
assert len(cmd.stderr) == 0
|
|
|
|
if default_format or configtype == "toml":
|
|
output = toml.loads(cmd.stdout)
|
|
elif configtype == "yaml":
|
|
output = yaml.safe_load(cmd.stdout)
|
|
else:
|
|
raise NotImplementedError()
|
|
|
|
assert isinstance(output, dict)
|
|
assert set(output.keys()) == {"trees"}
|
|
assert isinstance(output["trees"], list)
|
|
assert len(output["trees"]) == 1
|
|
for tree in output["trees"]:
|
|
assert set(tree.keys()) == {"root", "repos"}
|
|
assert tree["root"] == tmpdir
|
|
assert isinstance(tree["repos"], list)
|
|
assert len(tree["repos"]) == (1 if exclude == "^.*/repo2$" else 2)
|
|
|
|
repo1 = [r for r in tree["repos"] if r["name"] == "repo1"][0]
|
|
assert repo1["worktree_setup"] is False
|
|
assert isinstance(repo1["remotes"], list)
|
|
assert len(repo1["remotes"]) == 2
|
|
|
|
origin = [r for r in repo1["remotes"] if r["name"] == "origin"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert origin["type"] == "https"
|
|
assert origin["url"] == "https://example.com/repo2.git"
|
|
|
|
someremote = [r for r in repo1["remotes"] if r["name"] == "someremote"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert someremote["type"] == "ssh"
|
|
assert someremote["url"] == "ssh://example.com/repo2.git"
|
|
|
|
if exclude == "^.*/repo2$":
|
|
assert [r for r in tree["repos"] if r["name"] == "repo2"] == []
|
|
else:
|
|
repo2 = [r for r in tree["repos"] if r["name"] == "repo2"][0]
|
|
assert repo2["worktree_setup"] is False
|
|
assert isinstance(repo1["remotes"], list)
|
|
assert len(repo2["remotes"]) == 1
|
|
|
|
origin = [r for r in repo2["remotes"] if r["name"] == "origin"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert origin["type"] == "https"
|
|
assert origin["url"] == "https://example.com/repo2.git"
|
|
|
|
|
|
@pytest.mark.parametrize("default_format", [True, False])
|
|
@pytest.mark.parametrize("configtype", ["toml", "yaml"])
|
|
def test_repos_find_in_root(configtype, default_format):
|
|
with TempGitRepository() as repo_dir:
|
|
args = ["repos", "find", "local", repo_dir]
|
|
if not default_format:
|
|
args += ["--format", configtype]
|
|
cmd = grm(args)
|
|
assert cmd.returncode == 0
|
|
assert len(cmd.stderr) == 0
|
|
|
|
if default_format or configtype == "toml":
|
|
output = toml.loads(cmd.stdout)
|
|
elif configtype == "yaml":
|
|
output = yaml.safe_load(cmd.stdout)
|
|
else:
|
|
raise NotImplementedError()
|
|
|
|
assert isinstance(output, dict)
|
|
assert set(output.keys()) == {"trees"}
|
|
assert isinstance(output["trees"], list)
|
|
assert len(output["trees"]) == 1
|
|
for tree in output["trees"]:
|
|
assert set(tree.keys()) == {"root", "repos"}
|
|
assert tree["root"] == os.path.dirname(repo_dir)
|
|
assert isinstance(tree["repos"], list)
|
|
assert len(tree["repos"]) == 1
|
|
|
|
repo1 = [
|
|
r for r in tree["repos"] if r["name"] == os.path.basename(repo_dir)
|
|
][0]
|
|
assert repo1["worktree_setup"] is False
|
|
assert isinstance(repo1["remotes"], list)
|
|
assert len(repo1["remotes"]) == 2
|
|
|
|
origin = [r for r in repo1["remotes"] if r["name"] == "origin"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert origin["type"] == "file"
|
|
|
|
someremote = [r for r in repo1["remotes"] if r["name"] == "otherremote"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert someremote["type"] == "file"
|
|
|
|
|
|
@pytest.mark.parametrize("configtype", ["toml", "yaml"])
|
|
@pytest.mark.parametrize("default_format", [True, False])
|
|
def test_repos_find_with_invalid_repo(configtype, default_format):
|
|
with tempfile.TemporaryDirectory() as tmpdir:
|
|
shell(
|
|
f"""
|
|
cd {tmpdir}
|
|
mkdir repo1
|
|
(
|
|
cd ./repo1
|
|
git -c init.defaultBranch=master init
|
|
echo test > test
|
|
git add test
|
|
git commit -m "commit1"
|
|
git remote add origin https://example.com/repo2.git
|
|
git remote add someremote ssh://example.com/repo2.git
|
|
)
|
|
mkdir repo2
|
|
(
|
|
cd ./repo2
|
|
git -c init.defaultBranch=master init
|
|
git checkout -b main
|
|
echo test > test
|
|
git add test
|
|
git commit -m "commit1"
|
|
git remote add origin https://example.com/repo2.git
|
|
)
|
|
mkdir broken_repo
|
|
(
|
|
cd broken_repo
|
|
echo "broken" > .git
|
|
)
|
|
"""
|
|
)
|
|
|
|
args = ["repos", "find", "local", tmpdir]
|
|
if not default_format:
|
|
args += ["--format", configtype]
|
|
cmd = grm(args)
|
|
assert cmd.returncode == 0
|
|
assert "broken" in cmd.stderr
|
|
|
|
if default_format or configtype == "toml":
|
|
output = toml.loads(cmd.stdout)
|
|
elif configtype == "yaml":
|
|
output = yaml.safe_load(cmd.stdout)
|
|
else:
|
|
raise NotImplementedError()
|
|
|
|
assert isinstance(output, dict)
|
|
assert set(output.keys()) == {"trees"}
|
|
assert isinstance(output["trees"], list)
|
|
assert len(output["trees"]) == 1
|
|
for tree in output["trees"]:
|
|
assert set(tree.keys()) == {"root", "repos"}
|
|
assert tree["root"] == tmpdir
|
|
assert isinstance(tree["repos"], list)
|
|
assert len(tree["repos"]) == 2
|
|
|
|
repo1 = [r for r in tree["repos"] if r["name"] == "repo1"][0]
|
|
assert repo1["worktree_setup"] is False
|
|
assert isinstance(repo1["remotes"], list)
|
|
assert len(repo1["remotes"]) == 2
|
|
|
|
origin = [r for r in repo1["remotes"] if r["name"] == "origin"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert origin["type"] == "https"
|
|
assert origin["url"] == "https://example.com/repo2.git"
|
|
|
|
someremote = [r for r in repo1["remotes"] if r["name"] == "someremote"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert someremote["type"] == "ssh"
|
|
assert someremote["url"] == "ssh://example.com/repo2.git"
|
|
|
|
repo2 = [r for r in tree["repos"] if r["name"] == "repo2"][0]
|
|
assert repo2["worktree_setup"] is False
|
|
assert isinstance(repo1["remotes"], list)
|
|
assert len(repo2["remotes"]) == 1
|
|
|
|
origin = [r for r in repo2["remotes"] if r["name"] == "origin"][0]
|
|
assert set(origin.keys()) == {"name", "type", "url"}
|
|
assert origin["type"] == "https"
|
|
assert origin["url"] == "https://example.com/repo2.git"
|