OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # | |
3 # Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
4 # Use of this source code is governed by a BSD-style license that can be | |
5 # found in the LICENSE file. | |
6 | |
7 import collections | |
8 import copy | |
9 import json | |
10 import os | |
11 import pipes | |
12 import re | |
13 import subprocess | |
14 import sys | |
15 | |
16 import bb_utils | |
17 | |
18 sys.path.append(os.path.join(os.path.dirname(__file__), '..')) | |
19 from pylib import constants | |
20 | |
21 | |
22 CHROMIUM_COVERAGE_BUCKET = 'chromium-code-coverage' | |
23 | |
24 _BotConfig = collections.namedtuple( | |
25 'BotConfig', ['bot_id', 'host_obj', 'test_obj']) | |
26 | |
27 HostConfig = collections.namedtuple( | |
28 'HostConfig', | |
29 ['script', 'host_steps', 'extra_args', 'extra_gyp_defines', 'target_arch']) | |
30 | |
31 TestConfig = collections.namedtuple('Tests', ['script', 'tests', 'extra_args']) | |
32 | |
33 | |
34 def BotConfig(bot_id, host_object, test_object=None): | |
35 return _BotConfig(bot_id, host_object, test_object) | |
36 | |
37 | |
38 def DictDiff(d1, d2): | |
39 diff = [] | |
40 for key in sorted(set(d1.keys() + d2.keys())): | |
41 if key in d1 and d1[key] != d2.get(key): | |
42 diff.append('- %s=%s' % (key, pipes.quote(d1[key]))) | |
43 if key in d2 and d2[key] != d1.get(key): | |
44 diff.append('+ %s=%s' % (key, pipes.quote(d2[key]))) | |
45 return '\n'.join(diff) | |
46 | |
47 | |
48 def GetEnvironment(host_obj, testing, extra_env_vars=None): | |
49 init_env = dict(os.environ) | |
50 init_env['GYP_GENERATORS'] = 'ninja' | |
51 if extra_env_vars: | |
52 init_env.update(extra_env_vars) | |
53 envsetup_cmd = '. build/android/envsetup.sh' | |
54 if testing: | |
55 # Skip envsetup to avoid presubmit dependence on android deps. | |
56 print 'Testing mode - skipping "%s"' % envsetup_cmd | |
57 envsetup_cmd = ':' | |
58 else: | |
59 print 'Running %s' % envsetup_cmd | |
60 proc = subprocess.Popen(['bash', '-exc', | |
61 envsetup_cmd + ' >&2; python build/android/buildbot/env_to_json.py'], | |
62 stdout=subprocess.PIPE, stderr=subprocess.PIPE, | |
63 cwd=bb_utils.CHROME_SRC, env=init_env) | |
64 json_env, envsetup_output = proc.communicate() | |
65 if proc.returncode != 0: | |
66 print >> sys.stderr, 'FATAL Failure in envsetup.' | |
67 print >> sys.stderr, envsetup_output | |
68 sys.exit(1) | |
69 env = json.loads(json_env) | |
70 env['GYP_DEFINES'] = env.get('GYP_DEFINES', '') + \ | |
71 ' OS=android fastbuild=1 use_goma=1 gomadir=%s' % bb_utils.GOMA_DIR | |
72 if host_obj.target_arch: | |
73 env['GYP_DEFINES'] += ' target_arch=%s' % host_obj.target_arch | |
74 extra_gyp = host_obj.extra_gyp_defines | |
75 if extra_gyp: | |
76 env['GYP_DEFINES'] += ' %s' % extra_gyp | |
77 if re.search('(asan|clang)=1', extra_gyp): | |
78 env.pop('CXX_target', None) | |
79 | |
80 # Bots checkout chrome in /b/build/slave/<name>/build/src | |
81 build_internal_android = os.path.abspath(os.path.join( | |
82 bb_utils.CHROME_SRC, '..', '..', '..', '..', '..', 'build_internal', | |
83 'scripts', 'slave', 'android')) | |
84 if os.path.exists(build_internal_android): | |
85 env['PATH'] = os.pathsep.join([build_internal_android, env['PATH']]) | |
86 return env | |
87 | |
88 | |
89 def GetCommands(options, bot_config): | |
90 """Get a formatted list of commands. | |
91 | |
92 Args: | |
93 options: Options object. | |
94 bot_config: A BotConfig named tuple. | |
95 host_step_script: Host step script. | |
96 device_step_script: Device step script. | |
97 Returns: | |
98 list of Command objects. | |
99 """ | |
100 property_args = bb_utils.EncodeProperties(options) | |
101 commands = [[bot_config.host_obj.script, | |
102 '--steps=%s' % ','.join(bot_config.host_obj.host_steps)] + | |
103 property_args + (bot_config.host_obj.extra_args or [])] | |
104 | |
105 test_obj = bot_config.test_obj | |
106 if test_obj: | |
107 run_test_cmd = [test_obj.script] + property_args | |
108 for test in test_obj.tests: | |
109 run_test_cmd.extend(['-f', test]) | |
110 if test_obj.extra_args: | |
111 run_test_cmd.extend(test_obj.extra_args) | |
112 commands.append(run_test_cmd) | |
113 return commands | |
114 | |
115 | |
116 def GetBotStepMap(): | |
117 compile_step = ['compile'] | |
118 chrome_proxy_tests = ['chrome_proxy'] | |
119 python_unittests = ['python_unittests'] | |
120 std_host_tests = ['check_webview_licenses'] | |
121 std_build_steps = ['compile', 'zip_build'] | |
122 std_test_steps = ['extract_build'] | |
123 std_tests = ['ui', 'unit'] | |
124 telemetry_tests = ['telemetry_perf_unittests'] | |
125 telemetry_tests_user_build = ['telemetry_unittests', | |
126 'telemetry_perf_unittests'] | |
127 trial_tests = [ | |
128 'base_junit_tests', | |
129 'components_browsertests', | |
130 'gfx_unittests', | |
131 'gl_unittests', | |
132 ] | |
133 flakiness_server = ( | |
134 '--flakiness-server=%s' % constants.UPSTREAM_FLAKINESS_SERVER) | |
135 experimental = ['--experimental'] | |
136 bisect_chrome_output_dir = os.path.abspath( | |
137 os.path.join(os.path.dirname(__file__), os.pardir, os.pardir, os.pardir, | |
138 os.pardir, 'bisect', 'src', 'out')) | |
139 B = BotConfig | |
140 H = (lambda steps, extra_args=None, extra_gyp=None, target_arch=None: | |
141 HostConfig('build/android/buildbot/bb_host_steps.py', steps, extra_args, | |
142 extra_gyp, target_arch)) | |
143 T = (lambda tests, extra_args=None: | |
144 TestConfig('build/android/buildbot/bb_device_steps.py', tests, | |
145 extra_args)) | |
146 | |
147 bot_configs = [ | |
148 # Main builders | |
149 B('main-builder-dbg', H(std_build_steps + std_host_tests)), | |
150 B('main-builder-rel', H(std_build_steps)), | |
151 B('main-clang-builder', | |
152 H(compile_step, extra_gyp='clang=1 component=shared_library')), | |
153 B('main-clobber', H(compile_step)), | |
154 B('main-tests-rel', H(std_test_steps), | |
155 T(std_tests + telemetry_tests + chrome_proxy_tests, | |
156 ['--cleanup', flakiness_server])), | |
157 B('main-tests', H(std_test_steps), | |
158 T(std_tests, ['--cleanup', flakiness_server])), | |
159 | |
160 # Other waterfalls | |
161 B('asan-builder-tests', H(compile_step, | |
162 extra_gyp='asan=1 component=shared_library'), | |
163 T(std_tests, ['--asan', '--asan-symbolize'])), | |
164 B('blink-try-builder', H(compile_step)), | |
165 B('chromedriver-fyi-tests-dbg', H(std_test_steps), | |
166 T(['chromedriver'], | |
167 ['--install=ChromeShell', '--install=ChromeDriverWebViewShell', | |
168 '--skip-wipe', '--disable-location', '--cleanup'])), | |
169 B('fyi-x86-builder-dbg', | |
170 H(compile_step + std_host_tests, experimental, target_arch='ia32')), | |
171 B('fyi-builder-dbg', | |
172 H(std_build_steps + std_host_tests, experimental, | |
173 extra_gyp='emma_coverage=1')), | |
174 B('x86-builder-dbg', | |
175 H(compile_step + std_host_tests, target_arch='ia32')), | |
176 B('fyi-builder-rel', H(std_build_steps, experimental)), | |
177 B('fyi-tests', H(std_test_steps), | |
178 T(std_tests + python_unittests, | |
179 ['--experimental', flakiness_server, | |
180 '--coverage-bucket', CHROMIUM_COVERAGE_BUCKET, | |
181 '--cleanup'])), | |
182 B('user-build-fyi-tests-dbg', H(std_test_steps), | |
183 T(sorted(telemetry_tests_user_build + trial_tests))), | |
184 B('fyi-component-builder-tests-dbg', | |
185 H(compile_step, extra_gyp='component=shared_library'), | |
186 T(std_tests, ['--experimental', flakiness_server])), | |
187 B('gpu-builder-tests-dbg', | |
188 H(compile_step), | |
189 T(['gpu'], ['--install=ContentShell'])), | |
190 # Pass empty T([]) so that logcat monitor and device status check are run. | |
191 B('perf-bisect-builder-tests-dbg', | |
192 H(['bisect_perf_regression']), | |
193 T([], ['--chrome-output-dir', bisect_chrome_output_dir])), | |
194 B('perf-tests-rel', H(std_test_steps), | |
195 T([], ['--install=ChromeShell', '--cleanup'])), | |
196 B('webkit-latest-webkit-tests', H(std_test_steps), | |
197 T(['webkit_layout', 'webkit'], ['--cleanup', '--auto-reconnect'])), | |
198 B('webkit-latest-contentshell', H(compile_step), | |
199 T(['webkit_layout'], ['--auto-reconnect'])), | |
200 B('builder-unit-tests', H(compile_step), T(['unit'])), | |
201 | |
202 # Generic builder config (for substring match). | |
203 B('builder', H(std_build_steps)), | |
204 ] | |
205 | |
206 bot_map = dict((config.bot_id, config) for config in bot_configs) | |
207 | |
208 # These bots have identical configuration to ones defined earlier. | |
209 copy_map = [ | |
210 ('lkgr-clobber', 'main-clobber'), | |
211 ('try-builder-dbg', 'main-builder-dbg'), | |
212 ('try-builder-rel', 'main-builder-rel'), | |
213 ('try-clang-builder', 'main-clang-builder'), | |
214 ('try-fyi-builder-dbg', 'fyi-builder-dbg'), | |
215 ('try-x86-builder-dbg', 'x86-builder-dbg'), | |
216 ('try-tests-rel', 'main-tests-rel'), | |
217 ('try-tests', 'main-tests'), | |
218 ('try-fyi-tests', 'fyi-tests'), | |
219 ('webkit-latest-tests', 'main-tests'), | |
220 ] | |
221 for to_id, from_id in copy_map: | |
222 assert to_id not in bot_map | |
223 # pylint: disable=W0212 | |
224 bot_map[to_id] = copy.deepcopy(bot_map[from_id])._replace(bot_id=to_id) | |
225 | |
226 # Trybots do not upload to flakiness dashboard. They should be otherwise | |
227 # identical in configuration to their trunk building counterparts. | |
228 test_obj = bot_map[to_id].test_obj | |
229 if to_id.startswith('try') and test_obj: | |
230 extra_args = test_obj.extra_args | |
231 if extra_args and flakiness_server in extra_args: | |
232 extra_args.remove(flakiness_server) | |
233 return bot_map | |
234 | |
235 | |
236 # Return an object from the map, looking first for an exact id match. | |
237 # If this fails, look for an id which is a substring of the specified id. | |
238 # Choose the longest of all substring matches. | |
239 # pylint: disable=W0622 | |
240 def GetBestMatch(id_map, id): | |
241 config = id_map.get(id) | |
242 if not config: | |
243 substring_matches = [x for x in id_map.iterkeys() if x in id] | |
244 if substring_matches: | |
245 max_id = max(substring_matches, key=len) | |
246 print 'Using config from id="%s" (substring match).' % max_id | |
247 config = id_map[max_id] | |
248 return config | |
249 | |
250 | |
251 def GetRunBotOptParser(): | |
252 parser = bb_utils.GetParser() | |
253 parser.add_option('--bot-id', help='Specify bot id directly.') | |
254 parser.add_option('--testing', action='store_true', | |
255 help='For testing: print, but do not run commands') | |
256 | |
257 return parser | |
258 | |
259 | |
260 def GetBotConfig(options, bot_step_map): | |
261 bot_id = options.bot_id or options.factory_properties.get('android_bot_id') | |
262 if not bot_id: | |
263 print (sys.stderr, | |
264 'A bot id must be specified through option or factory_props.') | |
265 return | |
266 | |
267 bot_config = GetBestMatch(bot_step_map, bot_id) | |
268 if not bot_config: | |
269 print 'Error: config for id="%s" cannot be inferred.' % bot_id | |
270 return bot_config | |
271 | |
272 | |
273 def RunBotCommands(options, commands, env): | |
274 print 'Environment changes:' | |
275 print DictDiff(dict(os.environ), env) | |
276 | |
277 for command in commands: | |
278 print bb_utils.CommandToString(command) | |
279 sys.stdout.flush() | |
280 if options.testing: | |
281 env['BUILDBOT_TESTING'] = '1' | |
282 return_code = subprocess.call(command, cwd=bb_utils.CHROME_SRC, env=env) | |
283 if return_code != 0: | |
284 return return_code | |
285 | |
286 | |
287 def main(argv): | |
288 proc = subprocess.Popen( | |
289 ['/bin/hostname', '-f'], stdout=subprocess.PIPE, stderr=subprocess.PIPE) | |
290 hostname_stdout, hostname_stderr = proc.communicate() | |
291 if proc.returncode == 0: | |
292 print 'Running on: ' + hostname_stdout | |
293 else: | |
294 print >> sys.stderr, 'WARNING: failed to run hostname' | |
295 print >> sys.stderr, hostname_stdout | |
296 print >> sys.stderr, hostname_stderr | |
297 sys.exit(1) | |
298 | |
299 parser = GetRunBotOptParser() | |
300 options, args = parser.parse_args(argv[1:]) | |
301 if args: | |
302 parser.error('Unused args: %s' % args) | |
303 | |
304 bot_config = GetBotConfig(options, GetBotStepMap()) | |
305 if not bot_config: | |
306 sys.exit(1) | |
307 | |
308 print 'Using config:', bot_config | |
309 | |
310 commands = GetCommands(options, bot_config) | |
311 for command in commands: | |
312 print 'Will run: ', bb_utils.CommandToString(command) | |
313 print | |
314 | |
315 env = GetEnvironment(bot_config.host_obj, options.testing) | |
316 return RunBotCommands(options, commands, env) | |
317 | |
318 | |
319 if __name__ == '__main__': | |
320 sys.exit(main(sys.argv)) | |
OLD | NEW |