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
|
#!/usr/bin/env python
# --------------------------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# --------------------------------------------------------------------------------------------
import sys
import os
import logging
import re
from subprocess import run
from code_cov_report import create_coverage_report
from common_tasks import run_check_call
logging.getLogger().setLevel(logging.INFO)
root_dir = os.path.abspath(os.path.join(os.path.abspath(__file__), "..", "..", ".."))
coverage_dir = os.path.join(root_dir, "_coverage/")
coveragerc = os.path.join(root_dir, ".coveragerc")
def collect_tox_coverage_files():
coverage_version_cmd = [sys.executable, "-m", "coverage", "--version"]
run(coverage_version_cmd, cwd=root_dir)
logging.info("Running collect tox coverage files...")
coverage_files = []
for root, _, files in os.walk(coverage_dir):
for f in files:
if re.match(".coverage_*", f):
coverage_files.append(os.path.join(root, f))
logging.info(".coverage files: {}".format(coverage_files))
if len(coverage_files):
cov_cmd_array = [sys.executable, "-m", "coverage", "combine"]
cov_cmd_array.extend(coverage_files)
# merge them with coverage combine and copy to root
run(cov_cmd_array, cwd=root_dir)
logging.info("after running coverage combine")
for root, _, files in os.walk(root_dir):
for f in files:
if re.match(".coverage*", f):
print(os.path.join(root, f))
def generate_coverage_xml():
if os.path.exists(coverage_dir):
logging.info("Generating coverage XML")
commands = ["coverage", "xml", "-i", "--rcfile", coveragerc]
run_check_call(commands, root_dir, always_exit=False)
else:
logging.error("Coverage file is not available in {} to generate coverage XML".format(coverage_dir))
def fix_coverage_xml(coverage_file):
print("running 'fix_dot_coverage_file' on {}".format(coverage_file))
out = None
with open(coverage_file, encoding="utf-8") as cov_file:
line = cov_file.read()
# replace relative paths in folder structure pattern
out = re.sub(r"\/\.tox\/[\s\S_]*?\/site-packages", "", line)
# replace relative paths in python import pattern
out = re.sub(r"\.?\.tox[\s\S\.\d]*?\.site-packages", "", out)
if out:
with open(coverage_file, "w") as cov_file:
cov_file.write(out)
if __name__ == "__main__":
coverage_xml = os.path.join(root_dir, "coverage.xml")
collect_tox_coverage_files()
generate_coverage_xml()
create_coverage_report()
if os.path.exists(coverage_xml):
fix_coverage_xml(coverage_xml)
|
