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
|
# 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 __future__ import absolute_import, print_function, unicode_literals
import unittest
from .. import target_tasks
from .. import try_option_syntax
from ..graph import Graph
from ..taskgraph import TaskGraph
from .util import TestTask
from mozunit import main
class FakeTryOptionSyntax(object):
def __init__(self, message, task_graph):
self.trigger_tests = 0
self.notifications = None
def task_matches(self, attributes):
return 'at-at' in attributes
class TestTargetTasks(unittest.TestCase):
def default_matches(self, run_on_projects, project):
method = target_tasks.get_method('default')
graph = TaskGraph(tasks={
'a': TestTask(kind='build', label='a',
attributes={'run_on_projects': run_on_projects}),
}, graph=Graph(nodes={'a'}, edges=set()))
parameters = {'project': project}
return 'a' in method(graph, parameters)
def test_default_all(self):
"""run_on_projects=[all] includes release, integration, and other projects"""
self.assertTrue(self.default_matches(['all'], 'mozilla-central'))
self.assertTrue(self.default_matches(['all'], 'mozilla-inbound'))
self.assertTrue(self.default_matches(['all'], 'mozilla-aurora'))
self.assertTrue(self.default_matches(['all'], 'baobab'))
def test_default_integration(self):
"""run_on_projects=[integration] includes integration projects"""
self.assertFalse(self.default_matches(['integration'], 'mozilla-central'))
self.assertTrue(self.default_matches(['integration'], 'mozilla-inbound'))
self.assertFalse(self.default_matches(['integration'], 'baobab'))
def test_default_relesae(self):
"""run_on_projects=[release] includes release projects"""
self.assertTrue(self.default_matches(['release'], 'mozilla-central'))
self.assertFalse(self.default_matches(['release'], 'mozilla-inbound'))
self.assertFalse(self.default_matches(['release'], 'baobab'))
def test_default_nothing(self):
"""run_on_projects=[] includes nothing"""
self.assertFalse(self.default_matches([], 'mozilla-central'))
self.assertFalse(self.default_matches([], 'mozilla-inbound'))
self.assertFalse(self.default_matches([], 'baobab'))
def test_try_option_syntax(self):
tasks = {
'a': TestTask(kind=None, label='a'),
'b': TestTask(kind=None, label='b', attributes={'at-at': 'yep'}),
}
graph = Graph(nodes=set('ab'), edges=set())
tg = TaskGraph(tasks, graph)
params = {'message': 'try me'}
orig_TryOptionSyntax = try_option_syntax.TryOptionSyntax
try:
try_option_syntax.TryOptionSyntax = FakeTryOptionSyntax
method = target_tasks.get_method('try_option_syntax')
self.assertEqual(method(tg, params), ['b'])
finally:
try_option_syntax.TryOptionSyntax = orig_TryOptionSyntax
if __name__ == '__main__':
main()
|
