mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-03 04:19:18 +08:00
116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
import os
|
|
import threading
|
|
import time
|
|
import glob
|
|
from chaos import constants
|
|
from yaml import full_load
|
|
from utils.util_log import test_log as log
|
|
from delayed_assert import expect
|
|
import pytest
|
|
|
|
|
|
def check_config(chaos_config):
|
|
if not chaos_config.get("kind", None):
|
|
raise Exception("kind must be specified")
|
|
if not chaos_config.get("spec", None):
|
|
raise Exception("spec must be specified")
|
|
if "action" not in chaos_config.get("spec", None):
|
|
raise Exception("action must be specified in spec")
|
|
if "selector" not in chaos_config.get("spec", None):
|
|
raise Exception("selector must be specified in spec")
|
|
return True
|
|
|
|
|
|
def reset_counting(checkers={}):
|
|
"""reset checker counts for all checker threads"""
|
|
for ch in checkers.values():
|
|
ch.reset()
|
|
|
|
|
|
def gen_experiment_config(yaml):
|
|
"""load the yaml file of chaos experiment"""
|
|
with open(yaml) as f:
|
|
_config = full_load(f)
|
|
f.close()
|
|
return _config
|
|
|
|
|
|
def start_monitor_threads(checkers={}):
|
|
"""start the threads by checkers"""
|
|
for k, ch in checkers.items():
|
|
ch._keep_running = True
|
|
t = threading.Thread(target=ch.keep_running, args=(), name=k, daemon=True)
|
|
t.start()
|
|
|
|
|
|
def get_env_variable_by_name(name):
|
|
"""get env variable by name"""
|
|
try:
|
|
env_var = os.environ[name]
|
|
log.debug(f"env_variable: {env_var}")
|
|
return str(env_var)
|
|
except Exception as e:
|
|
log.debug(f"fail to get env variables, error: {str(e)}")
|
|
return None
|
|
|
|
|
|
def get_chaos_yamls():
|
|
"""get chaos yaml file(s) from configured environment path"""
|
|
chaos_env = get_env_variable_by_name(constants.CHAOS_CONFIG_ENV)
|
|
if chaos_env is not None:
|
|
if os.path.isdir(chaos_env):
|
|
log.debug(f"chaos_env is a dir: {chaos_env}")
|
|
return glob.glob(chaos_env + "chaos_*.yaml")
|
|
elif os.path.isfile(chaos_env):
|
|
log.debug(f"chaos_env is a file: {chaos_env}")
|
|
return [chaos_env]
|
|
else:
|
|
# not a valid directory, return default
|
|
pass
|
|
log.debug("not a valid directory or file, return default chaos config path")
|
|
return glob.glob(constants.TESTS_CONFIG_LOCATION + constants.ALL_CHAOS_YAMLS)
|
|
|
|
|
|
def reconnect(connections, alias="default", timeout=360):
|
|
"""trying to connect by connection alias"""
|
|
is_connected = False
|
|
start = time.time()
|
|
end = time.time()
|
|
while not is_connected or end - start < timeout:
|
|
try:
|
|
connections.connect(alias)
|
|
is_connected = True
|
|
except Exception as e:
|
|
log.debug(f"fail to connect, error: {str(e)}")
|
|
time.sleep(10)
|
|
end = time.time()
|
|
else:
|
|
log.info(f"failed to reconnect after {timeout} seconds")
|
|
return connections.connect(alias)
|
|
|
|
|
|
def assert_statistic(
|
|
checkers, expectations={}, succ_rate_threshold=0.95, fail_rate_threshold=0.49
|
|
):
|
|
for k in checkers.keys():
|
|
# expect succ if no expectations
|
|
succ_rate = checkers[k].succ_rate()
|
|
total = checkers[k].total()
|
|
average_time = checkers[k].average_time
|
|
if expectations.get(k, "") == constants.FAIL:
|
|
log.info(
|
|
f"Expect Fail: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}"
|
|
)
|
|
pytest.assume(
|
|
succ_rate < fail_rate_threshold or total < 2,
|
|
f"Expect Fail: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}",
|
|
)
|
|
else:
|
|
log.info(
|
|
f"Expect Succ: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}"
|
|
)
|
|
pytest.assume(
|
|
succ_rate > succ_rate_threshold and total > 2,
|
|
f"Expect Succ: {str(k)} succ rate {succ_rate}, total: {total}, average time: {average_time:.4f}",
|
|
)
|