mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-04 21:42:39 +00:00
81 lines
2.0 KiB
Python
81 lines
2.0 KiB
Python
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()
|