aboutsummaryrefslogtreecommitdiffstats
path: root/conftest.py
blob: 0c42f66c0506739ba4e89e0f41c5b1ab9535a5d0 (plain)
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
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
# -*- coding:utf-8 -*-
import pytest
import json
import shutil
import subprocess
from jinja2 import Environment,FileSystemLoader

import plugins.agl_test_conf as conf


@pytest.fixture(scope='session' ,autouse=True)
def setup_compress_function():
    #Before the test start, clean the env
    report_json = conf.TMP_LOGS_DIR + "report.json"
    output = subprocess.run(['ls',report_json],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    if(output.returncode == 0):
        subprocess.run(['rm',report_json])

    #Makdir of TMP_TEST_REPORT and REPORT_LOGS_DIR
    subprocess.run(['mkdir','-p',conf.TMP_TEST_REPORT])
    subprocess.run(['mkdir','-p',conf.REPORT_LOGS_DIR])

    yield
    #Collect report.json from all test sets to generate a report.json for all the test sets
    report_files = conf.TMP_LOGS_DIR + "report_files"
    with open(report_files,'w') as report_f:
        subprocess.run(['find','-name','report.json'],cwd=conf.TMP_LOGS_DIR,stdout=report_f)
    report_f.close()

    #Get the summary data and write to report.json file
    summary_data = get_summary_data(report_files)
    summary_json = conf.TMP_LOGS_DIR + "/report.json"
    with open(summary_json, 'w') as summary_file:
        json.dump(summary_data,summary_file,indent=4,sort_keys=False)
    summary_file.close()

    #Get zip file name
    issue = subprocess.getoutput('cat /etc/issue')
    version = issue[23:-7]
    machine_name = subprocess.getoutput('uname -n')
    date = subprocess.getoutput('date +%Y%m%d')
    zip_name = "agl-test-log-" + version + '-' + machine_name + '-' + date

    #Creat summary report in html
    summary_html_data = format_test_suite_table(summary_data)
    summary_html_data["log_zip_name"] = zip_name
    env = Environment(loader=FileSystemLoader(conf.get_tpl_dir()))
    template = env.get_template("all_test_suites_tpl.html")
    html_path = conf.TMP_LOGS_DIR + "test-report/summary-report.html"
    with open(html_path, "w") as html_file:
        html_content = template.render(data=summary_html_data)
        html_file.write(html_content)
    html_file.close()

    #Copy summary report file
    source_file = conf.TMP_LOGS_DIR + "test-report/summary-report.html"
    target_file = conf.REPORT_LOGS_DIR + "summary-report.html"
    shutil.copyfile(source_file,target_file)

    #Package the test report
    base_name = conf.REPORT_LOGS_DIR + zip_name
    root_dir = conf.TMP_LOGS_DIR + "test-report"
    shutil.make_archive(base_name,"zip",root_dir)


#Summarize all reports.json file
def get_summary_data(report_files):
    summary_data = {}
    summary_total = summary_passed = summary_failed = summary_skipped = 0
    files = open(report_files)
    while True:
        report = files.readline()
        if not report:
            break
        report = report[1:-1]
        report_json = conf.TMP_LOGS_DIR + report
        with open(report_json,'r') as f:
            data = json.load(f)

            total = passed = xpassed = failed = xfailed = skipped = 0
            total = data["collected"]
            passed = data["passed"]
            xpassed = data["xpassed"]
            failed = data["failed"]
            xfailed = data["xfailed"]
            skipped = data["skipped"]
            test_status = data["test_status"]
            test_name = data["test_name"]

            this_summary = {
                'total': total,
                'passed': passed,
                'xpassed': xpassed,
                'failed': failed,
                'xfailed': xfailed,
                'skipped': skipped,
                'test_status': test_status,
            }
            summary_data[test_name] = this_summary

            summary_total = summary_total + 1
            if(test_status=="passed"):
                summary_passed = summary_passed + 1
            elif(test_status=="failed"):
                summary_failed = summary_failed + 1
            else:
                summary_skipped = summary_skipped + 1
        f.close()
    summary_data["summary"] = {
        "summary_total": summary_total,
        "summary_passed": summary_passed,
        "summary_failed": summary_failed,
        "summary_skipped": summary_skipped,
    }

    status = ""
    if (summary_data["summary"]["summary_total"] == summary_data["summary"]["summary_skipped"]):
        status = "skip"
    elif (summary_data["summary"]["summary_failed"] == 0):
        status = "pass"
    else:
        status = "fail"
    summary_data["summary"]["status"] = status

    return summary_data

def format_test_suite_table(summary_data):
    html_data = ""
    #init table title
    bgcolor = conf.BGCOLOR_TABLE_TITLE
    color = conf.COLOR_TABLE_TITLE
    html_data += "\t<tr bgcolor = \"" + bgcolor + "\">\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> test suite </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> status </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> total </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> pass </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> xpass </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> fail </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> xfail </font></th>\n"
    html_data += "\t\t<th><font color = \"" + color + "\"> skip </font></th>\n"
    html_data += "\t</tr>\n"

    #init all rows
    bgcolor = conf.BGCOLOR_DARK
    for key in summary_data:
        if(key != "summary"):
            html_data += "\t<tr bgcolor = \"" + bgcolor + "\">\n"
            html_data += "\t\t<th>" + key + "</th>\n"
            html_data += "\t\t<th>" + summary_data[key]["test_status"] + "</th>\n"
            html_data += "\t\t<th>" + str(summary_data[key]["total"]) + "</th>\n"
            html_data += "\t\t<th>" + str(summary_data[key]["passed"]) + "</th>\n"
            html_data += "\t\t<th>" + str(summary_data[key]["xpassed"]) + "</th>\n"
            html_data += "\t\t<th>" + str(summary_data[key]["failed"]) + "</th>\n"
            html_data += "\t\t<th>" + str(summary_data[key]["xfailed"]) + "</th>\n"
            html_data += "\t\t<th>" + str(summary_data[key]["skipped"]) + "</th>\n"
            html_data += "\t</tr>\n"

            if (bgcolor == conf.BGCOLOR_DARK):
                bgcolor = conf.BGCOLOR_LIGHT
            else:
                bgcolor = conf.BGCOLOR_DARK
    summary_data["test_suite_table_html"] = html_data
    return summary_data