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
|
# 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 unittest
import pytest
from mozunit import main
from gecko_taskgraph.util.attributes import (
match_run_on_projects,
match_run_on_repo_type,
release_level,
)
class MatchRunOnProjects(unittest.TestCase):
def test_empty(self):
self.assertFalse(match_run_on_projects({"project": "birch"}, []))
def test_all(self):
self.assertTrue(match_run_on_projects({"project": "birch"}, ["all"]))
self.assertTrue(match_run_on_projects({"project": "larch"}, ["all"]))
self.assertTrue(match_run_on_projects({"project": "autoland"}, ["all"]))
self.assertTrue(match_run_on_projects({"project": "mozilla-central"}, ["all"]))
self.assertTrue(match_run_on_projects({"project": "mozilla-beta"}, ["all"]))
self.assertTrue(match_run_on_projects({"project": "mozilla-release"}, ["all"]))
def test_release(self):
self.assertFalse(match_run_on_projects({"project": "birch"}, ["release"]))
self.assertTrue(match_run_on_projects({"project": "larch"}, ["release"]))
self.assertFalse(match_run_on_projects({"project": "autoland"}, ["release"]))
self.assertTrue(
match_run_on_projects({"project": "mozilla-central"}, ["release"])
)
self.assertTrue(match_run_on_projects({"project": "mozilla-beta"}, ["release"]))
self.assertTrue(
match_run_on_projects({"project": "mozilla-release"}, ["release"])
)
def test_integration(self):
self.assertFalse(match_run_on_projects({"project": "birch"}, ["integration"]))
self.assertFalse(match_run_on_projects({"project": "larch"}, ["integration"]))
self.assertTrue(match_run_on_projects({"project": "autoland"}, ["integration"]))
self.assertFalse(
match_run_on_projects({"project": "mozilla-central"}, ["integration"])
)
self.assertFalse(
match_run_on_projects({"project": "mozilla-beta"}, ["integration"])
)
self.assertFalse(
match_run_on_projects({"project": "mozilla-integration"}, ["integration"])
)
def test_combo(self):
self.assertTrue(
match_run_on_projects({"project": "birch"}, ["release", "birch", "maple"])
)
self.assertTrue(
match_run_on_projects({"project": "larch"}, ["release", "birch", "maple"])
)
self.assertTrue(
match_run_on_projects({"project": "maple"}, ["release", "birch", "maple"])
)
self.assertFalse(
match_run_on_projects(
{"project": "autoland"}, ["release", "birch", "maple"]
)
)
self.assertTrue(
match_run_on_projects(
{"project": "mozilla-central"}, ["release", "birch", "maple"]
)
)
self.assertTrue(
match_run_on_projects(
{"project": "mozilla-beta"}, ["release", "birch", "maple"]
)
)
self.assertTrue(
match_run_on_projects(
{"project": "mozilla-release"}, ["release", "birch", "maple"]
)
)
self.assertTrue(match_run_on_projects({"project": "birch"}, ["birch", "trunk"]))
@pytest.mark.parametrize(
"repo_type,run_on_repo_types,expected",
(
("hg", ["hg"], True),
("hg", [], False),
("hg", ["all"], True),
("git", ["git", "hg"], True),
("git", ["hg"], False),
),
)
def test_match_run_on_repo_type(repo_type, run_on_repo_types, expected):
assert match_run_on_repo_type(repo_type, run_on_repo_types) == expected
@pytest.mark.parametrize(
"params,expected",
(
({"project": "autoland"}, "staging"),
({"project": "mozilla-central"}, "production"),
({"project": "firefox", "head_ref": "refs/heads/test"}, "staging"),
({"project": "firefox", "head_ref": "refs/tags/beta"}, "staging"),
({"project": "firefox", "head_ref": "refs/heads/beta"}, "production"),
),
)
def test_release_level(params, expected):
assert release_level(params) == expected
if __name__ == "__main__":
main()
|
