ClickHouse/tests/integration/test_replication_without_zookeeper/test.py
2022-03-22 17:39:58 +01:00

67 lines
1.7 KiB
Python

import time
import pytest
from helpers.cluster import ClickHouseCluster
cluster = ClickHouseCluster(__file__)
node1 = cluster.add_instance(
"node1",
main_configs=["configs/remote_servers.xml"],
with_zookeeper=True,
stay_alive=True,
)
@pytest.fixture(scope="module")
def start_cluster():
try:
cluster.start()
node1.query(
"""
CREATE DATABASE test;
CREATE TABLE test_table(date Date, id UInt32)
ENGINE = ReplicatedMergeTree('/clickhouse/tables/test/replicated', 'node1') ORDER BY id PARTITION BY toYYYYMM(date);
"""
)
yield cluster
except Exception as ex:
print(ex)
finally:
cluster.shutdown()
def drop_zk(zk):
zk.delete(path="/clickhouse", recursive=True)
def test_startup_without_zookeeper(start_cluster):
node1.query(
"INSERT INTO test_table VALUES ('2018-10-01', 1), ('2018-10-02', 2), ('2018-10-03', 3)"
)
assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
assert (
node1.query("SELECT is_readonly from system.replicas where table='test_table'")
== "0\n"
)
cluster.run_kazoo_commands_with_retries(drop_zk)
time.sleep(5)
assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
with pytest.raises(Exception):
node1.query(
"INSERT INTO test_table VALUES ('2018-10-01', 1), ('2018-10-02', 2), ('2018-10-03', 3)"
)
node1.restart_clickhouse()
assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
assert (
node1.query("SELECT is_readonly from system.replicas where table='test_table'")
== "1\n"
)