Files
wehub-resource-sync 498b235461
Build and test / Build and test AMD64 Ubuntu 22.04 (push) Failing after 0s
Publish Builder / amazonlinux2023 (push) Failing after 1s
Build and test / UT for Go (push) Has been skipped
Publish KRTE Images / KRTE (push) Failing after 1s
Build and test / Integration Test (push) Has been skipped
Build and test / Upload Code Coverage (push) Has been skipped
Publish Builder / rockylinux9 (push) Failing after 1s
Publish Builder / ubuntu22.04 (push) Failing after 0s
Publish Builder / ubuntu24.04 (push) Failing after 0s
Publish Gpu Builder / publish-gpu-builder (push) Failing after 1s
Publish Test Images / PyTest (push) Failing after 0s
Build and test / UT for Cpp (push) Has been cancelled
chore: import upstream snapshot with attribution
2026-07-13 12:31:17 +08:00

130 lines
3.9 KiB
Python

import pytest
def pytest_addoption(parser):
parser.addoption("--chaos_type", action="store", default="pod_kill", help="chaos_type")
parser.addoption("--role_type", action="store", default="activated", help="role_type")
parser.addoption("--target_component", action="store", default="querynode", help="target_component")
parser.addoption("--target_pod", action="store", default="etcd_leader", help="target_pod")
parser.addoption("--target_scope", action="store", default="all", help="target_scope")
parser.addoption("--target_number", action="store", default="1", help="target_number")
parser.addoption("--chaos_duration", action="store", default="7m", help="chaos_duration")
parser.addoption("--chaos_interval", action="store", default="2m", help="chaos_interval")
parser.addoption("--wait_signal", action="store", type=bool, default=True, help="wait_signal")
parser.addoption("--enable_import", action="store", type=bool, default=False, help="enable_import")
parser.addoption(
"--target_rgs",
action="store",
default="",
help="comma-separated resource group names to target for chaos (e.g. rg1,rg2)",
)
parser.addoption(
"--chaos_mode",
action="store",
default="one",
help="chaos mode: 'one' (random single pod) or 'all' (all matching pods)",
)
parser.addoption(
"--target_components",
action="store",
default="querynode,streamingnode",
help="comma-separated components to inject chaos (e.g. querynode,streamingnode)",
)
parser.addoption(
"--chaos_template",
action="store",
default="",
help="path to external ChaosMesh YAML template (overrides built-in config)",
)
parser.addoption("--collection_num", action="store", default="1", help="collection_num")
parser.addoption("--search_timeout", action="store", type=float, default=None, help="search API timeout in seconds")
parser.addoption("--query_timeout", action="store", type=float, default=None, help="query API timeout in seconds")
@pytest.fixture(scope="session", autouse=True)
def configure_client_timeouts(request):
import chaos.checker as checker
search_timeout = request.config.getoption("--search_timeout")
query_timeout = request.config.getoption("--query_timeout")
if search_timeout is not None:
checker.search_timeout = search_timeout
if query_timeout is not None:
checker.query_timeout = query_timeout
@pytest.fixture
def chaos_type(request):
return request.config.getoption("--chaos_type")
@pytest.fixture
def role_type(request):
return request.config.getoption("--role_type")
@pytest.fixture
def target_component(request):
return request.config.getoption("--target_component")
@pytest.fixture
def target_pod(request):
return request.config.getoption("--target_pod")
@pytest.fixture
def target_scope(request):
return request.config.getoption("--target_scope")
@pytest.fixture
def target_number(request):
return request.config.getoption("--target_number")
@pytest.fixture
def collection_num(request):
return request.config.getoption("--collection_num")
@pytest.fixture
def chaos_duration(request):
return request.config.getoption("--chaos_duration")
@pytest.fixture
def chaos_interval(request):
return request.config.getoption("--chaos_interval")
@pytest.fixture
def wait_signal(request):
return request.config.getoption("--wait_signal")
@pytest.fixture
def enable_import(request):
return request.config.getoption("--enable_import")
@pytest.fixture
def target_rgs(request):
return request.config.getoption("--target_rgs")
@pytest.fixture
def chaos_mode(request):
return request.config.getoption("--chaos_mode")
@pytest.fixture
def target_components(request):
return request.config.getoption("--target_components")
@pytest.fixture
def chaos_template(request):
return request.config.getoption("--chaos_template")