TestBench/parse_jacoco.py at main · iSEngLab/TestBench · GitHub

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
import xml.etree.ElementTree as ET
import os
import random

def extract_coverage(report_path, package_name, class_name, method_name):
    tree = ET.parse(report_path)
    root = tree.getroot()

    package_coverage, class_coverage, method_coverage = None, None, 0.0

    for package in root.findall(".//package"):
        if package.get('name') == package_name.replace('.', '/'):
            package_counter = package.find('counter[@type="LINE"]')
            if package_counter is not None:
                package_coverage  = calculate_coverage(package_counter)

            for clazz in package.findall('class'):
                if clazz.get('name').endswith("/" + class_name):
                    class_counter = clazz.find('counter[@type="LINE"]')
                    if class_counter is not None:
                        class_coverage  = calculate_coverage(class_counter)

                    count = 0
                    for method in clazz.findall('method'):
                        if method.get('name') == method_name:
                            method_counter = method.find('counter[@type="LINE"]')
                            if method_counter is not None and method_counter.get('covered') != '0':
                                count += 1
                                method_coverage  += calculate_coverage(method_counter)
                    if count > 0:
                        method_coverage = method_coverage / count
                    break

    return package_coverage, class_coverage, method_coverage

def calculate_coverage(counter):
    missed = int(counter.get('missed'))
    covered = int(counter.get('covered'))
    total = missed + covered
    if total > 0:
        return round((covered / total) * 100, 2)
    else:
        return 0.0

def find_jacoco_xml(base_path):
    target_dir = 'target'
    target_file = 'jacoco.xml'
    target_path = f'{target_dir}/site/jacoco/{target_file}'

    found_paths = []

    for root, dirs, files in os.walk(base_path):
        if target_dir in dirs:
            full_path = os.path.join(root, target_path)
            if os.path.isfile(full_path):
                found_paths.append(os.path.relpath(full_path, base_path))

    return found_paths

def handle_class_name(class_name):
    # If the class name ends with Test, remove Test. This is a bug that has been resolved, and the new coverage generated will not have the Test suffix.
    if class_name.endswith('Test'):
        class_name = class_name[:len(class_name)-4]
    if class_name.endswith('Tests'):
        class_name = class_name[:len(class_name)-5]
    return class_name

def get_coverage(base_path, package_name, class_name, method_name):
    xml_path = find_jacoco_xml(base_path)
    package_coverage = 0
    class_coverage = 0
    method_coverage = 0
    class_name = handle_class_name(class_name)
    for xml_file in xml_path:
        package_cov, class_cov, method_cov = extract_coverage(xml_file, package_name, class_name, method_name)
        if 1 == 0:
            xml_append_path = "/home/joseph/TestCaseGeneration/xmls/"
            if not os.path.exists(xml_append_path):
                os.makedirs(xml_append_path)
            with open(xml_append_path + class_name + "_" + method_name + "_" + str(random.randint(1, 50)) + "_jacoco.xml", 'w') as f:
                with open(xml_file, 'r') as file:
                    xmlFile = file.read()
                    f.write(xmlFile)
        if package_cov:
            package_coverage = package_coverage if package_coverage > package_cov else package_cov
        if class_cov:
            class_coverage = class_coverage if class_coverage > class_cov else class_cov
        if method_cov:
            method_coverage = method_coverage if method_coverage > method_cov else method_cov
    return package_coverage, class_coverage, method_coverage


if __name__ == "__main__":
    # Example function call
    report_path = 'jacoco.xml'
    package_name = 'org.apache.commons.math4.legacy.core.dfp'
    class_name = 'Dfp'
    method_name = 'add'

    package_coverage, class_coverage, method_coverage = extract_coverage(report_path, package_name, class_name, method_name)
    print(package_coverage)
    print(class_coverage)
    print(method_coverage)