ClickHouse/tests/ci/upload_result_helper.py

Ignoring revisions in .git-blame-ignore-revs. Click here to bypass and see the normal blame view.

105 lines
3.0 KiB
Python
Raw Normal View History

2021-11-12 11:39:00 +00:00
import os
import logging
import ast
2022-08-09 16:49:55 +00:00
from env_helper import (
GITHUB_JOB_URL,
GITHUB_REPOSITORY,
GITHUB_RUN_URL,
GITHUB_SERVER_URL,
)
from report import ReportColorTheme, create_test_html_report
2021-11-12 11:39:00 +00:00
2021-11-26 14:00:09 +00:00
def process_logs(
s3_client, additional_logs, s3_path_prefix, test_results, with_raw_logs
):
2022-10-26 12:01:12 +00:00
logging.info("Upload files to s3 %s", additional_logs)
2022-10-26 08:30:20 +00:00
2021-11-26 14:00:09 +00:00
processed_logs = {}
2021-11-12 11:39:00 +00:00
# Firstly convert paths of logs from test_results to urls to s3.
for test_result in test_results:
2021-11-12 19:57:26 +00:00
if len(test_result) <= 3 or with_raw_logs:
2021-11-12 11:39:00 +00:00
continue
# Convert from string repr of list to list.
test_log_paths = ast.literal_eval(test_result[3])
test_log_urls = []
for log_path in test_log_paths:
2021-11-26 14:00:09 +00:00
if log_path in processed_logs:
test_log_urls.append(processed_logs[log_path])
2021-11-12 11:39:00 +00:00
elif log_path:
url = s3_client.upload_test_report_to_s3(
log_path, s3_path_prefix + "/" + os.path.basename(log_path)
)
2021-11-12 11:39:00 +00:00
test_log_urls.append(url)
2021-11-26 14:00:09 +00:00
processed_logs[log_path] = url
2021-11-12 11:39:00 +00:00
test_result[3] = test_log_urls
additional_urls = []
for log_path in additional_logs:
if log_path:
additional_urls.append(
s3_client.upload_test_report_to_s3(
log_path, s3_path_prefix + "/" + os.path.basename(log_path)
)
)
2021-11-12 11:39:00 +00:00
return additional_urls
2021-11-26 14:00:09 +00:00
def upload_results(
s3_client,
pr_number,
commit_sha,
test_results,
additional_files,
check_name,
with_raw_logs=True,
statuscolors=None,
):
s3_path_prefix = f"{pr_number}/{commit_sha}/" + check_name.lower().replace(
" ", "_"
).replace("(", "_").replace(")", "_").replace(",", "_")
additional_urls = process_logs(
s3_client, additional_files, s3_path_prefix, test_results, with_raw_logs
)
2021-11-12 11:39:00 +00:00
2021-11-26 14:00:09 +00:00
branch_url = f"{GITHUB_SERVER_URL}/{GITHUB_REPOSITORY}/commits/master"
2021-11-12 11:39:00 +00:00
branch_name = "master"
if pr_number != 0:
branch_name = f"PR #{pr_number}"
2021-11-26 14:00:09 +00:00
branch_url = f"{GITHUB_SERVER_URL}/{GITHUB_REPOSITORY}/pull/{pr_number}"
commit_url = f"{GITHUB_SERVER_URL}/{GITHUB_REPOSITORY}/commit/{commit_sha}"
2021-11-12 11:39:00 +00:00
if additional_urls:
raw_log_url = additional_urls[0]
additional_urls.pop(0)
else:
2022-08-09 16:49:55 +00:00
raw_log_url = GITHUB_JOB_URL()
2021-11-12 11:39:00 +00:00
statuscolors = (
ReportColorTheme.bugfixcheck if "bugfix validate check" in check_name else None
)
html_report = create_test_html_report(
check_name,
test_results,
raw_log_url,
2022-08-09 16:49:55 +00:00
GITHUB_RUN_URL,
GITHUB_JOB_URL(),
branch_url,
branch_name,
commit_url,
additional_urls,
with_raw_logs,
statuscolors=statuscolors,
)
with open("report.html", "w", encoding="utf-8") as f:
2021-11-12 11:39:00 +00:00
f.write(html_report)
url = s3_client.upload_test_report_to_s3("report.html", s3_path_prefix + ".html")
2021-11-12 11:39:00 +00:00
logging.info("Search result in url %s", url)
return url