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
|
# 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/.
from unittest import mock
import mozunit
import pytest
from mozversioncontrol import get_repository_object
STEPS = {
"hg": [],
"git": [
"git remote add blah https://example.com/blah",
"""
git remote add unified hg::https://hg.mozilla.org/mozilla-unified
git remote add central hg::https://hg.mozilla.org/central
git remote add try hg::https://hg.mozilla.org/try
git remote add firefox https://github.com/mozilla-firefox/firefox
""",
],
"jj": [],
}
@pytest.mark.parametrize(
"is_cinnabar,expected_remotes",
(
(
True,
[
"central",
"firefox",
"unified",
],
),
(False, ["firefox"]),
),
)
def test_get_upstream_remotes(is_cinnabar, expected_remotes, repo):
# Test is only relevant for Git.
if not repo.vcs == "git":
return
repo.execute_next_step()
vcs = get_repository_object(repo.dir)
vcs.is_cinnabar_repo = mock.MagicMock(name="is_cinnabar_repo")
vcs.is_cinnabar_repo.return_value = is_cinnabar
remotes = vcs.get_mozilla_upstream_remotes()
assert list(remotes) == [], "No official remotes should be found."
repo.execute_next_step()
remotes = sorted(list(vcs.get_mozilla_upstream_remotes()))
assert remotes == expected_remotes, "Multiple non-try remotes should be found."
if __name__ == "__main__":
mozunit.main()
|
