LXR qemu/scripts/mtest2make.py

   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.deps = set()
  17        self.speeds = ['quick']
  18
  19    def names(self, base):
  20        return [base if speed == 'quick' else f'{base}-{speed}' for speed in self.speeds]
  21
  22
  23print('''
  24SPEED = quick
  25
  26.speed.quick = $(foreach s,$(sort $(filter-out %-slow %-thorough, $1)), --suite $s)
  27.speed.slow = $(foreach s,$(sort $(filter-out %-thorough, $1)), --suite $s)
  28.speed.thorough = $(foreach s,$(sort $1), --suite $s)
  29
  30.mtestargs = --no-rebuild -t 0
  31ifneq ($(SPEED), quick)
  32.mtestargs += --setup $(SPEED)
  33endif
  34.mtestargs += $(subst -j,--num-processes , $(filter-out -j, $(lastword -j1 $(filter -j%, $(MAKEFLAGS)))))
  35
  36.check.mtestargs = $(MTESTARGS) $(.mtestargs) $(if $(V),--verbose,--print-errorlogs)
  37.bench.mtestargs = $(MTESTARGS) $(.mtestargs) --benchmark --verbose''')
  38
  39introspect = json.load(sys.stdin)
  40
  41def process_tests(test, targets, suites):
  42    executable = test['cmd'][0]
  43    try:
  44        executable = os.path.relpath(executable)
  45    except:
  46        pass
  47
  48    deps = (targets.get(x, []) for x in test['depends'])
  49    deps = itertools.chain.from_iterable(deps)
  50    deps = list(deps)
  51
  52    test_suites = test['suite'] or ['default']
  53    for s in test_suites:
  54        # The suite name in the introspection info is "PROJECT" or "PROJECT:SUITE"
  55        if ':' in s:
  56            s = s.split(':')[1]
  57            if s == 'slow' or s == 'thorough':
  58                continue
  59        if s.endswith('-slow'):
  60            s = s[:-5]
  61            suites[s].speeds.append('slow')
  62        if s.endswith('-thorough'):
  63            s = s[:-9]
  64            suites[s].speeds.append('thorough')
  65        suites[s].deps.update(deps)
  66
  67def emit_prolog(suites, prefix):
  68    all_targets = ' '.join((f'{prefix}-{k}' for k in suites.keys()))
  69    all_xml = ' '.join((f'{prefix}-report-{k}.junit.xml' for k in suites.keys()))
  70    print()
  71    print(f'all-{prefix}-targets = {all_targets}')
  72    print(f'all-{prefix}-xml = {all_xml}')
  73    print(f'.PHONY: {prefix} do-meson-{prefix} {prefix}-report.junit.xml $(all-{prefix}-targets) $(all-{prefix}-xml)')
  74    print(f'ifeq ($(filter {prefix}, $(MAKECMDGOALS)),)')
  75    print(f'.{prefix}.mtestargs += $(call .speed.$(SPEED), $(.{prefix}.mtest-suites))')
  76    print(f'endif')
  77    print(f'{prefix}-build: run-ninja')
  78    print(f'{prefix} $(all-{prefix}-targets): do-meson-{prefix}')
  79    print(f'do-meson-{prefix}: run-ninja; $(if $(MAKE.n),,+)$(MESON) test $(.{prefix}.mtestargs)')
  80    print(f'{prefix}-report.junit.xml $(all-{prefix}-xml): {prefix}-report%.junit.xml: run-ninja')
  81    print(f'\t$(MAKE) {prefix}$* MTESTARGS="$(MTESTARGS) --logbase {prefix}-report$*" && ln -f meson-logs/$@ .')
  82
  83def emit_suite_deps(name, suite, prefix):
  84    deps = ' '.join(suite.deps)
  85    targets = [f'{prefix}-{name}', f'{prefix}-report-{name}.junit.xml', f'{prefix}', f'{prefix}-report.junit.xml',
  86               f'{prefix}-build']
  87    print()
  88    print(f'.{prefix}-{name}.deps = {deps}')
  89    for t in targets:
  90        print(f'.ninja-goals.{t} += $(.{prefix}-{name}.deps)')
  91
  92def emit_suite(name, suite, prefix):
  93    emit_suite_deps(name, suite, prefix)
  94    targets = f'{prefix}-{name} {prefix}-report-{name}.junit.xml {prefix} {prefix}-report.junit.xml'
  95    print(f'ifneq ($(filter {targets}, $(MAKECMDGOALS)),)')
  96    print(f'.{prefix}.mtest-suites += ' + ' '.join(suite.names(name)))
  97    print(f'endif')
  98
  99targets = {t['id']: [os.path.relpath(f) for f in t['filename']]
 100           for t in introspect['targets']}
 101
 102testsuites = defaultdict(Suite)
 103for test in introspect['tests']:
 104    process_tests(test, targets, testsuites)
 105emit_prolog(testsuites, 'check')
 106for name, suite in testsuites.items():
 107    emit_suite(name, suite, 'check')
 108
 109benchsuites = defaultdict(Suite)
 110for test in introspect['benchmarks']:
 111    process_tests(test, targets, benchsuites)
 112emit_prolog(benchsuites, 'bench')
 113for name, suite in benchsuites.items():
 114    emit_suite(name, suite, 'bench')
 115