mirror of
https://github.com/hanzoai/datastore.git
synced 2026-09-26 13:55:07 +00:00
118 lines
3.7 KiB
Python
118 lines
3.7 KiB
Python
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
|
|
node = cluster.add_instance("node1", with_zookeeper=True)
|
|
node_with_compatibility = cluster.add_instance(
|
|
"node2", with_zookeeper=True, user_configs=["configs/compatibility.xml"]
|
|
)
|
|
node_with_compatibility_and_mt_setings = cluster.add_instance(
|
|
"node3",
|
|
with_zookeeper=True,
|
|
main_configs=["configs/mt_settings.xml"],
|
|
user_configs=["configs/compatibility.xml"],
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_check_projections_compatibility(started_cluster):
|
|
create_with_invalid_projection = """
|
|
CREATE TABLE tp (type Int32, eventcnt UInt64, PROJECTION p (SELECT sum(eventcnt), type GROUP BY type))
|
|
ENGINE = {} ORDER BY type;
|
|
"""
|
|
|
|
create_without_projection = """
|
|
CREATE TABLE tp (type Int32, eventcnt UInt64)
|
|
ENGINE = {} ORDER BY type;
|
|
"""
|
|
|
|
alter_add_projection = """
|
|
ALTER TABLE tp ADD PROJECTION p (SELECT sum(eventcnt), type GROUP BY type);
|
|
"""
|
|
|
|
# Create with invalid projection is not supported by default
|
|
|
|
assert "Projections are not supported" in node.query_and_get_error(
|
|
create_with_invalid_projection.format("ReplacingMergeTree")
|
|
)
|
|
assert "Projections are not supported" in node.query_and_get_error(
|
|
create_with_invalid_projection.format(
|
|
"ReplicatedReplacingMergeTree('/tables/tp', '0')"
|
|
)
|
|
)
|
|
|
|
# Adding invalid projection is not supported by default
|
|
|
|
node.query(create_without_projection.format("ReplacingMergeTree"))
|
|
assert "ADD PROJECTION is not supported" in node.query_and_get_error(
|
|
alter_add_projection
|
|
)
|
|
node.query("DROP TABLE tp SYNC")
|
|
|
|
node.query(
|
|
create_without_projection.format("ReplicatedReplacingMergeTree('/tables/tp', '0')")
|
|
)
|
|
assert "ADD PROJECTION is not supported" in node.query_and_get_error(
|
|
alter_add_projection
|
|
)
|
|
node.query("DROP TABLE tp SYNC")
|
|
|
|
# Create with invalid projection is supported with compatibility
|
|
|
|
node_with_compatibility.query(
|
|
create_with_invalid_projection.format("ReplacingMergeTree")
|
|
)
|
|
node_with_compatibility.query("DROP TABLE tp SYNC")
|
|
node_with_compatibility.query(
|
|
create_with_invalid_projection.format(
|
|
"ReplicatedReplacingMergeTree('/tables/tp2', '0')"
|
|
)
|
|
)
|
|
node_with_compatibility.query("DROP TABLE tp SYNC")
|
|
|
|
# Adding invalid projection is supported with compatibility
|
|
|
|
node_with_compatibility.query(create_without_projection.format("ReplacingMergeTree"))
|
|
node_with_compatibility.query(alter_add_projection)
|
|
node_with_compatibility.query("DROP TABLE tp SYNC")
|
|
|
|
node_with_compatibility.query(
|
|
create_without_projection.format("ReplicatedReplacingMergeTree('/tables/tp3', '0')")
|
|
)
|
|
node_with_compatibility.query(alter_add_projection)
|
|
node_with_compatibility.query("DROP TABLE tp SYNC")
|
|
|
|
|
|
def test_config_overrides_compatibility(started_cluster):
|
|
create_with_invalid_projection = """
|
|
CREATE TABLE tp (type Int32, eventcnt UInt64, PROJECTION p (select sum(eventcnt), type group by type))
|
|
engine = {} order by type;
|
|
"""
|
|
|
|
assert (
|
|
"Projections are not supported"
|
|
in node_with_compatibility_and_mt_setings.query_and_get_error(
|
|
create_with_invalid_projection.format("ReplacingMergeTree")
|
|
)
|
|
)
|
|
assert (
|
|
"Projections are not supported"
|
|
in node_with_compatibility_and_mt_setings.query_and_get_error(
|
|
create_with_invalid_projection.format(
|
|
"ReplicatedReplacingMergeTree('/tables/tp', '0')"
|
|
)
|
|
)
|
|
)
|