ClickHouse/tests/integration/test_user_defined_object_persistence/test.py

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
    )