ClickHouse/tests/integration/test_undrop_query/test.py
Yarik Briukhovetskyi 5275a7f8bf
black check
2024-03-26 18:26:32 +01:00

49 lines
1.3 KiB
Python

import pytest
import uuid
import logging
import time
from helpers.cluster import ClickHouseCluster
cluster = ClickHouseCluster(__file__)
node = cluster.add_instance("node", main_configs=["configs/with_delay_config.xml"])
@pytest.fixture(scope="module")
def started_cluster():
try:
cluster.start()
yield cluster
finally:
cluster.shutdown()
def test_undrop_drop_and_undrop_loop(started_cluster):
uuid_list = []
for i in range(4):
table_uuid = uuid.uuid1().__str__()
uuid_list.append(table_uuid)
logging.info(f"table_uuid: {table_uuid}")
node.query(
f"CREATE TABLE test_undrop_{i} UUID '{table_uuid}' (id Int32) ENGINE = MergeTree() ORDER BY id;"
)
node.query(f"DROP TABLE test_undrop_{i};")
for i in range(4):
if (
i >= 3
): # First 3 tables are dropped after 0, 5 and 10 seconds. Fourth is dropped after 21 seconds
time.sleep(6)
error = node.query_and_get_error(
f"UNDROP TABLE test_undrop_loop_{i} UUID '{uuid_list[i]}';"
)
assert "UNKNOWN_TABLE" in error
else:
node.query(f"UNDROP TABLE test_undrop_loop_{i} UUID '{uuid_list[i]}';")
time.sleep(5)