1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
import os
import platform
import shutil
import subprocess
import tempfile
from pathlib import Path
import pytest
# Execute the first element in each list of steps within a `repo` directory,
# then copy the whole directory to a `remoterepo`, and finally execute the
# second element on just `repo`.
SETUP = {
"hg": [
"""
echo "foo" > foo
echo "bar" > bar
hg init
hg add *
hg commit -m "Initial commit"
hg phase --public .
""",
"""
echo [paths] > .hg/hgrc
echo "default = ../remoterepo" >> .hg/hgrc
""",
],
"git": [
"""
echo "foo" > foo
echo "bar" > bar
git init
git config user.name "Testing McTesterson"
git config user.email "<test@example.org>"
git add *
git commit -am "Initial commit"
""",
"""
git remote add upstream ../remoterepo
git fetch upstream
git branch -u upstream/master
""",
],
"jj": [
"""
echo "foo" > foo
echo "bar" > bar
git init
git config user.name "Testing McTesterson"
git config user.email "<test@example.org>"
git add *
git commit -am "Initial commit"
jj git init --colocate
jj config set --repo user.name "Testing McTesterson"
jj config set --repo user.email "test@example.org"
jj describe --reset-author --no-edit
""",
"""
jj git remote add upstream ../remoterepo
jj git fetch --remote upstream
jj bookmark track master@upstream
""",
],
"src": [
"""
echo "foo" > foo
echo "bar" > bar
mkdir config
echo 1.0 > config/milestone.txt
""",
"",
],
}
class RepoTestFixture:
def __init__(self, repo_dir: Path, vcs: str, steps: list[str]):
self.dir = repo_dir
self.vcs = vcs
# This creates a step iterator. Each time execute_next_step()
# is called the next set of instructions will be executed.
self.steps = (shell(cmd, self.dir) for cmd in steps)
def execute_next_step(self):
next(self.steps)
def shell(cmd, working_dir):
for step in cmd.split(os.linesep):
subprocess.check_call(step, shell=True, cwd=working_dir)
@pytest.fixture(params=["git", "hg", "jj", "src"])
def repo(request):
if request.param == "jj":
if os.getenv("MOZ_AUTOMATION") == 1:
fetches_dir = os.environ.get("MOZ_FETCHES_DIR", "")
jj_dir = Path(fetches_dir) / "jj"
if jj_dir.is_dir():
os.environ["PATH"] = os.pathsep.join([str(jj_dir), os.environ["PATH"]])
if platform.system() == "Darwin":
pytest.skip(
"jj tests disabled for MacOS CI due to incompatible git on workers"
)
if os.getenv("MOZ_AVOID_JJ_VCS") not in (None, "0", ""):
pytest.skip("jj support disabled")
try:
subprocess.call(["jj", "--version"], stdout=subprocess.DEVNULL)
except OSError:
pytest.skip("jj unavailable")
vcs = request.param
# Use tempfile since pytest's tempdir is too long for jj on Windows
td = tempfile.TemporaryDirectory(prefix=f"{vcs}-repo")
tmpdir = Path(td.name)
steps = SETUP[vcs]
if hasattr(request.module, "STEPS"):
if vcs == "src" and vcs not in request.module.STEPS:
# Special-case SourceRepository: most tests do not handle this case,
# so allow it to be skipped if STEPS is defined but not for src.
# (Tests without STEPS will need to skip manually.)
pytest.skip("not applicable for src repo")
steps.extend(request.module.STEPS[vcs])
repo_dir = (tmpdir / "repo").resolve()
(tmpdir / "repo").mkdir()
repo_test_fixture = RepoTestFixture(repo_dir, vcs, steps)
repo_test_fixture.execute_next_step()
shutil.copytree(str(repo_dir), str(tmpdir / "remoterepo"))
repo_test_fixture.execute_next_step()
yield repo_test_fixture
|
