verify.py revision 13005
112870Sgabeblack@google.com#!/usr/bin/env python2 212870Sgabeblack@google.com# 312870Sgabeblack@google.com# Copyright 2018 Google, Inc. 412870Sgabeblack@google.com# 512870Sgabeblack@google.com# Redistribution and use in source and binary forms, with or without 612870Sgabeblack@google.com# modification, are permitted provided that the following conditions are 712870Sgabeblack@google.com# met: redistributions of source code must retain the above copyright 812870Sgabeblack@google.com# notice, this list of conditions and the following disclaimer; 912870Sgabeblack@google.com# redistributions in binary form must reproduce the above copyright 1012870Sgabeblack@google.com# notice, this list of conditions and the following disclaimer in the 1112870Sgabeblack@google.com# documentation and/or other materials provided with the distribution; 1212870Sgabeblack@google.com# neither the name of the copyright holders nor the names of its 1312870Sgabeblack@google.com# contributors may be used to endorse or promote products derived from 1412870Sgabeblack@google.com# this software without specific prior written permission. 1512870Sgabeblack@google.com# 1612870Sgabeblack@google.com# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 1712870Sgabeblack@google.com# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 1812870Sgabeblack@google.com# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 1912870Sgabeblack@google.com# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 2012870Sgabeblack@google.com# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2112870Sgabeblack@google.com# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 2212870Sgabeblack@google.com# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2312870Sgabeblack@google.com# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2412870Sgabeblack@google.com# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2512870Sgabeblack@google.com# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 2612870Sgabeblack@google.com# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 2712870Sgabeblack@google.com# 2812870Sgabeblack@google.com# Authors: Gabe Black 2912870Sgabeblack@google.com 3012870Sgabeblack@google.comfrom __future__ import print_function 3112870Sgabeblack@google.com 3212870Sgabeblack@google.comimport argparse 3313003Sgabeblack@google.comimport collections 3413003Sgabeblack@google.comimport difflib 3512870Sgabeblack@google.comimport functools 3612870Sgabeblack@google.comimport inspect 3712870Sgabeblack@google.comimport itertools 3812870Sgabeblack@google.comimport json 3913000Sgabeblack@google.comimport multiprocessing.pool 4012870Sgabeblack@google.comimport os 4113003Sgabeblack@google.comimport re 4212870Sgabeblack@google.comimport subprocess 4312870Sgabeblack@google.comimport sys 4412870Sgabeblack@google.com 4512870Sgabeblack@google.comscript_path = os.path.abspath(inspect.getfile(inspect.currentframe())) 4612870Sgabeblack@google.comscript_dir = os.path.dirname(script_path) 4712870Sgabeblack@google.comconfig_path = os.path.join(script_dir, 'config.py') 4812870Sgabeblack@google.com 4912870Sgabeblack@google.comsystemc_rel_path = 'systemc' 5012870Sgabeblack@google.comtests_rel_path = os.path.join(systemc_rel_path, 'tests') 5112870Sgabeblack@google.comjson_rel_path = os.path.join(tests_rel_path, 'tests.json') 5212870Sgabeblack@google.com 5312870Sgabeblack@google.com 5412870Sgabeblack@google.com 5512870Sgabeblack@google.comdef scons(*args): 5612870Sgabeblack@google.com args = ['scons'] + list(args) 5712870Sgabeblack@google.com subprocess.check_call(args) 5812870Sgabeblack@google.com 5912870Sgabeblack@google.com 6012870Sgabeblack@google.com 6112870Sgabeblack@google.comclass Test(object): 6212870Sgabeblack@google.com def __init__(self, target, suffix, build_dir, props): 6312870Sgabeblack@google.com self.target = target 6412870Sgabeblack@google.com self.suffix = suffix 6512870Sgabeblack@google.com self.build_dir = build_dir 6613003Sgabeblack@google.com self.props = {} 6712870Sgabeblack@google.com 6812870Sgabeblack@google.com for key, val in props.iteritems(): 6913003Sgabeblack@google.com self.set_prop(key, val) 7013003Sgabeblack@google.com 7113003Sgabeblack@google.com def set_prop(self, key, val): 7213003Sgabeblack@google.com setattr(self, key, val) 7313003Sgabeblack@google.com self.props[key] = val 7412870Sgabeblack@google.com 7512870Sgabeblack@google.com def dir(self): 7612870Sgabeblack@google.com return os.path.join(self.build_dir, tests_rel_path, self.path) 7712870Sgabeblack@google.com 7812870Sgabeblack@google.com def src_dir(self): 7912897Sgabeblack@google.com return os.path.join(script_dir, self.path) 8012870Sgabeblack@google.com 8112870Sgabeblack@google.com def golden_dir(self): 8212870Sgabeblack@google.com return os.path.join(self.src_dir(), 'golden') 8312870Sgabeblack@google.com 8412870Sgabeblack@google.com def bin(self): 8512870Sgabeblack@google.com return '.'.join([self.name, self.suffix]) 8612870Sgabeblack@google.com 8712870Sgabeblack@google.com def full_path(self): 8812870Sgabeblack@google.com return os.path.join(self.dir(), self.bin()) 8912870Sgabeblack@google.com 9012870Sgabeblack@google.com def m5out_dir(self): 9112870Sgabeblack@google.com return os.path.join(self.dir(), 'm5out.' + self.suffix) 9212870Sgabeblack@google.com 9313002Sgabeblack@google.com def returncode_file(self): 9413002Sgabeblack@google.com return os.path.join(self.m5out_dir(), 'returncode') 9513002Sgabeblack@google.com 9612870Sgabeblack@google.com 9712870Sgabeblack@google.com 9812870Sgabeblack@google.comtest_phase_classes = {} 9912870Sgabeblack@google.com 10012870Sgabeblack@google.comclass TestPhaseMeta(type): 10112870Sgabeblack@google.com def __init__(cls, name, bases, d): 10212870Sgabeblack@google.com if not d.pop('abstract', False): 10312870Sgabeblack@google.com test_phase_classes[d['name']] = cls 10412870Sgabeblack@google.com 10512870Sgabeblack@google.com super(TestPhaseMeta, cls).__init__(name, bases, d) 10612870Sgabeblack@google.com 10712870Sgabeblack@google.comclass TestPhaseBase(object): 10812870Sgabeblack@google.com __metaclass__ = TestPhaseMeta 10912870Sgabeblack@google.com abstract = True 11012870Sgabeblack@google.com 11112870Sgabeblack@google.com def __init__(self, main_args, *args): 11212870Sgabeblack@google.com self.main_args = main_args 11312870Sgabeblack@google.com self.args = args 11412870Sgabeblack@google.com 11512870Sgabeblack@google.com def __lt__(self, other): 11612870Sgabeblack@google.com return self.number < other.number 11712870Sgabeblack@google.com 11812870Sgabeblack@google.comclass CompilePhase(TestPhaseBase): 11912870Sgabeblack@google.com name = 'compile' 12012870Sgabeblack@google.com number = 1 12112870Sgabeblack@google.com 12212870Sgabeblack@google.com def run(self, tests): 12312870Sgabeblack@google.com targets = list([test.full_path() for test in tests]) 12412870Sgabeblack@google.com scons_args = list(self.args) + targets 12512870Sgabeblack@google.com scons(*scons_args) 12612870Sgabeblack@google.com 12712870Sgabeblack@google.comclass RunPhase(TestPhaseBase): 12812870Sgabeblack@google.com name = 'execute' 12912870Sgabeblack@google.com number = 2 13012870Sgabeblack@google.com 13112870Sgabeblack@google.com def run(self, tests): 13213000Sgabeblack@google.com parser = argparse.ArgumentParser() 13313000Sgabeblack@google.com parser.add_argument('--timeout', type=int, metavar='SECONDS', 13413000Sgabeblack@google.com help='Time limit for each run in seconds.', 13513000Sgabeblack@google.com default=0) 13613000Sgabeblack@google.com parser.add_argument('-j', type=int, default=1, 13713000Sgabeblack@google.com help='How many tests to run in parallel.') 13813000Sgabeblack@google.com args = parser.parse_args(self.args) 13913000Sgabeblack@google.com 14013000Sgabeblack@google.com timeout_cmd = [ 14113000Sgabeblack@google.com 'timeout', 14213000Sgabeblack@google.com '--kill-after', str(args.timeout * 2), 14313000Sgabeblack@google.com str(args.timeout) 14413000Sgabeblack@google.com ] 14513000Sgabeblack@google.com def run_test(test): 14613000Sgabeblack@google.com cmd = [] 14713000Sgabeblack@google.com if args.timeout: 14813000Sgabeblack@google.com cmd.extend(timeout_cmd) 14913000Sgabeblack@google.com cmd.extend([ 15012870Sgabeblack@google.com test.full_path(), 15112870Sgabeblack@google.com '-red', test.m5out_dir(), 15212870Sgabeblack@google.com '--listener-mode=off', 15313003Sgabeblack@google.com '--quiet', 15412870Sgabeblack@google.com config_path 15513000Sgabeblack@google.com ]) 15613002Sgabeblack@google.com # Ensure the output directory exists. 15713002Sgabeblack@google.com if not os.path.exists(test.m5out_dir()): 15813002Sgabeblack@google.com os.makedirs(test.m5out_dir()) 15913001Sgabeblack@google.com try: 16013001Sgabeblack@google.com subprocess.check_call(cmd) 16113001Sgabeblack@google.com except subprocess.CalledProcessError, error: 16213001Sgabeblack@google.com returncode = error.returncode 16313001Sgabeblack@google.com else: 16413001Sgabeblack@google.com returncode = 0 16513002Sgabeblack@google.com with open(test.returncode_file(), 'w') as rc: 16613001Sgabeblack@google.com rc.write('%d\n' % returncode) 16713000Sgabeblack@google.com 16813000Sgabeblack@google.com runnable = filter(lambda t: not t.compile_only, tests) 16913000Sgabeblack@google.com if args.j == 1: 17013000Sgabeblack@google.com map(run_test, runnable) 17113000Sgabeblack@google.com else: 17213000Sgabeblack@google.com tp = multiprocessing.pool.ThreadPool(args.j) 17313000Sgabeblack@google.com map(lambda t: tp.apply_async(run_test, (t,)), runnable) 17413000Sgabeblack@google.com tp.close() 17513000Sgabeblack@google.com tp.join() 17612870Sgabeblack@google.com 17713003Sgabeblack@google.comclass Checker(object): 17813003Sgabeblack@google.com def __init__(self, ref, test, tag): 17913003Sgabeblack@google.com self.ref = ref 18013003Sgabeblack@google.com self.test = test 18113003Sgabeblack@google.com self.tag = tag 18213003Sgabeblack@google.com 18313003Sgabeblack@google.com def check(self): 18413003Sgabeblack@google.com with open(self.text) as test_f, open(self.ref) as ref_f: 18513003Sgabeblack@google.com return test_f.read() == ref_f.read() 18613003Sgabeblack@google.com 18713003Sgabeblack@google.comclass LogChecker(Checker): 18813003Sgabeblack@google.com def merge_filts(*filts): 18913003Sgabeblack@google.com filts = map(lambda f: '(' + f + ')', filts) 19013003Sgabeblack@google.com filts = '|'.join(filts) 19113003Sgabeblack@google.com return re.compile(filts, flags=re.MULTILINE) 19213003Sgabeblack@google.com 19313003Sgabeblack@google.com ref_filt = merge_filts( 19413003Sgabeblack@google.com r'^\nInfo: /OSCI/SystemC: Simulation stopped by user.\n', 19513004Sgabeblack@google.com r'^SystemC Simulation\n', 19613004Sgabeblack@google.com r'^\nWarning: .*\nIn file: .*\n' 19713003Sgabeblack@google.com ) 19813003Sgabeblack@google.com test_filt = merge_filts( 19913003Sgabeblack@google.com r'^Global frequency set at \d* ticks per second\n' 20013003Sgabeblack@google.com ) 20113003Sgabeblack@google.com 20213003Sgabeblack@google.com def __init__(self, ref, test, tag, out_dir): 20313003Sgabeblack@google.com super(LogChecker, self).__init__(ref, test, tag) 20413003Sgabeblack@google.com self.out_dir = out_dir 20513003Sgabeblack@google.com 20613003Sgabeblack@google.com def apply_filters(self, data, filts): 20713003Sgabeblack@google.com re.sub(filt, '', data) 20813003Sgabeblack@google.com 20913003Sgabeblack@google.com def check(self): 21013003Sgabeblack@google.com test_file = os.path.basename(self.test) 21113003Sgabeblack@google.com ref_file = os.path.basename(self.ref) 21213003Sgabeblack@google.com with open(self.test) as test_f, open(self.ref) as ref_f: 21313003Sgabeblack@google.com test = re.sub(self.test_filt, '', test_f.read()) 21413003Sgabeblack@google.com ref = re.sub(self.ref_filt, '', ref_f.read()) 21513005Sgabeblack@google.com diff_file = '.'.join([ref_file, 'diff']) 21613005Sgabeblack@google.com diff_path = os.path.join(self.out_dir, diff_file) 21713003Sgabeblack@google.com if test != ref: 21813003Sgabeblack@google.com with open(diff_path, 'w') as diff_f: 21913003Sgabeblack@google.com for line in difflib.unified_diff( 22013003Sgabeblack@google.com ref.splitlines(True), test.splitlines(True), 22113003Sgabeblack@google.com fromfile=ref_file, 22213003Sgabeblack@google.com tofile=test_file): 22313003Sgabeblack@google.com diff_f.write(line) 22413003Sgabeblack@google.com return False 22513005Sgabeblack@google.com else: 22613005Sgabeblack@google.com if os.path.exists(diff_path): 22713005Sgabeblack@google.com os.unlink(diff_path) 22813003Sgabeblack@google.com return True 22913003Sgabeblack@google.com 23012870Sgabeblack@google.comclass VerifyPhase(TestPhaseBase): 23112870Sgabeblack@google.com name = 'verify' 23212870Sgabeblack@google.com number = 3 23312870Sgabeblack@google.com 23413002Sgabeblack@google.com def reset_status(self): 23513002Sgabeblack@google.com self._passed = [] 23613002Sgabeblack@google.com self._failed = {} 23713002Sgabeblack@google.com 23813002Sgabeblack@google.com def passed(self, test): 23913002Sgabeblack@google.com self._passed.append(test) 24013002Sgabeblack@google.com 24113003Sgabeblack@google.com def failed(self, test, cause, note=''): 24213003Sgabeblack@google.com test.set_prop('note', note) 24313002Sgabeblack@google.com self._failed.setdefault(cause, []).append(test) 24413002Sgabeblack@google.com 24513002Sgabeblack@google.com def print_status(self): 24613002Sgabeblack@google.com total_passed = len(self._passed) 24713002Sgabeblack@google.com total_failed = sum(map(len, self._failed.values())) 24813002Sgabeblack@google.com print() 24913002Sgabeblack@google.com print('Passed: {passed:4} - Failed: {failed:4}'.format( 25013002Sgabeblack@google.com passed=total_passed, failed=total_failed)) 25113002Sgabeblack@google.com 25213002Sgabeblack@google.com def write_result_file(self, path): 25313003Sgabeblack@google.com results = { 25413003Sgabeblack@google.com 'passed': map(lambda t: t.props, self._passed), 25513003Sgabeblack@google.com 'failed': { 25613003Sgabeblack@google.com cause: map(lambda t: t.props, tests) for 25713002Sgabeblack@google.com cause, tests in self._failed.iteritems() 25813003Sgabeblack@google.com } 25913002Sgabeblack@google.com } 26013002Sgabeblack@google.com with open(path, 'w') as rf: 26113002Sgabeblack@google.com json.dump(results, rf) 26213002Sgabeblack@google.com 26313002Sgabeblack@google.com def print_results(self): 26413002Sgabeblack@google.com print() 26513002Sgabeblack@google.com print('Passed:') 26613003Sgabeblack@google.com for path in sorted(list([ t.path for t in self._passed ])): 26713003Sgabeblack@google.com print(' ', path) 26813002Sgabeblack@google.com 26913002Sgabeblack@google.com print() 27013002Sgabeblack@google.com print('Failed:') 27113002Sgabeblack@google.com 27213003Sgabeblack@google.com causes = [] 27313003Sgabeblack@google.com for cause, tests in sorted(self._failed.items()): 27413003Sgabeblack@google.com block = ' ' + cause.capitalize() + ':\n' 27513003Sgabeblack@google.com for test in sorted(tests, key=lambda t: t.path): 27613003Sgabeblack@google.com block += ' ' + test.path 27713003Sgabeblack@google.com if test.note: 27813003Sgabeblack@google.com block += ' - ' + test.note 27913003Sgabeblack@google.com block += '\n' 28013003Sgabeblack@google.com causes.append(block) 28113002Sgabeblack@google.com 28213003Sgabeblack@google.com print('\n'.join(causes)) 28313002Sgabeblack@google.com 28412870Sgabeblack@google.com def run(self, tests): 28513002Sgabeblack@google.com parser = argparse.ArgumentParser() 28613002Sgabeblack@google.com result_opts = parser.add_mutually_exclusive_group() 28713002Sgabeblack@google.com result_opts.add_argument('--result-file', action='store_true', 28813002Sgabeblack@google.com help='Create a results.json file in the current directory.') 28913002Sgabeblack@google.com result_opts.add_argument('--result-file-at', metavar='PATH', 29013002Sgabeblack@google.com help='Create a results json file at the given path.') 29113002Sgabeblack@google.com parser.add_argument('--print-results', action='store_true', 29213002Sgabeblack@google.com help='Print a list of tests that passed or failed') 29313002Sgabeblack@google.com args = parser.parse_args(self.args) 29412870Sgabeblack@google.com 29513002Sgabeblack@google.com self.reset_status() 29613002Sgabeblack@google.com 29713002Sgabeblack@google.com runnable = filter(lambda t: not t.compile_only, tests) 29813002Sgabeblack@google.com compile_only = filter(lambda t: t.compile_only, tests) 29913002Sgabeblack@google.com 30013002Sgabeblack@google.com for test in compile_only: 30113002Sgabeblack@google.com if os.path.exists(test.full_path()): 30213002Sgabeblack@google.com self.passed(test) 30313002Sgabeblack@google.com else: 30413002Sgabeblack@google.com self.failed(test, 'compile failed') 30513002Sgabeblack@google.com 30613002Sgabeblack@google.com for test in runnable: 30713002Sgabeblack@google.com with open(test.returncode_file()) as rc: 30813002Sgabeblack@google.com returncode = int(rc.read()) 30913002Sgabeblack@google.com 31013003Sgabeblack@google.com if returncode == 124: 31113002Sgabeblack@google.com self.failed(test, 'time out') 31213003Sgabeblack@google.com continue 31313003Sgabeblack@google.com elif returncode != 0: 31413002Sgabeblack@google.com self.failed(test, 'abort') 31513003Sgabeblack@google.com continue 31613003Sgabeblack@google.com 31713003Sgabeblack@google.com out_dir = test.m5out_dir() 31813003Sgabeblack@google.com 31913003Sgabeblack@google.com Diff = collections.namedtuple( 32013003Sgabeblack@google.com 'Diff', 'ref, test, tag, ref_filter') 32113003Sgabeblack@google.com 32213003Sgabeblack@google.com diffs = [] 32313003Sgabeblack@google.com 32413003Sgabeblack@google.com log_file = '.'.join([test.name, 'log']) 32513003Sgabeblack@google.com log_path = os.path.join(test.golden_dir(), log_file) 32613003Sgabeblack@google.com simout_path = os.path.join(out_dir, 'simout') 32713003Sgabeblack@google.com if not os.path.exists(simout_path): 32813003Sgabeblack@google.com self.failed(test, 'no log output') 32913003Sgabeblack@google.com if os.path.exists(log_path): 33013003Sgabeblack@google.com diffs.append(LogChecker( 33113003Sgabeblack@google.com log_path, simout_path, log_file, out_dir)) 33213003Sgabeblack@google.com 33313003Sgabeblack@google.com failed_diffs = filter(lambda d: not d.check(), diffs) 33413003Sgabeblack@google.com if failed_diffs: 33513003Sgabeblack@google.com tags = map(lambda d: d.tag, failed_diffs) 33613003Sgabeblack@google.com self.failed(test, 'failed diffs', ' '.join(tags)) 33713003Sgabeblack@google.com continue 33813003Sgabeblack@google.com 33913003Sgabeblack@google.com self.passed(test) 34013002Sgabeblack@google.com 34113002Sgabeblack@google.com if args.print_results: 34213002Sgabeblack@google.com self.print_results() 34313002Sgabeblack@google.com 34413002Sgabeblack@google.com self.print_status() 34513002Sgabeblack@google.com 34613002Sgabeblack@google.com result_path = None 34713002Sgabeblack@google.com if args.result_file: 34813002Sgabeblack@google.com result_path = os.path.join(os.getcwd(), 'results.json') 34913002Sgabeblack@google.com elif args.result_file_at: 35013002Sgabeblack@google.com result_path = args.result_file_at 35113002Sgabeblack@google.com 35213002Sgabeblack@google.com if result_path: 35313002Sgabeblack@google.com self.write_result_file(result_path) 35412870Sgabeblack@google.com 35512870Sgabeblack@google.com 35612870Sgabeblack@google.comparser = argparse.ArgumentParser(description='SystemC test utility') 35712870Sgabeblack@google.com 35812870Sgabeblack@google.comparser.add_argument('build_dir', metavar='BUILD_DIR', 35912870Sgabeblack@google.com help='The build directory (ie. build/ARM).') 36012870Sgabeblack@google.com 36112870Sgabeblack@google.comparser.add_argument('--update-json', action='store_true', 36212870Sgabeblack@google.com help='Update the json manifest of tests.') 36312870Sgabeblack@google.com 36412870Sgabeblack@google.comparser.add_argument('--flavor', choices=['debug', 'opt', 'fast'], 36512870Sgabeblack@google.com default='opt', 36612870Sgabeblack@google.com help='Flavor of binary to test.') 36712870Sgabeblack@google.com 36812870Sgabeblack@google.comparser.add_argument('--list', action='store_true', 36912870Sgabeblack@google.com help='List the available tests') 37012870Sgabeblack@google.com 37112903Sgabeblack@google.comfilter_opts = parser.add_mutually_exclusive_group() 37212903Sgabeblack@google.comfilter_opts.add_argument('--filter', default='True', 37312903Sgabeblack@google.com help='Python expression which filters tests based ' 37412903Sgabeblack@google.com 'on their properties') 37512903Sgabeblack@google.comfilter_opts.add_argument('--filter-file', default=None, 37612903Sgabeblack@google.com type=argparse.FileType('r'), 37712903Sgabeblack@google.com help='Same as --filter, but read from a file') 37812870Sgabeblack@google.com 37912870Sgabeblack@google.comdef collect_phases(args): 38012870Sgabeblack@google.com phase_groups = [list(g) for k, g in 38112870Sgabeblack@google.com itertools.groupby(args, lambda x: x != '--phase') if k] 38212870Sgabeblack@google.com main_args = parser.parse_args(phase_groups[0][1:]) 38312870Sgabeblack@google.com phases = [] 38412870Sgabeblack@google.com names = [] 38512870Sgabeblack@google.com for group in phase_groups[1:]: 38612870Sgabeblack@google.com name = group[0] 38712870Sgabeblack@google.com if name in names: 38812870Sgabeblack@google.com raise RuntimeException('Phase %s specified more than once' % name) 38912870Sgabeblack@google.com phase = test_phase_classes[name] 39012870Sgabeblack@google.com phases.append(phase(main_args, *group[1:])) 39112870Sgabeblack@google.com phases.sort() 39212870Sgabeblack@google.com return main_args, phases 39312870Sgabeblack@google.com 39412870Sgabeblack@google.commain_args, phases = collect_phases(sys.argv) 39512870Sgabeblack@google.com 39612870Sgabeblack@google.comif len(phases) == 0: 39712870Sgabeblack@google.com phases = [ 39812870Sgabeblack@google.com CompilePhase(main_args), 39912870Sgabeblack@google.com RunPhase(main_args), 40012870Sgabeblack@google.com VerifyPhase(main_args) 40112870Sgabeblack@google.com ] 40212870Sgabeblack@google.com 40312870Sgabeblack@google.com 40412870Sgabeblack@google.com 40512870Sgabeblack@google.comjson_path = os.path.join(main_args.build_dir, json_rel_path) 40612870Sgabeblack@google.com 40712870Sgabeblack@google.comif main_args.update_json: 40812870Sgabeblack@google.com scons(os.path.join(json_path)) 40912870Sgabeblack@google.com 41012870Sgabeblack@google.comwith open(json_path) as f: 41112870Sgabeblack@google.com test_data = json.load(f) 41212870Sgabeblack@google.com 41312903Sgabeblack@google.com if main_args.filter_file: 41412903Sgabeblack@google.com f = main_args.filter_file 41512903Sgabeblack@google.com filt = compile(f.read(), f.name, 'eval') 41612903Sgabeblack@google.com else: 41712903Sgabeblack@google.com filt = compile(main_args.filter, '<string>', 'eval') 41812903Sgabeblack@google.com 41912903Sgabeblack@google.com filtered_tests = { 42012903Sgabeblack@google.com target: props for (target, props) in 42112903Sgabeblack@google.com test_data.iteritems() if eval(filt, dict(props)) 42212903Sgabeblack@google.com } 42312903Sgabeblack@google.com 42412870Sgabeblack@google.com if main_args.list: 42512903Sgabeblack@google.com for target, props in sorted(filtered_tests.iteritems()): 42612870Sgabeblack@google.com print('%s.%s' % (target, main_args.flavor)) 42712870Sgabeblack@google.com for key, val in props.iteritems(): 42812870Sgabeblack@google.com print(' %s: %s' % (key, val)) 42912937Sgabeblack@google.com print('Total tests: %d' % len(filtered_tests)) 43012870Sgabeblack@google.com else: 43112903Sgabeblack@google.com tests_to_run = list([ 43212903Sgabeblack@google.com Test(target, main_args.flavor, main_args.build_dir, props) for 43312903Sgabeblack@google.com target, props in sorted(filtered_tests.iteritems()) 43412903Sgabeblack@google.com ]) 43512870Sgabeblack@google.com 43612870Sgabeblack@google.com for phase in phases: 43712870Sgabeblack@google.com phase.run(tests_to_run) 438