mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 09:22:05 +00:00
1105 lines
41 KiB
Python
Executable File
1105 lines
41 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import csv
|
|
import glob
|
|
import json
|
|
import logging
|
|
import os
|
|
import random
|
|
import re
|
|
import shlex
|
|
import shutil
|
|
import signal
|
|
import string
|
|
import subprocess
|
|
import sys
|
|
import time
|
|
from collections import defaultdict
|
|
from itertools import chain
|
|
from typing import Any, Dict, List, Optional
|
|
|
|
from ci_utils import kill_ci_runner
|
|
from env_helper import IS_CI
|
|
from integration_test_images import IMAGES
|
|
from report import JOB_TIMEOUT_TEST_NAME
|
|
from stopwatch import Stopwatch
|
|
from tee_popen import TeePopen
|
|
|
|
MAX_RETRY = 1
|
|
NUM_WORKERS = 5
|
|
SLEEP_BETWEEN_RETRIES = 5
|
|
PARALLEL_GROUP_SIZE = 100
|
|
CLICKHOUSE_BINARY_PATH = "usr/bin/clickhouse"
|
|
CLICKHOUSE_ODBC_BRIDGE_BINARY_PATH = "usr/bin/clickhouse-odbc-bridge"
|
|
CLICKHOUSE_LIBRARY_BRIDGE_BINARY_PATH = "usr/bin/clickhouse-library-bridge"
|
|
|
|
FLAKY_TRIES_COUNT = 3 # run whole pytest several times
|
|
FLAKY_REPEAT_COUNT = 5 # runs test case in single module several times
|
|
MAX_TIME_SECONDS = 3600
|
|
|
|
MAX_TIME_IN_SANDBOX = 20 * 60 # 20 minutes
|
|
TASK_TIMEOUT = 8 * 60 * 60 # 8 hours
|
|
|
|
NO_CHANGES_MSG = "Nothing to run"
|
|
|
|
|
|
# Search test by the common prefix.
|
|
# This is accept tests w/o parameters in skip list.
|
|
#
|
|
# Examples:
|
|
# - has_test(['foobar'], 'foobar[param]') == True
|
|
# - has_test(['foobar[param]'], 'foobar') == True
|
|
def has_test(tests, test_to_match):
|
|
for test in tests:
|
|
if len(test_to_match) < len(test):
|
|
if test[0 : len(test_to_match)] == test_to_match:
|
|
return True
|
|
else:
|
|
if test_to_match[0 : len(test)] == test:
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_changed_tests_to_run(pr_info, repo_path):
|
|
result = set()
|
|
changed_files = pr_info["changed_files"]
|
|
|
|
if changed_files is None:
|
|
return []
|
|
|
|
for fpath in changed_files:
|
|
if re.search(r"tests/integration/test_.*/test.*\.py", fpath) is not None:
|
|
logging.info("File %s changed and seems like integration test", fpath)
|
|
result.add("/".join(fpath.split("/")[2:]))
|
|
return filter_existing_tests(result, repo_path)
|
|
|
|
|
|
def filter_existing_tests(tests_to_run, repo_path):
|
|
result = []
|
|
for relative_test_path in tests_to_run:
|
|
if os.path.exists(
|
|
os.path.join(repo_path, "tests/integration", relative_test_path)
|
|
):
|
|
result.append(relative_test_path)
|
|
else:
|
|
logging.info(
|
|
"Skipping test %s, seems like it was removed", relative_test_path
|
|
)
|
|
return result
|
|
|
|
|
|
def _get_deselect_option(tests):
|
|
return " ".join([f"--deselect {t}" for t in tests])
|
|
|
|
|
|
# https://stackoverflow.com/questions/312443/how-do-you-split-a-list-into-evenly-sized-chunks
|
|
def chunks(lst, n):
|
|
"""Yield successive n-sized chunks from lst."""
|
|
for i in range(0, len(lst), n):
|
|
yield lst[i : i + n]
|
|
|
|
|
|
def get_counters(fname):
|
|
counters = {
|
|
"ERROR": set([]),
|
|
"PASSED": set([]),
|
|
"FAILED": set([]),
|
|
"SKIPPED": set([]),
|
|
} # type: Dict[str, Any]
|
|
|
|
with open(fname, "r", encoding="utf-8") as out:
|
|
for line in out:
|
|
line = line.strip()
|
|
# Example of log:
|
|
#
|
|
# test_mysql_protocol/test.py::test_golang_client
|
|
# [gw0] [ 7%] ERROR test_mysql_protocol/test.py::test_golang_client
|
|
#
|
|
# And only the line with test status should be matched
|
|
if not (".py::" in line and " " in line):
|
|
continue
|
|
|
|
line = line.strip()
|
|
# [gw0] [ 7%] ERROR test_mysql_protocol/test.py::test_golang_client
|
|
# ^^^^^^^^^^^^^
|
|
if line.strip().startswith("["):
|
|
line = re.sub(r"^\[[^\[\]]*\] \[[^\[\]]*\] ", "", line)
|
|
|
|
line_arr = line.split(" ")
|
|
if len(line_arr) < 2:
|
|
logging.debug("Strange line %s", line)
|
|
continue
|
|
|
|
# Lines like:
|
|
#
|
|
# ERROR test_mysql_protocol/test.py::test_golang_client
|
|
# PASSED test_replicated_users/test.py::test_rename_replicated[QUOTA]
|
|
# PASSED test_drop_is_lock_free/test.py::test_query_is_lock_free[detach part]
|
|
#
|
|
state = line_arr.pop(0)
|
|
test_name = " ".join(line_arr)
|
|
|
|
# Normalize test names for lines like this:
|
|
#
|
|
# FAILED test_storage_s3/test.py::test_url_reconnect_in_the_middle - Exception
|
|
# FAILED test_distributed_ddl/test.py::test_default_database[configs] - AssertionError: assert ...
|
|
#
|
|
test_name = re.sub(
|
|
r"^(?P<test_name>[^\[\] ]+)(?P<test_param>\[[^\[\]]*\]|)(?P<test_error> - .*|)$",
|
|
r"\g<test_name>\g<test_param>",
|
|
test_name,
|
|
)
|
|
|
|
if state in counters:
|
|
counters[state].add(test_name)
|
|
else:
|
|
# will skip lines like:
|
|
# 30.76s call test_host_ip_change/test.py::test_ip_drop_cache
|
|
# 5.71s teardown test_host_ip_change/test.py::test_ip_change[node1]
|
|
# and similar
|
|
logging.debug("Strange state in line %s", line)
|
|
|
|
return {k: list(v) for k, v in counters.items()}
|
|
|
|
|
|
def parse_test_times(fname):
|
|
read = False
|
|
description_output = []
|
|
with open(fname, "r", encoding="utf-8") as out:
|
|
for line in out:
|
|
if read and "==" in line:
|
|
break
|
|
if read and line.strip():
|
|
description_output.append(line.strip())
|
|
if "slowest durations" in line:
|
|
read = True
|
|
return description_output
|
|
|
|
|
|
def get_test_times(output):
|
|
result = defaultdict(float)
|
|
for line in output:
|
|
if ".py" in line:
|
|
line_arr = line.strip().split(" ")
|
|
test_time = line_arr[0]
|
|
test_name = " ".join([elem for elem in line_arr[2:] if elem])
|
|
if test_name not in result:
|
|
result[test_name] = 0.0
|
|
result[test_name] += float(test_time[:-1])
|
|
return result
|
|
|
|
|
|
def clear_ip_tables_and_restart_daemons():
|
|
logging.info(
|
|
"Dump iptables after run %s",
|
|
subprocess.check_output("sudo iptables -nvL", shell=True),
|
|
)
|
|
try:
|
|
logging.info("Killing all alive docker containers")
|
|
subprocess.check_output(
|
|
"timeout --signal=KILL 10m docker ps --quiet | xargs --no-run-if-empty docker kill",
|
|
shell=True,
|
|
)
|
|
except subprocess.CalledProcessError as err:
|
|
logging.info("docker kill excepted: %s", str(err))
|
|
|
|
try:
|
|
logging.info("Removing all docker containers")
|
|
subprocess.check_output(
|
|
"timeout --signal=KILL 10m docker ps --all --quiet | xargs --no-run-if-empty docker rm --force",
|
|
shell=True,
|
|
)
|
|
except subprocess.CalledProcessError as err:
|
|
logging.info("docker rm excepted: %s", str(err))
|
|
|
|
# don't restart docker if it's disabled
|
|
if os.environ.get("CLICKHOUSE_TESTS_RUNNER_RESTART_DOCKER", "1") == "1":
|
|
try:
|
|
logging.info("Stopping docker daemon")
|
|
subprocess.check_output("service docker stop", shell=True)
|
|
except subprocess.CalledProcessError as err:
|
|
logging.info("docker stop excepted: %s", str(err))
|
|
|
|
try:
|
|
for i in range(200):
|
|
try:
|
|
logging.info("Restarting docker %s", i)
|
|
subprocess.check_output("service docker start", shell=True)
|
|
subprocess.check_output("docker ps", shell=True)
|
|
break
|
|
except subprocess.CalledProcessError as err:
|
|
time.sleep(0.5)
|
|
logging.info("Waiting docker to start, current %s", str(err))
|
|
else:
|
|
raise RuntimeError("Docker daemon doesn't responding")
|
|
except subprocess.CalledProcessError as err:
|
|
logging.info("Can't reload docker: %s", str(err))
|
|
|
|
iptables_iter = 0
|
|
try:
|
|
for i in range(1000):
|
|
iptables_iter = i
|
|
# when rules will be empty, it will raise exception
|
|
subprocess.check_output("sudo iptables -D DOCKER-USER 1", shell=True)
|
|
except subprocess.CalledProcessError as err:
|
|
logging.info(
|
|
"All iptables rules cleared, %s iterations, last error: %s",
|
|
iptables_iter,
|
|
str(err),
|
|
)
|
|
|
|
|
|
class ClickhouseIntegrationTestsRunner:
|
|
def __init__(self, repo_path: str, result_path: str, params: dict):
|
|
self.repo_path = repo_path
|
|
self.result_path = result_path
|
|
self.params = params
|
|
|
|
self.image_versions = self.params["docker_images_with_versions"]
|
|
self.shuffle_groups = self.params["shuffle_test_groups"]
|
|
self.flaky_check = "flaky check" in self.params["context_name"]
|
|
self.bugfix_validate_check = "bugfix" in self.params["context_name"].lower()
|
|
# if use_tmpfs is not set we assume it to be true, otherwise check
|
|
self.use_tmpfs = "use_tmpfs" not in self.params or self.params["use_tmpfs"]
|
|
self.disable_net_host = (
|
|
"disable_net_host" in self.params and self.params["disable_net_host"]
|
|
)
|
|
self.start_time = time.time()
|
|
self.soft_deadline_time = self.start_time + (TASK_TIMEOUT - MAX_TIME_IN_SANDBOX)
|
|
|
|
self.use_old_analyzer = (
|
|
os.environ.get("CLICKHOUSE_USE_OLD_ANALYZER") is not None
|
|
)
|
|
|
|
if "run_by_hash_total" in self.params:
|
|
self.run_by_hash_total = self.params["run_by_hash_total"]
|
|
self.run_by_hash_num = self.params["run_by_hash_num"]
|
|
else:
|
|
self.run_by_hash_total = 0
|
|
self.run_by_hash_num = 0
|
|
|
|
def path(self):
|
|
return self.result_path
|
|
|
|
def base_path(self):
|
|
return os.path.join(str(self.result_path), "../")
|
|
|
|
@staticmethod
|
|
def should_skip_tests():
|
|
return []
|
|
|
|
def get_image_with_version(self, name):
|
|
if name in self.image_versions:
|
|
return name + ":" + self.image_versions[name]
|
|
logging.warning(
|
|
"Cannot find image %s in params list %s", name, self.image_versions
|
|
)
|
|
if ":" not in name:
|
|
return name + ":latest"
|
|
return name
|
|
|
|
def get_image_version(self, name: str) -> Any:
|
|
if name in self.image_versions:
|
|
return self.image_versions[name]
|
|
logging.warning(
|
|
"Cannot find image %s in params list %s", name, self.image_versions
|
|
)
|
|
return "latest"
|
|
|
|
def shuffle_test_groups(self):
|
|
return self.shuffle_groups != 0
|
|
|
|
def _pre_pull_images(self):
|
|
image_cmd = self._get_runner_image_cmd()
|
|
|
|
cmd = (
|
|
f"cd {self.repo_path}/tests/integration && "
|
|
f"timeout --signal=KILL 1h ./runner {self._get_runner_opts()} {image_cmd} "
|
|
"--pre-pull --command ' echo Pre Pull finished ' "
|
|
)
|
|
|
|
for i in range(5):
|
|
logging.info("Pulling images before running tests. Attempt %s", i)
|
|
try:
|
|
subprocess.check_output(
|
|
cmd,
|
|
shell=True,
|
|
)
|
|
return
|
|
except subprocess.CalledProcessError as err:
|
|
logging.info("docker-compose pull failed: %s", str(err))
|
|
continue
|
|
message = "Pulling images failed for 5 attempts. Will fail the worker."
|
|
logging.error(message)
|
|
kill_ci_runner(message)
|
|
# We pass specific retcode to to ci/integration_test_check.py to skip status reporting and restart job
|
|
sys.exit(13)
|
|
|
|
@staticmethod
|
|
def _can_run_with(path, opt):
|
|
with open(path, "r", encoding="utf-8") as script:
|
|
for line in script:
|
|
if opt in line:
|
|
return True
|
|
return False
|
|
|
|
def _install_clickhouse(self, debs_path):
|
|
for package in (
|
|
"clickhouse-common-static_",
|
|
"clickhouse-server_",
|
|
"clickhouse-client",
|
|
"clickhouse-odbc-bridge_",
|
|
"clickhouse-library-bridge_",
|
|
"clickhouse-common-static-dbg_",
|
|
): # order matters
|
|
logging.info("Installing package %s", package)
|
|
for f in os.listdir(debs_path):
|
|
if package in f:
|
|
full_path = os.path.join(debs_path, f)
|
|
logging.info("Package found in %s", full_path)
|
|
log_name = "install_" + f + ".log"
|
|
log_path = os.path.join(str(self.path()), log_name)
|
|
cmd = f"dpkg -x {full_path} ."
|
|
logging.info("Executing installation cmd %s", cmd)
|
|
with TeePopen(cmd, log_file=log_path) as proc:
|
|
if proc.wait() == 0:
|
|
logging.info("Installation of %s successfull", full_path)
|
|
else:
|
|
raise RuntimeError(f"Installation of {full_path} failed")
|
|
break
|
|
else:
|
|
raise FileNotFoundError(f"Package with {package} not found")
|
|
# logging.info("Unstripping binary")
|
|
# logging.info(
|
|
# "Unstring %s",
|
|
# subprocess.check_output(
|
|
# "eu-unstrip /usr/bin/clickhouse {}".format(CLICKHOUSE_BINARY_PATH),
|
|
# shell=True,
|
|
# ),
|
|
# )
|
|
|
|
logging.info("All packages installed")
|
|
os.chmod(CLICKHOUSE_BINARY_PATH, 0o777)
|
|
os.chmod(CLICKHOUSE_ODBC_BRIDGE_BINARY_PATH, 0o777)
|
|
os.chmod(CLICKHOUSE_LIBRARY_BRIDGE_BINARY_PATH, 0o777)
|
|
shutil.copy(
|
|
CLICKHOUSE_BINARY_PATH, os.getenv("CLICKHOUSE_TESTS_SERVER_BIN_PATH") # type: ignore
|
|
)
|
|
shutil.copy(
|
|
CLICKHOUSE_ODBC_BRIDGE_BINARY_PATH,
|
|
os.getenv("CLICKHOUSE_TESTS_ODBC_BRIDGE_BIN_PATH"), # type: ignore
|
|
)
|
|
shutil.copy(
|
|
CLICKHOUSE_LIBRARY_BRIDGE_BINARY_PATH,
|
|
os.getenv("CLICKHOUSE_TESTS_LIBRARY_BRIDGE_BIN_PATH"), # type: ignore
|
|
)
|
|
|
|
@staticmethod
|
|
def _compress_logs(directory, relpaths, result_path):
|
|
retcode = subprocess.call(
|
|
f"sudo tar --use-compress-program='zstd --threads=0' "
|
|
f"-cf {result_path} -C {directory} {' '.join(relpaths)}",
|
|
shell=True,
|
|
)
|
|
# tar return 1 when the files are changed on compressing, we ignore it
|
|
if retcode in (0, 1):
|
|
return
|
|
# but even on the fatal errors it's better to retry
|
|
logging.error("Fatal error on compressing %s: %s", result_path, retcode)
|
|
|
|
def _get_runner_opts(self):
|
|
result = []
|
|
if self.use_tmpfs:
|
|
result.append("--tmpfs")
|
|
if self.disable_net_host:
|
|
result.append("--disable-net-host")
|
|
if self.use_old_analyzer:
|
|
result.append("--old-analyzer")
|
|
|
|
return " ".join(result)
|
|
|
|
def _get_all_tests(self) -> List[str]:
|
|
image_cmd = self._get_runner_image_cmd()
|
|
runner_opts = self._get_runner_opts()
|
|
out_file_full = os.path.join(self.result_path, "runner_get_all_tests.log")
|
|
cmd = (
|
|
f"cd {self.repo_path}/tests/integration && "
|
|
f"timeout --signal=KILL 1h ./runner {runner_opts} {image_cmd} -- --setup-plan "
|
|
)
|
|
|
|
logging.info(
|
|
"Getting all tests to the file %s with cmd: \n%s", out_file_full, cmd
|
|
)
|
|
with open(out_file_full, "wb") as ofd:
|
|
try:
|
|
subprocess.check_call(cmd, shell=True, stdout=ofd, stderr=ofd)
|
|
except subprocess.CalledProcessError as ex:
|
|
print("ERROR: Setting test plan failed. Output:")
|
|
with open(out_file_full, "r", encoding="utf-8") as file:
|
|
for line in file:
|
|
print(" " + line, end="")
|
|
raise ex
|
|
|
|
all_tests = set()
|
|
with open(out_file_full, "r", encoding="utf-8") as all_tests_fd:
|
|
for line in all_tests_fd:
|
|
if (
|
|
line[0] in string.whitespace # test names at the start of lines
|
|
or "::test" not in line # test names contain '::test'
|
|
or "SKIPPED" in line # pytest.mark.skip/-if
|
|
):
|
|
continue
|
|
all_tests.add(line.strip())
|
|
|
|
assert all_tests
|
|
|
|
return list(sorted(all_tests))
|
|
|
|
@staticmethod
|
|
def _get_parallel_tests_skip_list(repo_path):
|
|
skip_list_file_path = f"{repo_path}/tests/integration/parallel_skip.json"
|
|
if (
|
|
not os.path.isfile(skip_list_file_path)
|
|
or os.path.getsize(skip_list_file_path) == 0
|
|
):
|
|
raise ValueError(
|
|
"There is something wrong with getting all tests list: "
|
|
f"file '{skip_list_file_path}' is empty or does not exist."
|
|
)
|
|
|
|
skip_list_tests = []
|
|
with open(skip_list_file_path, "r", encoding="utf-8") as skip_list_file:
|
|
skip_list_tests = json.load(skip_list_file)
|
|
return list(sorted(skip_list_tests))
|
|
|
|
@staticmethod
|
|
def group_test_by_file(tests):
|
|
result = {} # type: Dict
|
|
for test in tests:
|
|
test_file = test.split("::")[0]
|
|
if test_file not in result:
|
|
result[test_file] = []
|
|
result[test_file].append(test)
|
|
return result
|
|
|
|
@staticmethod
|
|
def _update_counters(main_counters, current_counters):
|
|
for test in current_counters["PASSED"]:
|
|
if test not in main_counters["PASSED"]:
|
|
if test in main_counters["FAILED"]:
|
|
main_counters["FAILED"].remove(test)
|
|
if test in main_counters["BROKEN"]:
|
|
main_counters["BROKEN"].remove(test)
|
|
|
|
main_counters["PASSED"].append(test)
|
|
|
|
for state in ("ERROR", "FAILED"):
|
|
for test in current_counters[state]:
|
|
if test in main_counters["PASSED"]:
|
|
main_counters["PASSED"].remove(test)
|
|
if test not in main_counters[state]:
|
|
main_counters[state].append(test)
|
|
|
|
for state in ("SKIPPED",):
|
|
for test in current_counters[state]:
|
|
main_counters[state].append(test)
|
|
|
|
def _get_runner_image_cmd(self):
|
|
image_cmd = ""
|
|
if self._can_run_with(
|
|
os.path.join(self.repo_path, "tests/integration", "runner"),
|
|
"--docker-image-version",
|
|
):
|
|
for img in IMAGES:
|
|
if img == "clickhouse/integration-tests-runner":
|
|
runner_version = self.get_image_version(img)
|
|
logging.info(
|
|
"Can run with custom docker image version %s", runner_version
|
|
)
|
|
image_cmd += f" --docker-image-version={runner_version} "
|
|
else:
|
|
if self._can_run_with(
|
|
os.path.join(self.repo_path, "tests/integration", "runner"),
|
|
"--docker-compose-images-tags",
|
|
):
|
|
image_cmd += (
|
|
"--docker-compose-images-tags="
|
|
f"{self.get_image_with_version(img)} "
|
|
)
|
|
else:
|
|
image_cmd = ""
|
|
logging.info("Cannot run with custom docker image version :(")
|
|
return image_cmd
|
|
|
|
@staticmethod
|
|
def _find_test_data_dirs(repo_path, test_names):
|
|
relpaths = {}
|
|
for test_name in test_names:
|
|
if "/" in test_name:
|
|
test_dir = test_name[: test_name.find("/")]
|
|
else:
|
|
test_dir = test_name
|
|
if os.path.isdir(os.path.join(repo_path, "tests/integration", test_dir)):
|
|
for name in os.listdir(
|
|
os.path.join(repo_path, "tests/integration", test_dir)
|
|
):
|
|
relpath = os.path.join(os.path.join(test_dir, name))
|
|
mtime = os.path.getmtime(
|
|
os.path.join(repo_path, "tests/integration", relpath)
|
|
)
|
|
relpaths[relpath] = mtime
|
|
return relpaths
|
|
|
|
@staticmethod
|
|
def _get_test_data_dirs_difference(new_snapshot, old_snapshot):
|
|
res = set()
|
|
for path in new_snapshot:
|
|
if (path not in old_snapshot) or (old_snapshot[path] != new_snapshot[path]):
|
|
res.add(path)
|
|
return res
|
|
|
|
def try_run_test_group(
|
|
self,
|
|
test_group,
|
|
tests_in_group,
|
|
num_tries,
|
|
num_workers,
|
|
repeat_count,
|
|
):
|
|
try:
|
|
return self.run_test_group(
|
|
test_group,
|
|
tests_in_group,
|
|
num_tries,
|
|
num_workers,
|
|
repeat_count,
|
|
)
|
|
except Exception as e:
|
|
logging.info("Failed to run %s:\n%s", test_group, e)
|
|
counters = {
|
|
"ERROR": [],
|
|
"PASSED": [],
|
|
"FAILED": [],
|
|
"SKIPPED": [],
|
|
} # type: Dict
|
|
tests_times = defaultdict(float) # type: Dict
|
|
for test in tests_in_group:
|
|
counters["ERROR"].append(test)
|
|
tests_times[test] = 0
|
|
return counters, tests_times, []
|
|
|
|
def run_test_group(
|
|
self,
|
|
test_group,
|
|
tests_in_group,
|
|
num_tries,
|
|
num_workers,
|
|
repeat_count,
|
|
):
|
|
counters = {
|
|
"ERROR": [],
|
|
"PASSED": [],
|
|
"FAILED": [],
|
|
"SKIPPED": [],
|
|
"BROKEN": [],
|
|
"NOT_FAILED": [],
|
|
} # type: Dict
|
|
tests_times = defaultdict(float) # type: Dict
|
|
|
|
if self.soft_deadline_time < time.time():
|
|
for test in tests_in_group:
|
|
logging.info("Task timeout exceeded, skipping %s", test)
|
|
counters["SKIPPED"].append(test)
|
|
tests_times[test] = 0
|
|
return counters, tests_times, []
|
|
|
|
image_cmd = self._get_runner_image_cmd()
|
|
test_group_str = test_group.replace("/", "_").replace(".", "_")
|
|
|
|
log_paths = []
|
|
test_data_dirs = {}
|
|
|
|
for i in range(num_tries):
|
|
if timeout_expired:
|
|
print("Timeout expired - break test group execution")
|
|
break
|
|
logging.info("Running test group %s for the %s retry", test_group, i)
|
|
clear_ip_tables_and_restart_daemons()
|
|
|
|
test_names = set([])
|
|
for test_name in tests_in_group:
|
|
if test_name not in counters["PASSED"]:
|
|
test_names.add(test_name)
|
|
|
|
if i == 0:
|
|
test_data_dirs = self._find_test_data_dirs(self.repo_path, test_names)
|
|
|
|
info_basename = test_group_str + "_" + str(i) + ".nfo"
|
|
info_path = os.path.join(self.repo_path, "tests/integration", info_basename)
|
|
|
|
test_cmd = " ".join([shlex.quote(test) for test in sorted(test_names)])
|
|
parallel_cmd = f" --parallel {num_workers} " if num_workers > 0 else ""
|
|
repeat_cmd = f" --count {repeat_count} " if repeat_count > 0 else ""
|
|
# -r -- show extra test summary:
|
|
# -f -- (f)ailed
|
|
# -E -- (E)rror
|
|
# -p -- (p)assed
|
|
# -s -- (s)kipped
|
|
cmd = (
|
|
f"cd {self.repo_path}/tests/integration && "
|
|
f"timeout --signal=KILL 1h ./runner {self._get_runner_opts()} "
|
|
f"{image_cmd} -t {test_cmd} {parallel_cmd} {repeat_cmd} -- -rfEps --run-id={i} "
|
|
f"--color=no --durations=0 {_get_deselect_option(self.should_skip_tests())} "
|
|
f"| tee {info_path}"
|
|
)
|
|
|
|
log_basename = test_group_str + "_" + str(i) + ".log"
|
|
log_path = os.path.join(self.repo_path, "tests/integration", log_basename)
|
|
logging.info("Executing cmd: %s", cmd)
|
|
# ignore retcode, since it meaningful due to pipe to tee
|
|
with TeePopen(cmd, log_path) as proc:
|
|
global runner_subprocess # pylint:disable=global-statement
|
|
runner_subprocess = proc
|
|
proc.wait()
|
|
|
|
extra_logs_names = [log_basename]
|
|
log_result_path = os.path.join(
|
|
str(self.path()), "integration_run_" + log_basename
|
|
)
|
|
shutil.copy(log_path, log_result_path)
|
|
log_paths.append(log_result_path)
|
|
|
|
for pytest_log_path in glob.glob(
|
|
os.path.join(self.repo_path, "tests/integration/pytest*.log")
|
|
):
|
|
new_name = (
|
|
test_group_str
|
|
+ "_"
|
|
+ str(i)
|
|
+ "_"
|
|
+ os.path.basename(pytest_log_path)
|
|
)
|
|
os.rename(
|
|
pytest_log_path,
|
|
os.path.join(self.repo_path, "tests/integration", new_name),
|
|
)
|
|
extra_logs_names.append(new_name)
|
|
|
|
dockerd_log_path = os.path.join(
|
|
self.repo_path, "tests/integration/dockerd.log"
|
|
)
|
|
if os.path.exists(dockerd_log_path):
|
|
new_name = (
|
|
test_group_str
|
|
+ "_"
|
|
+ str(i)
|
|
+ "_"
|
|
+ os.path.basename(dockerd_log_path)
|
|
)
|
|
os.rename(
|
|
dockerd_log_path,
|
|
os.path.join(self.repo_path, "tests/integration", new_name),
|
|
)
|
|
extra_logs_names.append(new_name)
|
|
|
|
if os.path.exists(info_path):
|
|
extra_logs_names.append(info_basename)
|
|
new_counters = get_counters(info_path)
|
|
for state, tests in new_counters.items():
|
|
logging.info(
|
|
"Tests with %s state (%s): %s", state, len(tests), tests
|
|
)
|
|
times_lines = parse_test_times(info_path)
|
|
new_tests_times = get_test_times(times_lines)
|
|
self._update_counters(counters, new_counters)
|
|
for test_name, test_time in new_tests_times.items():
|
|
tests_times[test_name] = test_time
|
|
|
|
test_data_dirs_new = self._find_test_data_dirs(self.repo_path, test_names)
|
|
test_data_dirs_diff = self._get_test_data_dirs_difference(
|
|
test_data_dirs_new, test_data_dirs
|
|
)
|
|
test_data_dirs = test_data_dirs_new
|
|
|
|
if extra_logs_names or test_data_dirs_diff:
|
|
extras_result_path = os.path.join(
|
|
str(self.path()),
|
|
"integration_run_" + test_group_str + "_" + str(i) + ".tar.zst",
|
|
)
|
|
self._compress_logs(
|
|
os.path.join(self.repo_path, "tests/integration"),
|
|
extra_logs_names + list(test_data_dirs_diff),
|
|
extras_result_path,
|
|
)
|
|
log_paths.append(extras_result_path)
|
|
|
|
if len(counters["PASSED"]) == len(tests_in_group):
|
|
logging.info("All tests from group %s passed", test_group)
|
|
break
|
|
if (
|
|
len(counters["PASSED"]) >= 0
|
|
and len(counters["FAILED"]) == 0
|
|
and len(counters["ERROR"]) == 0
|
|
):
|
|
logging.info(
|
|
"Seems like all tests passed but some of them are skipped or "
|
|
"deselected. Ignoring them and finishing group."
|
|
)
|
|
break
|
|
else:
|
|
# Mark all non tried tests as errors, with '::' in name
|
|
# (example test_partition/test.py::test_partition_simple). For flaky check
|
|
# we run whole test dirs like "test_odbc_interaction" and don't
|
|
# want to mark them as error so we filter by '::'.
|
|
for test in tests_in_group:
|
|
if (
|
|
test
|
|
not in chain(
|
|
counters["PASSED"],
|
|
counters["ERROR"],
|
|
counters["SKIPPED"],
|
|
counters["FAILED"],
|
|
counters["BROKEN"],
|
|
)
|
|
and "::" in test
|
|
):
|
|
counters["ERROR"].append(test)
|
|
|
|
return counters, tests_times, log_paths
|
|
|
|
def run_flaky_check(self, build_path, should_fail=False):
|
|
pr_info = self.params["pr_info"]
|
|
|
|
tests_to_run = get_changed_tests_to_run(pr_info, self.repo_path)
|
|
if not tests_to_run:
|
|
logging.info("No integration tests to run found")
|
|
return "success", NO_CHANGES_MSG, [(NO_CHANGES_MSG, "OK")], ""
|
|
|
|
self._install_clickhouse(build_path)
|
|
logging.info("Found '%s' tests to run", " ".join(tests_to_run))
|
|
result_state = "success"
|
|
description_prefix = "No flaky tests: "
|
|
logging.info("Starting check with retries")
|
|
final_retry = 0
|
|
counters = {
|
|
"ERROR": [],
|
|
"PASSED": [],
|
|
"FAILED": [],
|
|
"SKIPPED": [],
|
|
"BROKEN": [],
|
|
"NOT_FAILED": [],
|
|
} # type: Dict
|
|
tests_times = defaultdict(float) # type: Dict
|
|
tests_log_paths = defaultdict(list)
|
|
id_counter = 0
|
|
for test_to_run in tests_to_run:
|
|
tries_num = 1 if should_fail else FLAKY_TRIES_COUNT
|
|
for i in range(tries_num):
|
|
if timeout_expired:
|
|
print("Timeout expired - break flaky check execution")
|
|
break
|
|
final_retry += 1
|
|
logging.info("Running tests for the %s time", i)
|
|
group_counters, group_test_times, log_paths = self.try_run_test_group(
|
|
f"bugfix_{id_counter}" if should_fail else f"flaky{id_counter}",
|
|
[test_to_run],
|
|
1,
|
|
1,
|
|
FLAKY_REPEAT_COUNT,
|
|
)
|
|
id_counter = id_counter + 1
|
|
for counter, value in group_counters.items():
|
|
logging.info(
|
|
"Tests from group %s stats, %s count %s",
|
|
test_to_run,
|
|
counter,
|
|
len(value),
|
|
)
|
|
counters[counter] += value
|
|
|
|
for test_name, test_time in group_test_times.items():
|
|
tests_times[test_name] = test_time
|
|
tests_log_paths[test_name] = log_paths
|
|
if not should_fail and (
|
|
group_counters["FAILED"] or group_counters["ERROR"]
|
|
):
|
|
logging.info(
|
|
"Unexpected failure in group %s. Fail fast for current group",
|
|
test_to_run,
|
|
)
|
|
break
|
|
|
|
if counters["FAILED"]:
|
|
logging.info("Found failed tests: %s", " ".join(counters["FAILED"]))
|
|
description_prefix = "Failed tests found: "
|
|
result_state = "failure"
|
|
if counters["ERROR"]:
|
|
description_prefix = "Failed tests found: "
|
|
logging.info("Found error tests: %s", " ".join(counters["ERROR"]))
|
|
# NOTE "error" result state will restart the whole test task,
|
|
# so we use "failure" here
|
|
result_state = "failure"
|
|
logging.info("Try is OK, all tests passed, going to clear env")
|
|
clear_ip_tables_and_restart_daemons()
|
|
logging.info("And going to sleep for some time")
|
|
time.sleep(5)
|
|
|
|
test_result = []
|
|
for state in ("ERROR", "FAILED", "PASSED", "SKIPPED"):
|
|
if state == "PASSED":
|
|
text_state = "OK"
|
|
elif state == "FAILED":
|
|
text_state = "FAIL"
|
|
else:
|
|
text_state = state
|
|
test_result += [
|
|
(c, text_state, f"{tests_times[c]:.2f}", tests_log_paths[c])
|
|
for c in counters[state]
|
|
]
|
|
|
|
status_text = description_prefix + ", ".join(
|
|
[
|
|
str(n).lower().replace("failed", "fail") + ": " + str(len(c))
|
|
for n, c in counters.items()
|
|
]
|
|
)
|
|
|
|
return result_state, status_text, test_result, tests_log_paths
|
|
|
|
def run_impl(self, build_path):
|
|
stopwatch = Stopwatch()
|
|
if self.flaky_check or self.bugfix_validate_check:
|
|
result_state, status_text, test_result, tests_log_paths = (
|
|
self.run_flaky_check(build_path, should_fail=self.bugfix_validate_check)
|
|
)
|
|
else:
|
|
result_state, status_text, test_result, tests_log_paths = (
|
|
self.run_normal_check(build_path)
|
|
)
|
|
|
|
if self.soft_deadline_time < time.time():
|
|
status_text = "Timeout, " + status_text
|
|
result_state = "failure"
|
|
|
|
if timeout_expired:
|
|
logging.error(
|
|
"Job killed by external timeout signal - setting status to failure!"
|
|
)
|
|
status_text = "Job timeout expired, " + status_text
|
|
result_state = "failure"
|
|
# add mock test case to make timeout visible in job report and in ci db
|
|
test_result.insert(
|
|
0, (JOB_TIMEOUT_TEST_NAME, "FAIL", f"{stopwatch.duration_seconds}", "")
|
|
)
|
|
|
|
if "(memory)" in self.params["context_name"]:
|
|
result_state = "success"
|
|
|
|
return result_state, status_text, test_result, tests_log_paths
|
|
|
|
def _get_tests_by_hash(self) -> List[str]:
|
|
"Tries it's best to group the tests equally between groups"
|
|
all_tests = self._get_all_tests()
|
|
if self.run_by_hash_total == 0:
|
|
return all_tests
|
|
grouped_tests = self.group_test_by_file(all_tests)
|
|
groups_by_hash = {
|
|
g: [] for g in range(self.run_by_hash_total)
|
|
} # type: Dict[int, List[str]]
|
|
for tests_in_group in grouped_tests.values():
|
|
# It should work determenistic, because it searches groups with min tests
|
|
min_group = min(len(tests) for tests in groups_by_hash.values())
|
|
# And then it takes a group with min index
|
|
group_to_increase = min(
|
|
g for g, t in groups_by_hash.items() if len(t) == min_group
|
|
)
|
|
groups_by_hash[group_to_increase].extend(tests_in_group)
|
|
return groups_by_hash[self.run_by_hash_num]
|
|
|
|
def run_normal_check(self, build_path):
|
|
self._install_clickhouse(build_path)
|
|
logging.info("Pulling images")
|
|
self._pre_pull_images()
|
|
logging.info(
|
|
"Dump iptables before run %s",
|
|
subprocess.check_output("sudo iptables -nvL", shell=True),
|
|
)
|
|
all_tests = self._get_tests_by_hash()
|
|
parallel_skip_tests = self._get_parallel_tests_skip_list(self.repo_path)
|
|
logging.info(
|
|
"Found %s tests first 3 %s", len(all_tests), " ".join(all_tests[:3])
|
|
)
|
|
filtered_sequential_tests = list(
|
|
filter(lambda test: has_test(all_tests, test), parallel_skip_tests)
|
|
)
|
|
filtered_parallel_tests = list(
|
|
filter(
|
|
lambda test: not has_test(parallel_skip_tests, test),
|
|
all_tests,
|
|
)
|
|
)
|
|
not_found_tests = list(
|
|
filter(
|
|
lambda test: not has_test(all_tests, test),
|
|
parallel_skip_tests,
|
|
)
|
|
)
|
|
logging.info(
|
|
"Found %s tests first 3 %s, parallel %s, other %s",
|
|
len(all_tests),
|
|
" ".join(all_tests[:3]),
|
|
len(filtered_parallel_tests),
|
|
len(filtered_sequential_tests),
|
|
)
|
|
logging.info(
|
|
"Not found %s tests first 3 %s",
|
|
len(not_found_tests),
|
|
" ".join(not_found_tests[:3]),
|
|
)
|
|
grouped_tests = self.group_test_by_file(filtered_sequential_tests)
|
|
i = 0
|
|
for par_group in chunks(filtered_parallel_tests, PARALLEL_GROUP_SIZE):
|
|
grouped_tests[f"parallel{i}"] = par_group
|
|
i += 1
|
|
logging.info("Found %s tests groups", len(grouped_tests))
|
|
counters = {
|
|
"ERROR": [],
|
|
"PASSED": [],
|
|
"FAILED": [],
|
|
"SKIPPED": [],
|
|
"BROKEN": [],
|
|
"NOT_FAILED": [],
|
|
} # type: Dict
|
|
tests_times = defaultdict(float)
|
|
tests_log_paths = defaultdict(list)
|
|
items_to_run = list(grouped_tests.items())
|
|
logging.info("Total test groups %s", len(items_to_run))
|
|
if self.shuffle_test_groups():
|
|
logging.info("Shuffling test groups")
|
|
random.shuffle(items_to_run)
|
|
for group, tests in items_to_run:
|
|
if timeout_expired:
|
|
print("Timeout expired - break tests execution")
|
|
break
|
|
logging.info("Running test group %s containing %s tests", group, len(tests))
|
|
group_counters, group_test_times, log_paths = self.try_run_test_group(
|
|
group, tests, MAX_RETRY, NUM_WORKERS, 0
|
|
)
|
|
total_tests = 0
|
|
for counter, value in group_counters.items():
|
|
logging.info(
|
|
"Tests from group %s stats, %s count %s", group, counter, len(value)
|
|
)
|
|
counters[counter] += value
|
|
logging.info(
|
|
"Totally have %s with status %s", len(counters[counter]), counter
|
|
)
|
|
total_tests += len(counters[counter])
|
|
logging.info("Totally finished tests %s/%s", total_tests, len(all_tests))
|
|
|
|
for test_name, test_time in group_test_times.items():
|
|
tests_times[test_name] = test_time
|
|
tests_log_paths[test_name] = log_paths
|
|
|
|
if len(counters["FAILED"]) + len(counters["ERROR"]) >= 20:
|
|
logging.info("Collected more than 20 failed/error tests, stopping")
|
|
break
|
|
if counters["FAILED"] or counters["ERROR"]:
|
|
logging.info(
|
|
"Overall status failure, because we have tests in FAILED or ERROR state"
|
|
)
|
|
result_state = "failure"
|
|
else:
|
|
logging.info("Overall success!")
|
|
result_state = "success"
|
|
test_result = []
|
|
for state in (
|
|
"ERROR",
|
|
"FAILED",
|
|
"PASSED",
|
|
"SKIPPED",
|
|
"BROKEN",
|
|
"NOT_FAILED",
|
|
):
|
|
if state == "PASSED":
|
|
text_state = "OK"
|
|
elif state == "FAILED":
|
|
text_state = "FAIL"
|
|
else:
|
|
text_state = state
|
|
test_result += [
|
|
(c, text_state, f"{tests_times[c]:.2f}", tests_log_paths[c])
|
|
for c in counters[state]
|
|
]
|
|
failed_sum = len(counters["FAILED"]) + len(counters["ERROR"])
|
|
status_text = f"fail: {failed_sum}, passed: {len(counters['PASSED'])}"
|
|
|
|
if not counters or sum(len(counter) for counter in counters.values()) == 0:
|
|
status_text = "No tests found for some reason! It's a bug"
|
|
result_state = "failure"
|
|
|
|
return result_state, status_text, test_result, tests_log_paths
|
|
|
|
|
|
def write_results(results_file, status_file, results, status):
|
|
with open(results_file, "w", encoding="utf-8") as f:
|
|
out = csv.writer(f, delimiter="\t")
|
|
out.writerows(results)
|
|
with open(status_file, "w", encoding="utf-8") as f:
|
|
out = csv.writer(f, delimiter="\t")
|
|
out.writerow(status)
|
|
|
|
|
|
def run():
|
|
signal.signal(signal.SIGTERM, handle_sigterm)
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)s %(message)s")
|
|
|
|
repo_path = os.environ.get("CLICKHOUSE_TESTS_REPO_PATH", "")
|
|
build_path = os.environ.get("CLICKHOUSE_TESTS_BUILD_PATH", "")
|
|
result_path = os.environ.get("CLICKHOUSE_TESTS_RESULT_PATH", "")
|
|
params_path = os.environ.get("CLICKHOUSE_TESTS_JSON_PARAMS_PATH", "")
|
|
|
|
assert all((repo_path, build_path, result_path, params_path))
|
|
|
|
with open(params_path, "r", encoding="utf-8") as jfd:
|
|
params = json.loads(jfd.read())
|
|
runner = ClickhouseIntegrationTestsRunner(repo_path, result_path, params)
|
|
|
|
logging.info("Running tests")
|
|
|
|
if IS_CI:
|
|
# Avoid overlaps with previous runs
|
|
logging.info("Clearing dmesg before run")
|
|
subprocess.check_call("sudo -E dmesg --clear", shell=True)
|
|
|
|
state, description, test_results, _test_log_paths = runner.run_impl(build_path)
|
|
logging.info("Tests finished")
|
|
|
|
if IS_CI:
|
|
# Dump dmesg (to capture possible OOMs)
|
|
logging.info("Dumping dmesg")
|
|
subprocess.check_call("sudo -E dmesg -T", shell=True)
|
|
|
|
status = (state, description)
|
|
out_results_file = os.path.join(str(runner.path()), "test_results.tsv")
|
|
out_status_file = os.path.join(str(runner.path()), "check_status.tsv")
|
|
write_results(out_results_file, out_status_file, test_results, status)
|
|
logging.info("Result written")
|
|
|
|
|
|
timeout_expired = False
|
|
runner_subprocess = None # type:Optional[TeePopen]
|
|
|
|
|
|
def handle_sigterm(signum, _frame):
|
|
# TODO: think on how to process it without globals?
|
|
print(f"WARNING: Received signal {signum}")
|
|
global timeout_expired # pylint:disable=global-statement
|
|
timeout_expired = True
|
|
if runner_subprocess:
|
|
runner_subprocess.terminate()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run()
|