mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-16 04:32:33 +00:00
62 lines
2.4 KiB
Python
62 lines
2.4 KiB
Python
|
import pytest
|
||
|
|
|
||
|
|
from helpers.cluster import ClickHouseCluster
|
||
|
|
from multiprocessing.dummy import Pool
|
||
|
|
from helpers.network import PartitionManager
|
||
|
|
import time
|
||
|
|
|
||
|
|
cluster = ClickHouseCluster(__file__)
|
||
|
|
|
||
|
|
node1 = cluster.add_instance('node1', with_zookeeper=True)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="module")
|
||
|
|
def started_cluster():
|
||
|
|
try:
|
||
|
|
cluster.start()
|
||
|
|
|
||
|
|
node1.query('''
|
||
|
|
CREATE TABLE replicated_mt(date Date, id UInt32, value Int32)
|
||
|
|
ENGINE = ReplicatedMergeTree('/clickhouse/tables/replicated_mt', '{replica}') ORDER BY id;
|
||
|
|
'''.format(replica=node1.name))
|
||
|
|
|
||
|
|
yield cluster
|
||
|
|
|
||
|
|
finally:
|
||
|
|
cluster.shutdown()
|
||
|
|
|
||
|
|
def corrupt_data_part_on_disk(node, table, part_name):
|
||
|
|
part_path = node.query(
|
||
|
|
"SELECT path FROM system.parts WHERE table = '{}' and name = '{}'".format(table, part_name)).strip()
|
||
|
|
node.exec_in_container(['bash', '-c',
|
||
|
|
'cd {p} && ls *.bin | head -n 1 | xargs -I{{}} sh -c \'echo "1" >> $1\' -- {{}}'.format(
|
||
|
|
p=part_path)], privileged=True)
|
||
|
|
|
||
|
|
|
||
|
|
def test_merge_and_part_corruption(started_cluster):
|
||
|
|
node1.query("SYSTEM STOP REPLICATION QUEUES replicated_mt")
|
||
|
|
for i in range(4):
|
||
|
|
node1.query("INSERT INTO replicated_mt SELECT toDate('2019-10-01'), number, number * number FROM numbers ({f}, 100000)".format(f=i*100000))
|
||
|
|
|
||
|
|
assert node1.query("SELECT COUNT() FROM system.parts WHERE table='replicated_mt' AND active=1") == "4\n"
|
||
|
|
|
||
|
|
# Need to corrupt "border part" (left or right). If we will corrupt something in the middle
|
||
|
|
# clickhouse will not consider merge as broken, because we have parts with the same min and max
|
||
|
|
# block numbers.
|
||
|
|
corrupt_data_part_on_disk(node1, 'replicated_mt', 'all_3_3_0')
|
||
|
|
|
||
|
|
with Pool(1) as p:
|
||
|
|
def optimize_with_delay(x):
|
||
|
|
node1.query("OPTIMIZE TABLE replicated_mt FINAL", timeout=30)
|
||
|
|
|
||
|
|
# corrupt part after merge already assigned, but not started
|
||
|
|
res_opt = p.apply_async(optimize_with_delay, (1,))
|
||
|
|
node1.query("CHECK TABLE replicated_mt", settings={"check_query_single_value_result": 0})
|
||
|
|
# start merge
|
||
|
|
node1.query("SYSTEM START REPLICATION QUEUES replicated_mt")
|
||
|
|
res_opt.get()
|
||
|
|
|
||
|
|
# will hung if checked bug not fixed
|
||
|
|
node1.query("ALTER TABLE replicated_mt UPDATE value = 7 WHERE 1", settings={"mutations_sync": 2}, timeout=30)
|
||
|
|
assert node1.query("SELECT sum(value) FROM replicated_mt") == "2100000\n"
|