mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-30 03:22:14 +00:00
b75963d370
This PR formats all the `*.py` files found under the `tests/integration` folder. It also reorders the imports and cleans up a bunch of unused imports. The formatting also takes care of other things like wrapping lines and fixing spaces and indents such that the tests look more readable.
60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
from helpers.network import PartitionManager
|
|
from helpers.test_tools import assert_eq_with_retry
|
|
|
|
|
|
def fill_nodes(nodes, shard):
|
|
for node in nodes:
|
|
node.query(
|
|
'''
|
|
CREATE DATABASE test;
|
|
CREATE TABLE test_table(date Date, id UInt32)
|
|
ENGINE = ReplicatedMergeTree('/clickhouse/tables/test{shard}/replicated', '{replica}')
|
|
ORDER BY id PARTITION BY toYYYYMM(date)
|
|
SETTINGS min_replicated_logs_to_keep=3, max_replicated_logs_to_keep=5, cleanup_delay_period=0, cleanup_delay_period_random_add=0;
|
|
'''.format(shard=shard, replica=node.name))
|
|
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
node1 = cluster.add_instance('node1', main_configs=['configs/remote_servers.xml'], with_zookeeper=True)
|
|
node2 = cluster.add_instance('node2', main_configs=['configs/remote_servers.xml'], with_zookeeper=True)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def start_cluster():
|
|
try:
|
|
cluster.start()
|
|
fill_nodes([node1, node2], 1)
|
|
yield cluster
|
|
except Exception as ex:
|
|
print ex
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_inconsistent_parts_if_drop_while_replica_not_active(start_cluster):
|
|
with PartitionManager() as pm:
|
|
# insert into all replicas
|
|
for i in range(50):
|
|
node1.query("INSERT INTO test_table VALUES ('2019-08-16', {})".format(i))
|
|
assert_eq_with_retry(node2, "SELECT count(*) FROM test_table", node1.query("SELECT count(*) FROM test_table"))
|
|
|
|
# disable network on the first replica
|
|
pm.partition_instances(node1, node2)
|
|
pm.drop_instance_zk_connections(node1)
|
|
|
|
# drop all parts on the second replica
|
|
node2.query_with_retry("ALTER TABLE test_table DROP PARTITION 201908")
|
|
assert_eq_with_retry(node2, "SELECT count(*) FROM test_table", "0")
|
|
|
|
# insert into the second replica
|
|
# DROP_RANGE will be removed from the replication log and the first replica will be lost
|
|
for i in range(50):
|
|
node2.query("INSERT INTO test_table VALUES ('2019-08-16', {})".format(50 + i))
|
|
|
|
# the first replica will be cloned from the second
|
|
pm.heal_all()
|
|
assert_eq_with_retry(node1, "SELECT count(*) FROM test_table", node2.query("SELECT count(*) FROM test_table"))
|