ClickHouse/tests/integration/test_zookeeper_config/test_secure.py

54 lines
2.1 KiB
Python
Raw Normal View History

2021-05-21 13:29:43 +00:00
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):
2021-08-24 08:01:08 +00:00
# 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 = []
2021-05-21 13:29:43 +00:00
2021-08-24 08:01:08 +00:00
# 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)])))
2021-05-21 13:29:43 +00:00
2021-08-24 08:01:08 +00:00
for thread in threads:
thread.start()
2021-05-21 13:29:43 +00:00
2021-08-24 08:01:08 +00:00
for thread in threads:
thread.join()