|
|
@ -11,15 +11,44 @@ import time |
|
|
|
import jobset |
|
|
|
import jobset |
|
|
|
import watch_dirs |
|
|
|
import watch_dirs |
|
|
|
|
|
|
|
|
|
|
|
# flags required for make for each configuration |
|
|
|
# SimpleConfig: just compile with CONFIG=config, and run the binary to test |
|
|
|
_CONFIGS = ['dbg', 'opt', 'tsan', 'msan', 'asan'] |
|
|
|
class SimpleConfig(object): |
|
|
|
|
|
|
|
def __init__(self, config): |
|
|
|
|
|
|
|
self.build_config = config |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def run_command(self, binary): |
|
|
|
|
|
|
|
return [binary] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ValgrindConfig: compile with some CONFIG=config, but use valgrind to run |
|
|
|
|
|
|
|
class ValgrindConfig(object): |
|
|
|
|
|
|
|
def __init__(self, config): |
|
|
|
|
|
|
|
self.build_config = config |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def run_command(self, binary): |
|
|
|
|
|
|
|
return ['valgrind', binary] |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# different configurations we can run under |
|
|
|
|
|
|
|
_CONFIGS = { |
|
|
|
|
|
|
|
'dbg': SimpleConfig('dbg'), |
|
|
|
|
|
|
|
'opt': SimpleConfig('opt'), |
|
|
|
|
|
|
|
'tsan': SimpleConfig('tsan'), |
|
|
|
|
|
|
|
'msan': SimpleConfig('msan'), |
|
|
|
|
|
|
|
'asan': SimpleConfig('asan'), |
|
|
|
|
|
|
|
'valgrind': ValgrindConfig('dbg'), |
|
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
_DEFAULT = ['dbg', 'opt'] |
|
|
|
|
|
|
|
_MAKE_TEST_TARGETS = ['buildtests_c', 'buildtests_cxx'] |
|
|
|
|
|
|
|
|
|
|
|
# parse command line |
|
|
|
# parse command line |
|
|
|
argp = argparse.ArgumentParser(description='Run grpc tests.') |
|
|
|
argp = argparse.ArgumentParser(description='Run grpc tests.') |
|
|
|
argp.add_argument('-c', '--config', |
|
|
|
argp.add_argument('-c', '--config', |
|
|
|
choices=['all'] + _CONFIGS, |
|
|
|
choices=['all'] + sorted(_CONFIGS.keys()), |
|
|
|
nargs='+', |
|
|
|
nargs='+', |
|
|
|
default=['all']) |
|
|
|
default=_DEFAULT) |
|
|
|
argp.add_argument('-t', '--test-filter', nargs='*', default=['*']) |
|
|
|
argp.add_argument('-t', '--test-filter', nargs='*', default=['*']) |
|
|
|
argp.add_argument('-n', '--runs_per_test', default=1, type=int) |
|
|
|
argp.add_argument('-n', '--runs_per_test', default=1, type=int) |
|
|
|
argp.add_argument('-f', '--forever', |
|
|
|
argp.add_argument('-f', '--forever', |
|
|
@ -29,10 +58,11 @@ argp.add_argument('-f', '--forever', |
|
|
|
args = argp.parse_args() |
|
|
|
args = argp.parse_args() |
|
|
|
|
|
|
|
|
|
|
|
# grab config |
|
|
|
# grab config |
|
|
|
configs = [cfg |
|
|
|
run_configs = set(_CONFIGS[cfg] |
|
|
|
for cfg in itertools.chain.from_iterable( |
|
|
|
for cfg in itertools.chain.from_iterable( |
|
|
|
_CONFIGS if x == 'all' else [x] |
|
|
|
_CONFIGS.iterkeys() if x == 'all' else [x] |
|
|
|
for x in args.config)] |
|
|
|
for x in args.config)) |
|
|
|
|
|
|
|
build_configs = set(cfg.build_config for cfg in run_configs) |
|
|
|
filters = args.test_filter |
|
|
|
filters = args.test_filter |
|
|
|
runs_per_test = args.runs_per_test |
|
|
|
runs_per_test = args.runs_per_test |
|
|
|
forever = args.forever |
|
|
|
forever = args.forever |
|
|
@ -44,19 +74,22 @@ def _build_and_run(check_cancelled): |
|
|
|
if not jobset.run( |
|
|
|
if not jobset.run( |
|
|
|
(['make', |
|
|
|
(['make', |
|
|
|
'-j', '%d' % (multiprocessing.cpu_count() + 1), |
|
|
|
'-j', '%d' % (multiprocessing.cpu_count() + 1), |
|
|
|
'buildtests_c', |
|
|
|
target, |
|
|
|
'CONFIG=%s' % cfg] |
|
|
|
'CONFIG=%s' % cfg] |
|
|
|
for cfg in configs), check_cancelled, maxjobs=1): |
|
|
|
for cfg in build_configs |
|
|
|
|
|
|
|
for target in _MAKE_TEST_TARGETS), |
|
|
|
|
|
|
|
check_cancelled, maxjobs=1): |
|
|
|
sys.exit(1) |
|
|
|
sys.exit(1) |
|
|
|
|
|
|
|
|
|
|
|
# run all the tests |
|
|
|
# run all the tests |
|
|
|
jobset.run(([x] |
|
|
|
jobset.run(( |
|
|
|
for x in itertools.chain.from_iterable( |
|
|
|
config.run_command(x) |
|
|
|
itertools.chain.from_iterable(itertools.repeat( |
|
|
|
for config in run_configs |
|
|
|
glob.glob('bins/%s/%s_test' % (config, filt)), |
|
|
|
for filt in filters |
|
|
|
runs_per_test)) |
|
|
|
for x in itertools.chain.from_iterable(itertools.repeat( |
|
|
|
for config in configs |
|
|
|
glob.glob('bins/%s/%s_test' % ( |
|
|
|
for filt in filters)), check_cancelled) |
|
|
|
config.build_config, filt)), |
|
|
|
|
|
|
|
runs_per_test))), check_cancelled) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if forever: |
|
|
|
if forever: |
|
|
|