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

72 lines
2.4 KiB
Python

import pytest
from helpers.client import QueryRuntimeException
from helpers.cluster import ClickHouseCluster
from helpers.test_tools import TSV
cluster = ClickHouseCluster(__file__)
node = cluster.add_instance(
"node", main_configs=["configs/zookeeper_config.xml"], with_zookeeper=True
)
@pytest.fixture(scope="module")
def start_cluster():
try:
cluster.start()
yield cluster
finally:
cluster.shutdown()
def test_part_should_reset_mutation(start_cluster):
node.query(
"CREATE TABLE test (i Int64, s String) ENGINE = ReplicatedMergeTree('/clickhouse/tables/test', 'node') ORDER BY i;"
)
node.query("INSERT INTO test SELECT 1, 'a'")
node.query("optimize table test final")
node.query("optimize table test final")
expected = TSV("""all_0_0_2\t1\ta""")
assert TSV(node.query("SELECT _part, * FROM test")) == expected
node.query(
"ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
)
node.query(
"ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
)
node.query(
"ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
)
node.query(
"ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
)
expected = TSV("""all_0_0_2_4\t1\txxx""")
assert TSV(node.query("SELECT _part, * FROM test")) == expected
node.query(
"CREATE TABLE restore (i Int64, s String) ENGINE = ReplicatedMergeTree('/clickhouse/tables/restore', 'node') ORDER BY i;"
)
node.query(
"ALTER TABLE restore FETCH PARTITION tuple() FROM '/clickhouse/tables/test/'"
)
node.query("ALTER TABLE restore ATTACH PART 'all_0_0_2_4'")
node.query("INSERT INTO restore select 2, 'a'")
print(TSV(node.query("SELECT _part, * FROM restore")))
expected = TSV("""all_0_0_0\t1\txxx\nall_1_1_0\t2\ta""")
assert TSV(node.query("SELECT _part, * FROM restore ORDER BY i")) == expected
node.query(
"ALTER TABLE restore UPDATE s='yyy' WHERE 1", settings={"mutations_sync": "2"}
)
expected = TSV("""all_0_0_0_2\t1\tyyy\nall_1_1_0_2\t2\tyyy""")
assert TSV(node.query("SELECT _part, * FROM restore ORDER BY i")) == expected
node.query("ALTER TABLE restore DELETE WHERE 1", settings={"mutations_sync": "2"})
assert node.query("SELECT count() FROM restore").strip() == "0"