ClickHouse/tests/integration/test_create_query_constraints/test.py

93 lines
3.3 KiB
Python
Raw Normal View History

2022-12-07 00:29:48 +00:00
import pytest
import asyncio
import re
import random
import os.path
from helpers.cluster import ClickHouseCluster
from helpers.test_tools import assert_eq_with_retry, TSV
cluster = ClickHouseCluster(__file__)
2022-12-13 08:11:59 +00:00
instance = cluster.add_instance("instance")
2022-12-07 00:29:48 +00:00
@pytest.fixture(scope="module", autouse=True)
def start_cluster():
try:
cluster.start()
yield cluster
finally:
cluster.shutdown()
2022-12-13 08:11:59 +00:00
def test_create_query_const_constraints():
instance.query("CREATE USER u_const SETTINGS max_threads = 1 CONST")
instance.query("GRANT ALL ON *.* TO u_const")
2022-12-07 00:29:48 +00:00
expected_error = "Setting max_threads should not be changed"
assert expected_error in instance.query_and_get_error(
2022-12-13 08:11:59 +00:00
"CREATE USER inner_user SETTINGS max_threads = 1", user="u_const"
2022-12-07 00:29:48 +00:00
)
assert expected_error in instance.query_and_get_error(
2022-12-13 08:11:59 +00:00
"CREATE USER inner_user SETTINGS max_threads MIN 0 MAX 2", user="u_const"
)
assert expected_error in instance.query_and_get_error(
"CREATE USER inner_user SETTINGS max_threads WRITABLE", user="u_const"
)
assert expected_error in instance.query_and_get_error(
"CREATE ROLE inner_role SETTINGS max_threads = 1", user="u_const"
2022-12-07 00:29:48 +00:00
)
assert expected_error in instance.query_and_get_error(
2022-12-13 09:31:58 +00:00
"CREATE SETTINGS PROFILE inner_profile SETTINGS max_threads = 1", user="u_const"
2022-12-07 00:29:48 +00:00
)
2022-12-13 08:11:59 +00:00
instance.query(
2022-12-13 09:31:58 +00:00
"CREATE USER inner_user_1 SETTINGS max_threads CONST", user="u_const"
2022-12-13 08:11:59 +00:00
)
instance.query(
2022-12-13 09:31:58 +00:00
"CREATE USER inner_user_2 SETTINGS max_threads = 1 CONST", user="u_const"
2022-12-13 08:11:59 +00:00
)
instance.query("DROP USER u_const, inner_user_1, inner_user_2")
def test_create_query_minmax_constraints():
instance.query("CREATE USER u_minmax SETTINGS max_threads = 4 MIN 2 MAX 6")
instance.query("GRANT ALL ON *.* TO u_minmax")
expected_error = "Setting max_threads shouldn't be less than"
2022-12-07 00:29:48 +00:00
assert expected_error in instance.query_and_get_error(
2022-12-13 08:11:59 +00:00
"CREATE USER inner_user SETTINGS max_threads = 1", user="u_minmax"
2022-12-07 00:29:48 +00:00
)
2022-12-13 08:11:59 +00:00
assert expected_error in instance.query_and_get_error(
"CREATE USER inner_user SETTINGS max_threads MIN 1 MAX 3", user="u_minmax"
)
assert expected_error in instance.query_and_get_error(
"CREATE ROLE inner_role SETTINGS max_threads MIN 1 MAX 3", user="u_minmax"
)
assert expected_error in instance.query_and_get_error(
"CREATE SETTINGS PROFILE inner_profile SETTINGS max_threads MIN 1 MAX 3",
2022-12-13 09:31:58 +00:00
user="u_minmax",
2022-12-13 08:11:59 +00:00
)
expected_error = "Setting max_threads shouldn't be greater than"
2022-12-07 00:29:48 +00:00
assert expected_error in instance.query_and_get_error(
2022-12-13 08:11:59 +00:00
"CREATE USER inner_user SETTINGS max_threads = 8", user="u_minmax"
)
assert expected_error in instance.query_and_get_error(
"CREATE USER inner_user SETTINGS max_threads MIN 4 MAX 8", user="u_minmax"
)
assert expected_error in instance.query_and_get_error(
"CREATE ROLE inner_role SETTINGS max_threads MIN 4 MAX 8", user="u_minmax"
2022-12-07 00:29:48 +00:00
)
2022-12-13 08:11:59 +00:00
assert expected_error in instance.query_and_get_error(
"CREATE SETTINGS PROFILE inner_profile SETTINGS max_threads MIN 4 MAX 8",
2022-12-13 09:31:58 +00:00
user="u_minmax",
2022-12-13 08:11:59 +00:00
)
instance.query("CREATE USER inner_user SETTINGS max_threads = 3", user="u_minmax")
instance.query("DROP USER u_minmax, inner_user")