ClickHouse/tests/integration/test_send_crash_reports/test.py

# 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):
    if started_node.is_built_with_thread_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'