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)
|
