ClickHouse/tests/integration/test_inherit_multiple_profiles/test.py

import pytest

from helpers.client import QueryRuntimeException
from helpers.cluster import ClickHouseCluster
from helpers.test_tools import TSV

cluster = ClickHouseCluster(__file__)
instance = cluster.add_instance(
    "instance", user_configs=["configs/combined_profile.xml"]
)
q = instance.query


@pytest.fixture(scope="module")
def started_cluster():
    try:
        cluster.start()

        yield cluster

    finally:
        cluster.shutdown()


def test_combined_profile(started_cluster):
    settings = q(
        """
SELECT name, value FROM system.settings
    WHERE name IN
      ('max_insert_block_size', 'max_network_bytes', 'max_query_size',
       'max_parallel_replicas', 'readonly')
    AND changed
ORDER BY name
""",
        user="test_combined_profile",
    )

    expected1 = """\
max_insert_block_size	654321
max_network_bytes	1234567890
max_parallel_replicas	2
max_query_size	400000000
readonly	2"""

    assert TSV(settings) == TSV(expected1)

    with pytest.raises(QueryRuntimeException) as exc:
        q(
            """
          SET max_insert_block_size = 1000;
          """,
            user="test_combined_profile",
        )

    assert "max_insert_block_size shouldn't be less than 654320." in str(exc.value)

    with pytest.raises(QueryRuntimeException) as exc:
        q(
            """
          SET max_network_bytes = 2000000000;
          """,
            user="test_combined_profile",
        )

    assert "max_network_bytes shouldn't be greater than 1234567891." in str(exc.value)

    with pytest.raises(QueryRuntimeException) as exc:
        q(
            """
          SET max_parallel_replicas = 1000;
          """,
            user="test_combined_profile",
        )

    assert "max_parallel_replicas should not be changed." in str(exc.value)

    with pytest.raises(QueryRuntimeException) as exc:
        q(
            """
          SET max_memory_usage = 1000;
          """,
            user="test_combined_profile",
        )

    assert "max_memory_usage shouldn't be less than 300000000." in str(exc.value)