blob: bf21e87f31cd294076450d885fcd92dd13e721b2 [file] [log] [blame]
senorblanco@chromium.org782f3b42012-10-29 18:06:26 +00001#!/usr/bin/python
2
3'''
4Copyright 2012 Google Inc.
5
6Use of this source code is governed by a BSD-style license that can be
7found in the LICENSE file.
8'''
9
10'''
senorblanco@chromium.org123a0b52012-11-29 21:50:34 +000011Rebaselines the given GM tests, on all bots and all configurations.
senorblanco@chromium.org782f3b42012-10-29 18:06:26 +000012'''
13
epoger@google.com99ba65a2013-06-05 15:43:37 +000014# System-level imports
epoger@google.com9166bf52013-05-30 15:46:19 +000015import argparse
epoger@google.comfd040112013-08-20 16:21:55 +000016import json
epoger@google.comec3397b2013-05-29 17:09:43 +000017import os
epoger@google.come78d2072013-06-12 17:44:14 +000018import re
epoger@google.com27e1c002013-07-24 15:38:39 +000019import subprocess
epoger@google.comec3397b2013-05-29 17:09:43 +000020import sys
epoger@google.com99ba65a2013-06-05 15:43:37 +000021import urllib2
22
23# Imports from within Skia
24#
epoger@google.comdad53102013-06-12 14:25:30 +000025# We need to add the 'gm' directory, so that we can import gm_json.py within
26# that directory. That script allows us to parse the actual-results.json file
27# written out by the GM tool.
28# Make sure that the 'gm' dir is in the PYTHONPATH, but add it at the *end*
29# so any dirs that are already in the PYTHONPATH will be preferred.
30#
31# This assumes that the 'gm' directory has been checked out as a sibling of
32# the 'tools' directory containing this script, which will be the case if
33# 'trunk' was checked out as a single unit.
epoger@google.com99ba65a2013-06-05 15:43:37 +000034GM_DIRECTORY = os.path.realpath(
35 os.path.join(os.path.dirname(os.path.dirname(__file__)), 'gm'))
36if GM_DIRECTORY not in sys.path:
epoger@google.com2a192a82013-08-02 20:54:46 +000037 sys.path.append(GM_DIRECTORY)
epoger@google.com99ba65a2013-06-05 15:43:37 +000038import gm_json
39
epoger@google.com44f165f2013-08-24 20:45:31 +000040# TODO(epoger): In the long run, we want to build this list automatically,
41# but for now we hard-code it until we can properly address
42# https://code.google.com/p/skia/issues/detail?id=1544
43# ('live query of builder list makes rebaseline.py slow to start up')
44TEST_BUILDERS = [
45 'Test-Android-GalaxyNexus-SGX540-Arm7-Debug',
46 'Test-Android-GalaxyNexus-SGX540-Arm7-Release',
47 'Test-Android-IntelRhb-SGX544-x86-Debug',
48 'Test-Android-IntelRhb-SGX544-x86-Release',
49 'Test-Android-Nexus10-MaliT604-Arm7-Debug',
50 'Test-Android-Nexus10-MaliT604-Arm7-Release',
51 'Test-Android-Nexus4-Adreno320-Arm7-Debug',
52 'Test-Android-Nexus4-Adreno320-Arm7-Release',
53 'Test-Android-Nexus7-Tegra3-Arm7-Debug',
54 'Test-Android-Nexus7-Tegra3-Arm7-Release',
55 'Test-Android-NexusS-SGX540-Arm7-Debug',
56 'Test-Android-NexusS-SGX540-Arm7-Release',
57 'Test-Android-Xoom-Tegra2-Arm7-Debug',
58 'Test-Android-Xoom-Tegra2-Arm7-Release',
59 'Test-ChromeOS-Alex-GMA3150-x86-Debug',
60 'Test-ChromeOS-Alex-GMA3150-x86-Release',
61 'Test-ChromeOS-Daisy-MaliT604-Arm7-Debug',
62 'Test-ChromeOS-Daisy-MaliT604-Arm7-Release',
63 'Test-ChromeOS-Link-HD4000-x86_64-Debug',
64 'Test-ChromeOS-Link-HD4000-x86_64-Release',
65 'Test-Mac10.6-MacMini4.1-GeForce320M-x86-Debug',
66 'Test-Mac10.6-MacMini4.1-GeForce320M-x86-Release',
67 'Test-Mac10.6-MacMini4.1-GeForce320M-x86_64-Debug',
68 'Test-Mac10.6-MacMini4.1-GeForce320M-x86_64-Release',
69 'Test-Mac10.7-MacMini4.1-GeForce320M-x86-Debug',
70 'Test-Mac10.7-MacMini4.1-GeForce320M-x86-Release',
71 'Test-Mac10.7-MacMini4.1-GeForce320M-x86_64-Debug',
72 'Test-Mac10.7-MacMini4.1-GeForce320M-x86_64-Release',
73 'Test-Mac10.8-MacMini4.1-GeForce320M-x86-Debug',
74 'Test-Mac10.8-MacMini4.1-GeForce320M-x86-Release',
75 'Test-Mac10.8-MacMini4.1-GeForce320M-x86_64-Debug',
76 'Test-Mac10.8-MacMini4.1-GeForce320M-x86_64-Release',
77 'Test-Ubuntu12-ShuttleA-ATI5770-x86-Debug',
78 'Test-Ubuntu12-ShuttleA-ATI5770-x86-Release',
79 'Test-Ubuntu12-ShuttleA-ATI5770-x86_64-Debug',
80 'Test-Ubuntu12-ShuttleA-ATI5770-x86_64-Release',
81 'Test-Ubuntu12-ShuttleA-HD2000-x86_64-Release-Valgrind',
82 'Test-Ubuntu12-ShuttleA-NoGPU-x86_64-Debug',
83 'Test-Ubuntu13-ShuttleA-HD2000-x86_64-Debug-ASAN',
84 'Test-Win7-ShuttleA-HD2000-x86-Debug',
85 'Test-Win7-ShuttleA-HD2000-x86-Debug-ANGLE',
86 'Test-Win7-ShuttleA-HD2000-x86-Debug-DirectWrite',
87 'Test-Win7-ShuttleA-HD2000-x86-Release',
88 'Test-Win7-ShuttleA-HD2000-x86-Release-ANGLE',
89 'Test-Win7-ShuttleA-HD2000-x86-Release-DirectWrite',
90 'Test-Win7-ShuttleA-HD2000-x86_64-Debug',
91 'Test-Win7-ShuttleA-HD2000-x86_64-Release',
92]
epoger@google.com9166bf52013-05-30 15:46:19 +000093
epoger@google.com66ba9f92013-07-11 19:20:30 +000094class _InternalException(Exception):
epoger@google.com2a192a82013-08-02 20:54:46 +000095 pass
epoger@google.comdb29a312013-06-04 14:58:47 +000096
epoger@google.comffcbdbf2013-07-16 17:35:39 +000097# Object that handles exceptions, either raising them immediately or collecting
98# them to display later on.
99class ExceptionHandler(object):
100
epoger@google.com2a192a82013-08-02 20:54:46 +0000101 # params:
102 # keep_going_on_failure: if False, report failures and quit right away;
103 # if True, collect failures until
104 # ReportAllFailures() is called
105 def __init__(self, keep_going_on_failure=False):
106 self._keep_going_on_failure = keep_going_on_failure
107 self._failures_encountered = []
108 self._exiting = False
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000109
epoger@google.com2a192a82013-08-02 20:54:46 +0000110 # Exit the program with the given status value.
111 def _Exit(self, status=1):
112 self._exiting = True
113 sys.exit(status)
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000114
epoger@google.com2a192a82013-08-02 20:54:46 +0000115 # We have encountered an exception; either collect the info and keep going,
116 # or exit the program right away.
117 def RaiseExceptionOrContinue(self, e):
118 # If we are already quitting the program, propagate any exceptions
119 # so that the proper exit status will be communicated to the shell.
120 if self._exiting:
121 raise e
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000122
epoger@google.com2a192a82013-08-02 20:54:46 +0000123 if self._keep_going_on_failure:
124 print >> sys.stderr, 'WARNING: swallowing exception %s' % e
125 self._failures_encountered.append(e)
126 else:
127 print >> sys.stderr, e
128 print >> sys.stderr, (
129 'Halting at first exception; to keep going, re-run ' +
130 'with the --keep-going-on-failure option set.')
131 self._Exit()
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000132
epoger@google.com2a192a82013-08-02 20:54:46 +0000133 def ReportAllFailures(self):
134 if self._failures_encountered:
135 print >> sys.stderr, ('Encountered %d failures (see above).' %
136 len(self._failures_encountered))
137 self._Exit()
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000138
139
epoger@google.com99a8ec92013-06-19 18:56:59 +0000140# Object that rebaselines a JSON expectations file (not individual image files).
epoger@google.com99a8ec92013-06-19 18:56:59 +0000141class JsonRebaseliner(object):
epoger@google.com9166bf52013-05-30 15:46:19 +0000142
epoger@google.com2a192a82013-08-02 20:54:46 +0000143 # params:
144 # expectations_root: root directory of all expectations JSON files
145 # expectations_input_filename: filename (under expectations_root) of JSON
146 # expectations file to read; typically
147 # "expected-results.json"
148 # expectations_output_filename: filename (under expectations_root) to
149 # which updated expectations should be
150 # written; typically the same as
151 # expectations_input_filename, to overwrite
152 # the old content
153 # actuals_base_url: base URL from which to read actual-result JSON files
154 # actuals_filename: filename (under actuals_base_url) from which to read a
155 # summary of results; typically "actual-results.json"
156 # exception_handler: reference to rebaseline.ExceptionHandler object
157 # tests: list of tests to rebaseline, or None if we should rebaseline
158 # whatever files the JSON results summary file tells us to
159 # configs: which configs to run for each test, or None if we should
160 # rebaseline whatever configs the JSON results summary file tells
161 # us to
162 # add_new: if True, add expectations for tests which don't have any yet
epoger@google.com06e626d2013-09-03 17:32:15 +0000163 # bugs: optional list of bug numbers which pertain to these expectations
164 # notes: free-form text notes to add to all updated expectations
165 # mark_unreviewed: if True, mark these expectations as NOT having been
166 # reviewed by a human; otherwise, leave that field blank.
167 # Currently, there is no way to make this script mark
168 # expectations as reviewed-by-human=True.
169 # TODO(epoger): Add that capability to a review tool.
epoger@google.com2a192a82013-08-02 20:54:46 +0000170 def __init__(self, expectations_root, expectations_input_filename,
171 expectations_output_filename, actuals_base_url,
172 actuals_filename, exception_handler,
epoger@google.com06e626d2013-09-03 17:32:15 +0000173 tests=None, configs=None, add_new=False, bugs=None, notes=None,
174 mark_unreviewed=None):
epoger@google.com2a192a82013-08-02 20:54:46 +0000175 self._expectations_root = expectations_root
176 self._expectations_input_filename = expectations_input_filename
177 self._expectations_output_filename = expectations_output_filename
178 self._tests = tests
179 self._configs = configs
180 self._actuals_base_url = actuals_base_url
181 self._actuals_filename = actuals_filename
182 self._exception_handler = exception_handler
183 self._add_new = add_new
epoger@google.com06e626d2013-09-03 17:32:15 +0000184 self._bugs = bugs
185 self._notes = notes
186 self._mark_unreviewed = mark_unreviewed
epoger@google.com2a192a82013-08-02 20:54:46 +0000187 self._image_filename_re = re.compile(gm_json.IMAGE_FILENAME_PATTERN)
188 self._using_svn = os.path.isdir(os.path.join(expectations_root, '.svn'))
epoger@google.com27e1c002013-07-24 15:38:39 +0000189
epoger@google.com2a192a82013-08-02 20:54:46 +0000190 # Executes subprocess.call(cmd).
191 # Raises an Exception if the command fails.
192 def _Call(self, cmd):
193 if subprocess.call(cmd) != 0:
194 raise _InternalException('error running command: ' + ' '.join(cmd))
epoger@google.com9166bf52013-05-30 15:46:19 +0000195
epoger@google.com2a192a82013-08-02 20:54:46 +0000196 # Returns the full contents of filepath, as a single string.
197 # If filepath looks like a URL, try to read it that way instead of as
198 # a path on local storage.
199 #
200 # Raises _InternalException if there is a problem.
201 def _GetFileContents(self, filepath):
202 if filepath.startswith('http:') or filepath.startswith('https:'):
203 try:
204 return urllib2.urlopen(filepath).read()
205 except urllib2.HTTPError as e:
206 raise _InternalException('unable to read URL %s: %s' % (
207 filepath, e))
208 else:
209 return open(filepath, 'r').read()
epoger@google.com99ba65a2013-06-05 15:43:37 +0000210
epoger@google.com2a192a82013-08-02 20:54:46 +0000211 # Returns a dictionary of actual results from actual-results.json file.
212 #
213 # The dictionary returned has this format:
214 # {
215 # u'imageblur_565.png': [u'bitmap-64bitMD5', 3359963596899141322],
216 # u'imageblur_8888.png': [u'bitmap-64bitMD5', 4217923806027861152],
217 # u'shadertext3_8888.png': [u'bitmap-64bitMD5', 3713708307125704716]
218 # }
219 #
220 # If the JSON actual result summary file cannot be loaded, logs a warning
221 # message and returns None.
222 # If the JSON actual result summary file can be loaded, but we have
223 # trouble parsing it, raises an Exception.
224 #
225 # params:
226 # json_url: URL pointing to a JSON actual result summary file
227 # sections: a list of section names to include in the results, e.g.
228 # [gm_json.JSONKEY_ACTUALRESULTS_FAILED,
229 # gm_json.JSONKEY_ACTUALRESULTS_NOCOMPARISON] ;
230 # if None, then include ALL sections.
231 def _GetActualResults(self, json_url, sections=None):
232 try:
233 json_contents = self._GetFileContents(json_url)
234 except _InternalException:
235 print >> sys.stderr, (
236 'could not read json_url %s ; skipping this platform.' %
237 json_url)
238 return None
239 json_dict = gm_json.LoadFromString(json_contents)
240 results_to_return = {}
241 actual_results = json_dict[gm_json.JSONKEY_ACTUALRESULTS]
242 if not sections:
243 sections = actual_results.keys()
244 for section in sections:
245 section_results = actual_results[section]
246 if section_results:
247 results_to_return.update(section_results)
248 return results_to_return
epoger@google.come78d2072013-06-12 17:44:14 +0000249
epoger@google.com2a192a82013-08-02 20:54:46 +0000250 # Rebaseline all tests/types we specified in the constructor,
epoger@google.comfd040112013-08-20 16:21:55 +0000251 # within this builder's subdirectory in expectations/gm .
epoger@google.com2a192a82013-08-02 20:54:46 +0000252 #
253 # params:
epoger@google.com2a192a82013-08-02 20:54:46 +0000254 # builder : e.g. 'Test-Win7-ShuttleA-HD2000-x86-Release'
epoger@google.comfd040112013-08-20 16:21:55 +0000255 def RebaselineSubdir(self, builder):
epoger@google.com2a192a82013-08-02 20:54:46 +0000256 # Read in the actual result summary, and extract all the tests whose
257 # results we need to update.
258 actuals_url = '/'.join([self._actuals_base_url,
epoger@google.comfd040112013-08-20 16:21:55 +0000259 builder, self._actuals_filename])
epoger@google.com06e626d2013-09-03 17:32:15 +0000260 # Only update results for tests that are currently failing.
261 # We don't want to rewrite results for tests that are already succeeding,
262 # because we don't want to add annotation fields (such as
263 # JSONKEY_EXPECTEDRESULTS_BUGS) except for tests whose expectations we
264 # are actually modifying.
265 sections = [gm_json.JSONKEY_ACTUALRESULTS_FAILED]
epoger@google.com2a192a82013-08-02 20:54:46 +0000266 if self._add_new:
267 sections.append(gm_json.JSONKEY_ACTUALRESULTS_NOCOMPARISON)
268 results_to_update = self._GetActualResults(json_url=actuals_url,
269 sections=sections)
epoger@google.come78d2072013-06-12 17:44:14 +0000270
epoger@google.com2a192a82013-08-02 20:54:46 +0000271 # Read in current expectations.
272 expectations_input_filepath = os.path.join(
epoger@google.comfd040112013-08-20 16:21:55 +0000273 self._expectations_root, builder, self._expectations_input_filename)
epoger@google.com2a192a82013-08-02 20:54:46 +0000274 expectations_dict = gm_json.LoadFromFile(expectations_input_filepath)
275 expected_results = expectations_dict[gm_json.JSONKEY_EXPECTEDRESULTS]
epoger@google.coma783f2b2013-07-08 17:51:58 +0000276
epoger@google.com2a192a82013-08-02 20:54:46 +0000277 # Update the expectations in memory, skipping any tests/configs that
278 # the caller asked to exclude.
279 skipped_images = []
280 if results_to_update:
281 for (image_name, image_results) in results_to_update.iteritems():
282 (test, config) = self._image_filename_re.match(image_name).groups()
283 if self._tests:
284 if test not in self._tests:
285 skipped_images.append(image_name)
286 continue
287 if self._configs:
288 if config not in self._configs:
289 skipped_images.append(image_name)
290 continue
291 if not expected_results.get(image_name):
292 expected_results[image_name] = {}
epoger@google.com06e626d2013-09-03 17:32:15 +0000293 expected_results[image_name]\
294 [gm_json.JSONKEY_EXPECTEDRESULTS_ALLOWEDDIGESTS]\
295 = [image_results]
296 if self._mark_unreviewed:
297 expected_results[image_name]\
298 [gm_json.JSONKEY_EXPECTEDRESULTS_REVIEWED]\
299 = False
300 if self._bugs:
301 expected_results[image_name]\
302 [gm_json.JSONKEY_EXPECTEDRESULTS_BUGS]\
303 = self._bugs
304 if self._notes:
305 expected_results[image_name]\
306 [gm_json.JSONKEY_EXPECTEDRESULTS_NOTES]\
307 = self._notes
epoger@google.coma783f2b2013-07-08 17:51:58 +0000308
epoger@google.com2a192a82013-08-02 20:54:46 +0000309 # Write out updated expectations.
310 expectations_output_filepath = os.path.join(
epoger@google.comfd040112013-08-20 16:21:55 +0000311 self._expectations_root, builder, self._expectations_output_filename)
epoger@google.com2a192a82013-08-02 20:54:46 +0000312 gm_json.WriteToFile(expectations_dict, expectations_output_filepath)
epoger@google.coma783f2b2013-07-08 17:51:58 +0000313
epoger@google.com2a192a82013-08-02 20:54:46 +0000314 # Mark the JSON file as plaintext, so text-style diffs can be applied.
315 # Fixes https://code.google.com/p/skia/issues/detail?id=1442
316 if self._using_svn:
317 self._Call(['svn', 'propset', '--quiet', 'svn:mime-type',
318 'text/x-json', expectations_output_filepath])
epoger@google.comec3397b2013-05-29 17:09:43 +0000319
epoger@google.com9166bf52013-05-30 15:46:19 +0000320# main...
epoger@google.comec3397b2013-05-29 17:09:43 +0000321
epoger@google.com9166bf52013-05-30 15:46:19 +0000322parser = argparse.ArgumentParser()
epoger@google.coma783f2b2013-07-08 17:51:58 +0000323parser.add_argument('--actuals-base-url',
epoger@google.com06e626d2013-09-03 17:32:15 +0000324 help=('base URL from which to read files containing JSON '
325 'summaries of actual GM results; defaults to '
326 '%(default)s'),
epoger@google.coma783f2b2013-07-08 17:51:58 +0000327 default='http://skia-autogen.googlecode.com/svn/gm-actual')
328parser.add_argument('--actuals-filename',
epoger@google.com06e626d2013-09-03 17:32:15 +0000329 help=('filename (within builder-specific subdirectories '
330 'of ACTUALS_BASE_URL) to read a summary of results '
331 'from; defaults to %(default)s'),
epoger@google.coma783f2b2013-07-08 17:51:58 +0000332 default='actual-results.json')
333# TODO(epoger): Add test that exercises --add-new argument.
epoger@google.comdad53102013-06-12 14:25:30 +0000334parser.add_argument('--add-new', action='store_true',
epoger@google.com06e626d2013-09-03 17:32:15 +0000335 help=('in addition to the standard behavior of '
336 'updating expectations for failing tests, add '
337 'expectations for tests which don\'t have '
338 'expectations yet.'))
339parser.add_argument('--bugs', metavar='BUG', type=int, nargs='+',
340 help=('Skia bug numbers (under '
341 'https://code.google.com/p/skia/issues/list ) which '
342 'pertain to this set of rebaselines.'))
epoger@google.comfd040112013-08-20 16:21:55 +0000343parser.add_argument('--builders', metavar='BUILDER', nargs='+',
epoger@google.com06e626d2013-09-03 17:32:15 +0000344 help=('which platforms to rebaseline; '
345 'if unspecified, rebaseline all platforms, same as '
346 '"--builders %s"' % ' '.join(sorted(TEST_BUILDERS))))
epoger@google.coma783f2b2013-07-08 17:51:58 +0000347# TODO(epoger): Add test that exercises --configs argument.
epoger@google.com9166bf52013-05-30 15:46:19 +0000348parser.add_argument('--configs', metavar='CONFIG', nargs='+',
epoger@google.com06e626d2013-09-03 17:32:15 +0000349 help=('which configurations to rebaseline, e.g. '
350 '"--configs 565 8888", as a filter over the full set '
351 'of results in ACTUALS_FILENAME; if unspecified, '
352 'rebaseline *all* configs that are available.'))
epoger@google.coma783f2b2013-07-08 17:51:58 +0000353parser.add_argument('--expectations-filename',
epoger@google.com06e626d2013-09-03 17:32:15 +0000354 help=('filename (under EXPECTATIONS_ROOT) to read '
355 'current expectations from, and to write new '
356 'expectations into (unless a separate '
357 'EXPECTATIONS_FILENAME_OUTPUT has been specified); '
358 'defaults to %(default)s'),
epoger@google.coma783f2b2013-07-08 17:51:58 +0000359 default='expected-results.json')
epoger@google.comc60e7452013-07-24 19:36:51 +0000360parser.add_argument('--expectations-filename-output',
epoger@google.com06e626d2013-09-03 17:32:15 +0000361 help=('filename (under EXPECTATIONS_ROOT) to write '
362 'updated expectations into; by default, overwrites '
363 'the input file (EXPECTATIONS_FILENAME)'),
epoger@google.comc60e7452013-07-24 19:36:51 +0000364 default='')
epoger@google.com99a8ec92013-06-19 18:56:59 +0000365parser.add_argument('--expectations-root',
epoger@google.com06e626d2013-09-03 17:32:15 +0000366 help=('root of expectations directory to update-- should '
367 'contain one or more builder subdirectories. '
368 'Defaults to %(default)s'),
epoger@google.come94a7d22013-07-23 19:37:03 +0000369 default=os.path.join('expectations', 'gm'))
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000370parser.add_argument('--keep-going-on-failure', action='store_true',
epoger@google.com06e626d2013-09-03 17:32:15 +0000371 help=('instead of halting at the first error encountered, '
372 'keep going and rebaseline as many tests as '
373 'possible, and then report the full set of errors '
374 'at the end'))
375parser.add_argument('--notes',
376 help=('free-form text notes to add to all updated '
377 'expectations'))
epoger@google.coma783f2b2013-07-08 17:51:58 +0000378# TODO(epoger): Add test that exercises --tests argument.
epoger@google.com99ba65a2013-06-05 15:43:37 +0000379parser.add_argument('--tests', metavar='TEST', nargs='+',
epoger@google.com06e626d2013-09-03 17:32:15 +0000380 help=('which tests to rebaseline, e.g. '
381 '"--tests aaclip bigmatrix", as a filter over the '
382 'full set of results in ACTUALS_FILENAME; if '
383 'unspecified, rebaseline *all* tests that are '
384 'available.'))
385parser.add_argument('--unreviewed', action='store_true',
386 help=('mark all expectations modified by this run as '
387 '"%s": False' %
388 gm_json.JSONKEY_EXPECTEDRESULTS_REVIEWED))
epoger@google.com9166bf52013-05-30 15:46:19 +0000389args = parser.parse_args()
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000390exception_handler = ExceptionHandler(
391 keep_going_on_failure=args.keep_going_on_failure)
epoger@google.comfd040112013-08-20 16:21:55 +0000392if args.builders:
393 builders = args.builders
epoger@google.com2a192a82013-08-02 20:54:46 +0000394 missing_json_is_fatal = True
epoger@google.com99a8ec92013-06-19 18:56:59 +0000395else:
epoger@google.comfd040112013-08-20 16:21:55 +0000396 builders = sorted(TEST_BUILDERS)
epoger@google.com2a192a82013-08-02 20:54:46 +0000397 missing_json_is_fatal = False
epoger@google.comfd040112013-08-20 16:21:55 +0000398for builder in builders:
399 if not builder in TEST_BUILDERS:
400 raise Exception(('unrecognized builder "%s"; ' +
epoger@google.com2a192a82013-08-02 20:54:46 +0000401 'should be one of %s') % (
epoger@google.comfd040112013-08-20 16:21:55 +0000402 builder, TEST_BUILDERS))
epoger@google.com99a8ec92013-06-19 18:56:59 +0000403
epoger@google.comfd040112013-08-20 16:21:55 +0000404 expectations_json_file = os.path.join(args.expectations_root, builder,
epoger@google.com2a192a82013-08-02 20:54:46 +0000405 args.expectations_filename)
406 if os.path.isfile(expectations_json_file):
407 rebaseliner = JsonRebaseliner(
408 expectations_root=args.expectations_root,
409 expectations_input_filename=args.expectations_filename,
410 expectations_output_filename=(args.expectations_filename_output or
411 args.expectations_filename),
412 tests=args.tests, configs=args.configs,
413 actuals_base_url=args.actuals_base_url,
414 actuals_filename=args.actuals_filename,
415 exception_handler=exception_handler,
epoger@google.com06e626d2013-09-03 17:32:15 +0000416 add_new=args.add_new, bugs=args.bugs, notes=args.notes,
417 mark_unreviewed=args.unreviewed)
epoger@google.com3e7399f2013-07-10 17:23:47 +0000418 try:
epoger@google.comfd040112013-08-20 16:21:55 +0000419 rebaseliner.RebaselineSubdir(builder=builder)
epoger@google.com3e7399f2013-07-10 17:23:47 +0000420 except BaseException as e:
epoger@google.com2a192a82013-08-02 20:54:46 +0000421 exception_handler.RaiseExceptionOrContinue(e)
422 else:
423 exception_handler.RaiseExceptionOrContinue(_InternalException(
424 'expectations_json_file %s not found' % expectations_json_file))
epoger@google.comffcbdbf2013-07-16 17:35:39 +0000425
426exception_handler.ReportAllFailures()