mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
60 lines
1.7 KiB
Python
60 lines
1.7 KiB
Python
from collections.abc import Iterable
|
|
|
|
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster, ClickHouseInstance
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster() -> Iterable[ClickHouseCluster]:
|
|
try:
|
|
cluster.add_instance(
|
|
"node1",
|
|
main_configs=["configs/storage_conf.xml"],
|
|
with_minio=True,
|
|
with_zookeeper=True,
|
|
)
|
|
cluster.start()
|
|
|
|
yield cluster
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
@pytest.mark.parametrize("storage_policy", ["s3", "encrypted"])
|
|
def test_unfreeze(storage_policy: str, started_cluster: ClickHouseCluster) -> None:
|
|
node1: ClickHouseInstance = started_cluster.instances["node1"]
|
|
node1.query(
|
|
f"""\
|
|
CREATE TABLE test1 (a Int)
|
|
ENGINE = ReplicatedMergeTree('/clickhouse-tables/test1', 'r1')
|
|
ORDER BY a
|
|
SETTINGS storage_policy = '{storage_policy}'
|
|
"""
|
|
)
|
|
|
|
node1.query(
|
|
"""\
|
|
INSERT INTO test1
|
|
SELECT *
|
|
FROM system.numbers
|
|
LIMIT 20
|
|
"""
|
|
)
|
|
|
|
node1.query("ALTER TABLE test1 FREEZE WITH NAME 'test'")
|
|
node1.query("SYSTEM UNFREEZE WITH NAME 'test'")
|
|
uuid = node1.query("SELECT uuid FROM system.tables WHERE name = 'test1'").strip()
|
|
# ensure that zero copy lock parent still exists
|
|
kazoo = started_cluster.get_kazoo_client("zoo1")
|
|
part_path = f"/clickhouse/zero_copy/zero_copy_s3/{uuid}/all_0_0_0/"
|
|
children: list[str] = kazoo.get_children(part_path)
|
|
assert len(children) == 1
|
|
part_name = children[0]
|
|
assert len(kazoo.get_children(part_path + part_name)) == 1
|
|
assert node1.query("SELECT count() FROM test1").strip() == "20"
|
|
node1.query("DROP TABLE test1")
|
|
node1.query("SYSTEM DROP REPLICA 'r1' FROM ZKPATH '/clickhouse-tables/test1'")
|