ClickHouse/tests/integration/test_send_crash_reports/test.py
2024-09-27 10:19:49 +00:00

75 lines
2.0 KiB
Python

# pylint: disable=redefined-outer-name
# pylint: disable=unused-argument
# pylint: disable=line-too-long
# pylint: disable=bare-except
import os
import time
import pytest
import helpers.cluster
import helpers.test_tools
from . import fake_sentry_server
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
@pytest.fixture(scope="module")
def started_node():
cluster = helpers.cluster.ClickHouseCluster(__file__)
try:
node = cluster.add_instance(
"node",
main_configs=[
os.path.join(SCRIPT_DIR, "configs", "config_send_crash_reports.xml")
],
)
cluster.start()
yield node
finally:
# It will print Fatal message after pkill -SEGV, suppress it
try:
cluster.shutdown()
except:
pass
def test_send_segfault(started_node):
# NOTE: another option is to increase waiting time.
if (
started_node.is_built_with_thread_sanitizer()
or started_node.is_built_with_address_sanitizer()
or started_node.is_built_with_memory_sanitizer()
):
pytest.skip("doesn't fit in timeouts for stacktrace generation")
started_node.copy_file_to_container(
os.path.join(SCRIPT_DIR, "fake_sentry_server.py"), "/fake_sentry_server.py"
)
started_node.exec_in_container(
["bash", "-c", "python3 /fake_sentry_server.py > /fake_sentry_server.log 2>&1"],
detach=True,
user="root",
)
time.sleep(1)
started_node.exec_in_container(
["bash", "-c", "pkill -SEGV clickhouse"], user="root"
)
result = None
for attempt in range(1, 6):
time.sleep(attempt)
result = started_node.exec_in_container(
["cat", fake_sentry_server.RESULT_PATH], user="root"
)
if result == "OK":
break
if result == "INITIAL_STATE":
continue
if result:
assert False, "Unexpected state: " + result
assert result == "OK", "Crash report not sent"