1#! /usr/bin/env python3 2 3# Create Makefile targets to run tests, from Meson's test introspection data. 4# 5# Author: Paolo Bonzini <pbonzini@redhat.com> 6 7from collections import defaultdict 8import itertools 9import json 10import os 11import shlex 12import sys 13 14class Suite(object): 15 def __init__(self): 16 self.tests = list() 17 self.slow_tests = list() 18 self.executables = set() 19 20print(''' 21SPEED = quick 22 23# $1 = environment, $2 = test command, $3 = test name, $4 = dir 24.test-human-tap = $1 $(if $4,(cd $4 && $2),$2) < /dev/null | ./scripts/tap-driver.pl --test-name="$3" $(if $(V),,--show-failures-only) 25.test-human-exitcode = $1 $(PYTHON) scripts/test-driver.py $(if $4,-C$4) $(if $(V),--verbose) -- $2 < /dev/null 26.test-tap-tap = $1 $(if $4,(cd $4 && $2),$2) < /dev/null | sed "s/^[a-z][a-z]* [0-9]*/& $3/" || true 27.test-tap-exitcode = printf "%s\\n" 1..1 "`$1 $(if $4,(cd $4 && $2),$2) < /dev/null > /dev/null || echo "not "`ok 1 $3" 28.test.human-print = echo $(if $(V),'$1 $2','Running test $3') && 29.test.env = MALLOC_PERTURB_=$${MALLOC_PERTURB_:-$$(( $${RANDOM:-0} % 255 + 1))} 30 31# $1 = test name, $2 = test target (human or tap) 32.test.run = $(call .test.$2-print,$(.test.env.$1),$(.test.cmd.$1),$(.test.name.$1)) $(call .test-$2-$(.test.driver.$1),$(.test.env.$1),$(.test.cmd.$1),$(.test.name.$1),$(.test.dir.$1)) 33 34.test.output-format = human 35''') 36 37introspect = json.load(sys.stdin) 38i = 0 39 40def process_tests(test, targets, suites): 41 global i 42 env = ' '.join(('%s=%s' % (shlex.quote(k), shlex.quote(v)) 43 for k, v in test['env'].items())) 44 executable = test['cmd'][0] 45 try: 46 executable = os.path.relpath(executable) 47 except: 48 pass 49 if test['workdir'] is not None: 50 try: 51 test['cmd'][0] = os.path.relpath(executable, test['workdir']) 52 except: 53 test['cmd'][0] = executable 54 else: 55 test['cmd'][0] = executable 56 cmd = ' '.join((shlex.quote(x) for x in test['cmd'])) 57 driver = test['protocol'] if 'protocol' in test else 'exitcode' 58 59 i += 1 60 if test['workdir'] is not None: 61 print('.test.dir.%d := %s' % (i, shlex.quote(test['workdir']))) 62 63 if 'depends' in test: 64 deps = (targets.get(x, []) for x in test['depends']) 65 deps = itertools.chain.from_iterable(deps) 66 else: 67 deps = ['all'] 68 69 print('.test.name.%d := %s' % (i, test['name'])) 70 print('.test.driver.%d := %s' % (i, driver)) 71 print('.test.env.%d := $(.test.env) %s' % (i, env)) 72 print('.test.cmd.%d := %s' % (i, cmd)) 73 print('.PHONY: run-test-%d' % (i,)) 74 print('run-test-%d: %s' % (i, ' '.join(deps))) 75 print('\t@$(call .test.run,%d,$(.test.output-format))' % (i,)) 76 77 test_suites = test['suite'] or ['default'] 78 is_slow = any(s.endswith('-slow') for s in test_suites) 79 for s in test_suites: 80 # The suite name in the introspection info is "PROJECT:SUITE" 81 s = s.split(':')[1] 82 if s.endswith('-slow'): 83 s = s[:-5] 84 if is_slow: 85 suites[s].slow_tests.append(i) 86 else: 87 suites[s].tests.append(i) 88 suites[s].executables.add(executable) 89 90def emit_prolog(suites, prefix): 91 all_tap = ' '.join(('%s-report-%s.tap' % (prefix, k) for k in suites.keys())) 92 print('.PHONY: %s %s-report.tap %s' % (prefix, prefix, all_tap)) 93 print('%s: run-tests' % (prefix,)) 94 print('%s-report.tap %s: %s-report%%.tap: all' % (prefix, all_tap, prefix)) 95 print('''\t$(MAKE) .test.output-format=tap --quiet -Otarget V=1 %s$* | ./scripts/tap-merge.pl | tee "$@" \\ 96 | ./scripts/tap-driver.pl $(if $(V),, --show-failures-only)''' % (prefix, )) 97 98def emit_suite(name, suite, prefix): 99 executables = ' '.join(suite.executables) 100 slow_test_numbers = ' '.join((str(x) for x in suite.slow_tests)) 101 test_numbers = ' '.join((str(x) for x in suite.tests)) 102 target = '%s-%s' % (prefix, name) 103 print('.test.quick.%s := %s' % (target, test_numbers)) 104 print('.test.slow.%s := $(.test.quick.%s) %s' % (target, target, slow_test_numbers)) 105 print('%s-build: %s' % (prefix, executables)) 106 print('.PHONY: %s' % (target, )) 107 print('.PHONY: %s-report-%s.tap' % (prefix, name)) 108 print('%s: run-tests' % (target, )) 109 print('ifneq ($(filter %s %s, $(MAKECMDGOALS)),)' % (target, prefix)) 110 print('.tests += $(.test.$(SPEED).%s)' % (target, )) 111 print('endif') 112 113targets = {t['id']: [os.path.relpath(f) for f in t['filename']] 114 for t in introspect['targets']} 115 116testsuites = defaultdict(Suite) 117for test in introspect['tests']: 118 process_tests(test, targets, testsuites) 119emit_prolog(testsuites, 'check') 120for name, suite in testsuites.items(): 121 emit_suite(name, suite, 'check') 122 123benchsuites = defaultdict(Suite) 124for test in introspect['benchmarks']: 125 process_tests(test, targets, benchsuites) 126emit_prolog(benchsuites, 'bench') 127for name, suite in benchsuites.items(): 128 emit_suite(name, suite, 'bench') 129 130print('run-tests: $(patsubst %, run-test-%, $(.tests))') 131