|
| 1 | +#!/usr/bin/env python |
| 2 | + |
| 3 | +# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved. |
| 4 | +# SPDX-License-Identifier: Apache-2.0 |
| 5 | + |
| 6 | +import sys |
| 7 | +import xml.etree.ElementTree as ET |
| 8 | +import re |
| 9 | +import os.path |
| 10 | + |
| 11 | +# branch-rate="0.0" complexity="0.0" line-rate="1.0" |
| 12 | +# branch="true" hits="1" number="86" |
| 13 | + |
| 14 | +def find_lines(j_package, filename): |
| 15 | + """Return all <line> elements for a given source file in a package.""" |
| 16 | + lines = list() |
| 17 | + sourcefiles = j_package.findall("sourcefile") |
| 18 | + for sourcefile in sourcefiles: |
| 19 | + if sourcefile.attrib.get("name") == os.path.basename(filename): |
| 20 | + lines = lines + sourcefile.findall("line") |
| 21 | + return lines |
| 22 | + |
| 23 | +def line_is_after(jm, start_line): |
| 24 | + return int(jm.attrib.get('line', 0)) > start_line |
| 25 | + |
| 26 | +def method_lines(jmethod, jmethods, jlines): |
| 27 | + """Filter the lines from the given set of jlines that apply to the given jmethod.""" |
| 28 | + start_line = int(jmethod.attrib.get('line', 0)) |
| 29 | + larger = list(int(jm.attrib.get('line', 0)) for jm in jmethods if line_is_after(jm, start_line)) |
| 30 | + end_line = min(larger) if len(larger) else 99999999 |
| 31 | + |
| 32 | + for jline in jlines: |
| 33 | + if start_line <= int(jline.attrib['nr']) < end_line: |
| 34 | + yield jline |
| 35 | + |
| 36 | +def convert_lines(j_lines, into): |
| 37 | + """Convert the JaCoCo <line> elements into Cobertura <line> elements, add them under the given element.""" |
| 38 | + c_lines = ET.SubElement(into, 'lines') |
| 39 | + for jline in j_lines: |
| 40 | + mb = int(jline.attrib['mb']) |
| 41 | + cb = int(jline.attrib['cb']) |
| 42 | + ci = int(jline.attrib['ci']) |
| 43 | + |
| 44 | + cline = ET.SubElement(c_lines, 'line') |
| 45 | + cline.set('number', jline.attrib['nr']) |
| 46 | + cline.set('hits', '1' if ci > 0 else '0') # Probably not true but no way to know from JaCoCo XML file |
| 47 | + |
| 48 | + if mb + cb > 0: |
| 49 | + percentage = str(int(100 * (float(cb) / (float(cb) + float(mb))))) + '%' |
| 50 | + cline.set('branch', 'true') |
| 51 | + cline.set('condition-coverage', percentage + ' (' + str(cb) + '/' + str(cb + mb) + ')') |
| 52 | + |
| 53 | + cond = ET.SubElement(ET.SubElement(cline, 'conditions'), 'condition') |
| 54 | + cond.set('number', '0') |
| 55 | + cond.set('type', 'jump') |
| 56 | + cond.set('coverage', percentage) |
| 57 | + else: |
| 58 | + cline.set('branch', 'false') |
| 59 | + |
| 60 | +def guess_filename(path_to_class): |
| 61 | + m = re.match('([^$]*)', path_to_class) |
| 62 | + return (m.group(1) if m else path_to_class) + '.java' |
| 63 | + |
| 64 | +def add_counters(source, target): |
| 65 | + target.set('line-rate', counter(source, 'LINE')) |
| 66 | + target.set('branch-rate', counter(source, 'BRANCH')) |
| 67 | + target.set('complexity', counter(source, 'COMPLEXITY', sum)) |
| 68 | + |
| 69 | +def fraction(covered, missed): |
| 70 | + return covered / (covered + missed) |
| 71 | + |
| 72 | +def sum(covered, missed): |
| 73 | + return covered + missed |
| 74 | + |
| 75 | +def counter(source, type, operation=fraction): |
| 76 | + cs = source.findall('counter') |
| 77 | + c = next((ct for ct in cs if ct.attrib.get('type') == type), None) |
| 78 | + |
| 79 | + if c is not None: |
| 80 | + covered = float(c.attrib['covered']) |
| 81 | + missed = float(c.attrib['missed']) |
| 82 | + |
| 83 | + return str(operation(covered, missed)) |
| 84 | + else: |
| 85 | + return '0.0' |
| 86 | + |
| 87 | +def convert_method(j_method, j_lines): |
| 88 | + c_method = ET.Element('method') |
| 89 | + c_method.set('name', j_method.attrib['name']) |
| 90 | + c_method.set('signature', j_method.attrib['desc']) |
| 91 | + |
| 92 | + add_counters(j_method, c_method) |
| 93 | + convert_lines(j_lines, c_method) |
| 94 | + |
| 95 | + return c_method |
| 96 | + |
| 97 | +def convert_class(j_class, j_package): |
| 98 | + c_class = ET.Element('class') |
| 99 | + c_class.set('name', j_class.attrib['name'].replace('/', '.')) |
| 100 | + c_class.set('filename', guess_filename(j_class.attrib['name'])) |
| 101 | + |
| 102 | + all_j_lines = list(find_lines(j_package, c_class.attrib['filename'])) |
| 103 | + |
| 104 | + c_methods = ET.SubElement(c_class, 'methods') |
| 105 | + all_j_methods = list(j_class.findall('method')) |
| 106 | + for j_method in all_j_methods: |
| 107 | + j_method_lines = method_lines(j_method, all_j_methods, all_j_lines) |
| 108 | + c_methods.append(convert_method(j_method, j_method_lines)) |
| 109 | + |
| 110 | + add_counters(j_class, c_class) |
| 111 | + convert_lines(all_j_lines, c_class) |
| 112 | + |
| 113 | + return c_class |
| 114 | + |
| 115 | +def convert_package(j_package): |
| 116 | + c_package = ET.Element('package') |
| 117 | + c_package.attrib['name'] = j_package.attrib['name'].replace('/', '.') |
| 118 | + |
| 119 | + c_classes = ET.SubElement(c_package, 'classes') |
| 120 | + for j_class in j_package.findall('class'): |
| 121 | + # Only output the class if it has methods to be covered |
| 122 | + if j_class.findall('method'): |
| 123 | + c_classes.append(convert_class(j_class, j_package)) |
| 124 | + |
| 125 | + add_counters(j_package, c_package) |
| 126 | + |
| 127 | + return c_package |
| 128 | + |
| 129 | +def convert_root(source, target, source_roots): |
| 130 | + target.set('timestamp', str(int(source.find('sessioninfo').attrib['start']) / 1000)) |
| 131 | + |
| 132 | + sources = ET.SubElement(target, 'sources') |
| 133 | + for s in source_roots: |
| 134 | + ET.SubElement(sources, 'source').text = s |
| 135 | + |
| 136 | + packages = ET.SubElement(target, 'packages') |
| 137 | + for package in source.findall('package'): |
| 138 | + packages.append(convert_package(package)) |
| 139 | + |
| 140 | + add_counters(source, target) |
| 141 | + |
| 142 | +def jacoco2cobertura(filename, source_roots): |
| 143 | + if filename == '-': |
| 144 | + root = ET.fromstring(sys.stdin.read()) |
| 145 | + else: |
| 146 | + tree = ET.parse(filename) |
| 147 | + root = tree.getroot() |
| 148 | + |
| 149 | + into = ET.Element('coverage') |
| 150 | + convert_root(root, into, source_roots) |
| 151 | + print('<?xml version="1.0" ?>') |
| 152 | + print(ET.tostring(into, encoding='unicode')) |
| 153 | + |
| 154 | +if __name__ == '__main__': |
| 155 | + if len(sys.argv) < 2: |
| 156 | + print("Usage: cover2cover.py FILENAME [SOURCE_ROOTS]") |
| 157 | + sys.exit(1) |
| 158 | + |
| 159 | + filename = sys.argv[1] |
| 160 | + source_roots = sys.argv[2:] if 2 < len(sys.argv) else '.' |
| 161 | + |
| 162 | + jacoco2cobertura(filename, source_roots) |
0 commit comments