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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
import argparse
import tarfile
import zipfile
import os
import sys
import time
import subprocess
def mkdir_p(path):
try:
os.makedirs(path)
except OSError:
pass
class Timer(object):
def __init__(self):
self.start = time.time()
def step(self, msg):
sys.stderr.write("{} ({}s)\n".format(msg, int(time.time() - self.start)))
self.start = time.time()
def main(
source,
output,
java,
prefix_filter,
exclude_filter,
jars_list,
output_format,
tar_output,
agent_disposition,
runners_paths,
):
timer = Timer()
reports_dir = 'jacoco_reports_dir'
mkdir_p(reports_dir)
with tarfile.open(source) as tf:
tf.extractall(reports_dir)
timer.step("Coverage data extracted")
reports = [os.path.join(reports_dir, fname) for fname in os.listdir(reports_dir)]
with open(jars_list) as f:
jars = f.read().strip().split()
if jars and runners_paths:
for r in runners_paths:
try:
jars.remove(r)
except ValueError:
pass
src_dir = 'sources_dir'
cls_dir = 'classes_dir'
mkdir_p(src_dir)
mkdir_p(cls_dir)
for jar in jars:
if jar.endswith('devtools-jacoco-agent.jar'):
agent_disposition = jar
# Skip java contrib - it's irrelevant coverage
if jar.startswith('contrib/java'):
continue
with zipfile.ZipFile(jar) as jf:
for entry in jf.infolist():
if entry.filename.endswith('.java'):
dest = src_dir
elif entry.filename.endswith('.class'):
dest = cls_dir
else:
continue
entry.filename = entry.filename.encode('utf-8')
jf.extract(entry, dest)
timer.step("Jar files extracted")
if not agent_disposition:
print >> sys.stderr, 'Can\'t find jacoco agent. Will not generate html report for java coverage.'
if tar_output:
report_dir = 'java.report.temp'
else:
report_dir = output
mkdir_p(report_dir)
if agent_disposition:
agent_cmd = [
java,
'-jar',
agent_disposition,
src_dir,
cls_dir,
prefix_filter or '.',
exclude_filter or '__no_exclude__',
report_dir,
output_format,
]
agent_cmd += reports
subprocess.check_call(agent_cmd)
timer.step("Jacoco finished")
if tar_output:
with tarfile.open(output, 'w') as outf:
outf.add(report_dir, arcname='.')
if __name__ == '__main__':
if 'LC_ALL' in os.environ:
if os.environ['LC_ALL'] == 'C':
os.environ['LC_ALL'] = 'en_GB.UTF-8'
parser = argparse.ArgumentParser()
parser.add_argument('--source', action='store')
parser.add_argument('--output', action='store')
parser.add_argument('--java', action='store')
parser.add_argument('--prefix-filter', action='store')
parser.add_argument('--exclude-filter', action='store')
parser.add_argument('--jars-list', action='store')
parser.add_argument('--output-format', action='store', default="html")
parser.add_argument('--raw-output', dest='tar_output', action='store_false', default=True)
parser.add_argument('--agent-disposition', action='store')
parser.add_argument('--runner-path', dest='runners_paths', action='append', default=[])
args = parser.parse_args()
main(**vars(args))
|
