1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49
|
"""
simple scrpt for junitxml file merging
"""
from lxml.etree import parse, Element, tostring
from collections import defaultdict
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--out')
parser.add_argument('path', nargs='...')
opts = parser.parse_args()
files = []
for path in opts.path:
files.append(parse(path))
accum = defaultdict(int)
children = []
for item in files:
root = item.getroot()
for key, value in root.attrib.items():
if not value:
continue
value = float(value) if '.' in value else int(value)
accum[key] += value
children.extend(root)
assert len(children) == sum(accum[x] for x in 'tests errors skips'.split())
children.sort(key=lambda x:(x.attrib['classname'], x.attrib['name']))
new = Element('testsuite', dict((k, str(v)) for k, v in accum.items()))
new.extend(children)
data = tostring(new)
with open(opts.out, 'w') as fp:
fp.write(data)
|