mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
62 lines
1.9 KiB
Python
62 lines
1.9 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
|
|
import pytest
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
from helpers.network import PartitionManager
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
node = cluster.add_instance('node', with_zookeeper=True, main_configs=['configs/remote_servers.xml'])
|
|
node_1 = cluster.add_instance('node_1', with_zookeeper=True)
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
node_1.query_with_retry('DROP TABLE IF EXISTS replicated')
|
|
|
|
node_1.query_with_retry('''CREATE TABLE replicated (id UInt32, date Date) ENGINE =
|
|
ReplicatedMergeTree('/clickhouse/tables/replicated', 'node_1') ORDER BY id PARTITION BY toYYYYMM(date)''')
|
|
|
|
node.query_with_retry("CREATE TABLE distributed (id UInt32, date Date) ENGINE = Distributed('test_cluster', 'default', 'replicated')")
|
|
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
|
|
def test(started_cluster):
|
|
cluster.pause_container("node_1")
|
|
|
|
node.query("SYSTEM RELOAD CONFIG")
|
|
node.query_and_get_error("SELECT count() FROM distributed SETTINGS receive_timeout=1")
|
|
|
|
result = node.query("SELECT errors_count, estimated_recovery_time FROM system.clusters WHERE cluster='test_cluster' and host_name='node_1'")
|
|
errors_count, recovery_time = map(int, result.split())
|
|
assert errors_count == 3
|
|
|
|
while True:
|
|
time.sleep(1)
|
|
|
|
result = node.query("SELECT errors_count, estimated_recovery_time FROM system.clusters WHERE cluster='test_cluster' and host_name='node_1'")
|
|
prev_time = recovery_time
|
|
errors_count, recovery_time = map(int, result.split())
|
|
|
|
if recovery_time == 0:
|
|
break
|
|
|
|
assert recovery_time < prev_time
|
|
assert errors_count > 0
|
|
|
|
assert recovery_time == 0
|
|
assert errors_count == 0
|
|
|
|
cluster.unpause_container("node_1")
|
|
|