mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 10:52:30 +00:00
64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
instance = cluster.add_instance("instance", stay_alive=True)
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def test_persistence():
|
|
create_function_query1 = "CREATE FUNCTION MySum1 AS (a, b) -> a + b"
|
|
create_function_query2 = "CREATE FUNCTION MySum2 AS (a, b) -> MySum1(a, b) + b"
|
|
create_function_query3 = "CREATE FUNCTION MyUnion AS () -> (SELECT sum(s) FROM (SELECT 1 as s UNION ALL SELECT 1 as s))"
|
|
|
|
instance.query(create_function_query1)
|
|
instance.query(create_function_query2)
|
|
instance.query(create_function_query3)
|
|
|
|
assert instance.query("SELECT MySum1(1,2)") == "3\n"
|
|
assert instance.query("SELECT MySum2(1,2)") == "5\n"
|
|
assert instance.query("SELECT MyUnion()") == "2\n"
|
|
|
|
instance.restart_clickhouse()
|
|
|
|
assert instance.query("SELECT MySum1(1,2)") == "3\n"
|
|
assert instance.query("SELECT MySum2(1,2)") == "5\n"
|
|
assert instance.query("SELECT MyUnion()") == "2\n"
|
|
|
|
instance.query("DROP FUNCTION MySum2")
|
|
instance.query("DROP FUNCTION MySum1")
|
|
instance.query("DROP FUNCTION MyUnion")
|
|
|
|
instance.restart_clickhouse()
|
|
|
|
error_message = instance.query_and_get_error("SELECT MySum1(1, 2)")
|
|
assert (
|
|
"Unknown function MySum1" in error_message
|
|
or "Function with name `MySum1` does not exist. In scope SELECT MySum1(1, 2)"
|
|
in error_message
|
|
)
|
|
|
|
error_message = instance.query_and_get_error("SELECT MySum2(1, 2)")
|
|
assert (
|
|
"Unknown function MySum2" in error_message
|
|
or "Function with name `MySum2` does not exist. In scope SELECT MySum2(1, 2)"
|
|
in error_message
|
|
)
|
|
|
|
error_message = instance.query_and_get_error("SELECT MyUnion()")
|
|
assert (
|
|
"Unknown function MyUnion" in error_message
|
|
or "Function with name `MyUnion` does not exist. In scope SELECT MyUnion"
|
|
in error_message
|
|
)
|