mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 00:22:29 +00:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
|
import re
|
||
|
|
import pytest
|
||
|
|
from helpers.cluster import ClickHouseCluster
|
||
|
|
from helpers.test_tools import assert_eq_with_retry
|
||
|
|
|
||
|
|
cluster = ClickHouseCluster(__file__)
|
||
|
|
|
||
|
|
control_node = cluster.add_instance(
|
||
|
|
"control_node",
|
||
|
|
main_configs=["configs/clusters.xml"],
|
||
|
|
with_zookeeper=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
data_node = cluster.add_instance(
|
||
|
|
"data_node",
|
||
|
|
main_configs=["configs/clusters.xml"],
|
||
|
|
with_zookeeper=True,
|
||
|
|
macros={"shard": 1, "replica": 1},
|
||
|
|
)
|
||
|
|
|
||
|
|
uuid_regex = re.compile("[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}")
|
||
|
|
|
||
|
|
|
||
|
|
def assert_create_query(node, database_name, table_name, expected):
|
||
|
|
replace_uuid = lambda x: re.sub(uuid_regex, "uuid", x)
|
||
|
|
query = "SELECT create_table_query FROM system.tables WHERE database='{}' AND table='{}'".format(
|
||
|
|
database_name, table_name
|
||
|
|
)
|
||
|
|
assert_eq_with_retry(node, query, expected, get_result=replace_uuid)
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.fixture(scope="module")
|
||
|
|
def started_cluster():
|
||
|
|
try:
|
||
|
|
cluster.start()
|
||
|
|
yield cluster
|
||
|
|
finally:
|
||
|
|
cluster.shutdown()
|
||
|
|
|
||
|
|
|
||
|
|
def test_ddl(started_cluster):
|
||
|
|
control_node.query("CREATE DATABASE test_db ON CLUSTER 'external' ENGINE=Atomic")
|
||
|
|
control_node.query(
|
||
|
|
"CREATE TABLE test_db.test_table ON CLUSTER 'external' (id Int64) Engine=MergeTree ORDER BY id"
|
||
|
|
)
|
||
|
|
control_node.query(
|
||
|
|
"ALTER TABLE test_db.test_table ON CLUSTER 'external' add column data String"
|
||
|
|
)
|
||
|
|
|
||
|
|
expected = "CREATE TABLE test_db.test_table (`id` Int64, `data` String) ENGINE = MergeTree ORDER BY id SETTINGS index_granularity = 8192"
|
||
|
|
assert_create_query(data_node, "test_db", "test_table", expected)
|
||
|
|
|
||
|
|
control_node.query("DROP TABLE test_db.test_table ON CLUSTER 'external'")
|
||
|
|
control_node.query("DROP DATABASE test_db ON CLUSTER 'external'")
|
||
|
|
|
||
|
|
expected = ""
|
||
|
|
assert_create_query(data_node, "test_db", "test_table", expected)
|
||
|
|
|
||
|
|
|
||
|
|
def test_ddl_replicated(started_cluster):
|
||
|
|
control_node.query(
|
||
|
|
"CREATE DATABASE test_db ON CLUSTER 'external' ENGINE=Replicated('/replicated')",
|
||
|
|
settings={"allow_experimental_database_replicated": 1},
|
||
|
|
)
|
||
|
|
# Exception is expected
|
||
|
|
assert "It's not initial query" in control_node.query_and_get_error(
|
||
|
|
"CREATE TABLE test_db.test_table ON CLUSTER 'external' (id Int64) Engine=MergeTree ORDER BY id"
|
||
|
|
)
|
||
|
|
control_node.query("DROP DATABASE test_db ON CLUSTER 'external'")
|