ClickHouse/tests/integration/test_zookeeper_config/test_secure.py

import threading
import os 
from tempfile import NamedTemporaryFile

import pytest
from helpers.cluster import ClickHouseCluster

TEST_DIR = os.path.dirname(__file__)

cluster = ClickHouseCluster(__file__, name="secure",
                            zookeeper_certfile=os.path.join(TEST_DIR, "configs_secure", "client.crt"),
                            zookeeper_keyfile=os.path.join(TEST_DIR, "configs_secure", "client.key"))

node1 = cluster.add_instance('node1', main_configs=["configs_secure/client.crt", "configs_secure/client.key",
                                                    "configs_secure/conf.d/remote_servers.xml",
                                                    "configs_secure/conf.d/ssl_conf.xml",
                                                    "configs/zookeeper_config_with_ssl.xml"], 
                             with_zookeeper_secure=True)
node2 = cluster.add_instance('node2', main_configs=["configs_secure/client.crt", "configs_secure/client.key",
                                                    "configs_secure/conf.d/remote_servers.xml",
                                                    "configs_secure/conf.d/ssl_conf.xml",
                                                    "configs/zookeeper_config_with_ssl.xml"],
                             with_zookeeper_secure=True)

@pytest.fixture(scope="module", autouse=True)
def started_cluster():
    try:
        cluster.start()
        yield cluster

    finally:
        cluster.shutdown()

# NOTE this test have to be ported to Keeper
def test_secure_connection(started_cluster):
    # no asserts, connection works
    node1.query("SELECT count() FROM system.zookeeper WHERE path = '/'")
    node2.query("SELECT count() FROM system.zookeeper WHERE path = '/'")

    threads_number = 16
    iterations = 100
    threads = []

    # just checking for race conditions
    for _ in range(threads_number):
        threads.append(threading.Thread(target=(lambda:
            [node1.query("SELECT count() FROM system.zookeeper WHERE path = '/'") for _ in range(iterations)])))

    for thread in threads:
        thread.start()

    for thread in threads:
        thread.join()