mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-05 05:52:05 +00:00
1789 lines
43 KiB
Python
1789 lines
43 KiB
Python
import os
|
|
import re
|
|
import time
|
|
|
|
import pytest
|
|
from helpers.cluster import ClickHouseCluster
|
|
from helpers.test_tools import assert_eq_with_retry, TSV
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
instance = cluster.add_instance(
|
|
"instance",
|
|
main_configs=["configs/remote_servers.xml"],
|
|
user_configs=[
|
|
"configs/users.d/assign_myquota_to_default_user.xml",
|
|
"configs/users.d/drop_default_quota.xml",
|
|
"configs/users.d/myquota.xml",
|
|
"configs/users.d/user_with_no_quota.xml",
|
|
],
|
|
)
|
|
|
|
|
|
def check_system_quotas(canonical):
|
|
canonical_tsv = TSV(canonical)
|
|
r = TSV(instance.query("SELECT * FROM system.quotas ORDER BY name"))
|
|
print(("system_quotas: {},\ncanonical: {}".format(r, TSV(canonical_tsv))))
|
|
assert r == canonical_tsv
|
|
|
|
|
|
def system_quota_limits(canonical):
|
|
canonical_tsv = TSV(canonical)
|
|
r = TSV(
|
|
instance.query(
|
|
"SELECT * FROM system.quota_limits ORDER BY quota_name, duration"
|
|
)
|
|
)
|
|
print(("system_quota_limits: {},\ncanonical: {}".format(r, TSV(canonical_tsv))))
|
|
assert r == canonical_tsv
|
|
|
|
|
|
def system_quota_usage(canonical):
|
|
canonical_tsv = TSV(canonical)
|
|
query = (
|
|
"SELECT quota_name, quota_key, duration, queries, max_queries, query_selects, max_query_selects, query_inserts, max_query_inserts, errors, max_errors, result_rows, max_result_rows,"
|
|
"result_bytes, max_result_bytes, read_rows, max_read_rows, read_bytes, max_read_bytes, max_execution_time, max_failed_sequential_authentications "
|
|
"FROM system.quota_usage ORDER BY duration"
|
|
)
|
|
r = TSV(instance.query(query))
|
|
print(("system_quota_usage: {},\ncanonical: {}".format(r, TSV(canonical_tsv))))
|
|
assert r == canonical_tsv
|
|
|
|
|
|
def system_quotas_usage(canonical):
|
|
canonical_tsv = TSV(canonical)
|
|
query = (
|
|
"SELECT quota_name, quota_key, is_current, duration, queries, max_queries, query_selects, max_query_selects, query_inserts, max_query_inserts, errors, max_errors, result_rows, max_result_rows, "
|
|
"result_bytes, max_result_bytes, read_rows, max_read_rows, read_bytes, max_read_bytes, max_execution_time, max_failed_sequential_authentications "
|
|
"FROM system.quotas_usage ORDER BY quota_name, quota_key, duration"
|
|
)
|
|
r = TSV(instance.query(query))
|
|
print(("system_quotas_usage: {},\ncanonical: {}".format(r, TSV(canonical_tsv))))
|
|
assert r == canonical_tsv
|
|
|
|
|
|
def copy_quota_xml(local_file_name, reload_immediately=True):
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
instance.copy_file_to_container(
|
|
os.path.join(script_dir, local_file_name),
|
|
"/etc/clickhouse-server/users.d/myquota.xml",
|
|
)
|
|
if reload_immediately:
|
|
# We use the special user 'user_with_no_quota' here because
|
|
# we don't want SYSTEM RELOAD CONFIG to mess our quota consuming checks.
|
|
instance.query("SYSTEM RELOAD CONFIG", user="user_with_no_quota")
|
|
|
|
|
|
@pytest.fixture(scope="module", autouse=True)
|
|
def started_cluster():
|
|
try:
|
|
cluster.start()
|
|
instance.query("DROP TABLE IF EXISTS test_table")
|
|
instance.query(
|
|
"CREATE TABLE test_table(x UInt32) ENGINE = MergeTree ORDER BY tuple()"
|
|
)
|
|
instance.query("INSERT INTO test_table SELECT number FROM numbers(50)")
|
|
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def reset_quotas_and_usage_info():
|
|
try:
|
|
instance.query("DROP QUOTA IF EXISTS qA, qB")
|
|
copy_quota_xml("simpliest.xml") # To reset usage info.
|
|
copy_quota_xml("normal_limits.xml")
|
|
yield
|
|
finally:
|
|
pass
|
|
|
|
|
|
def test_quota_from_users_xml():
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
[31556952],
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quotas_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
1,
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
1,
|
|
1000,
|
|
1,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
"\\N",
|
|
50,
|
|
1000,
|
|
200,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT SUM(x) from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
2,
|
|
1000,
|
|
2,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
51,
|
|
"\\N",
|
|
208,
|
|
"\\N",
|
|
100,
|
|
1000,
|
|
400,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
|
|
def test_simpliest_quota():
|
|
# Simpliest quota doesn't even track usage.
|
|
copy_quota_xml("simpliest.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits("")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
|
|
def test_tracking_quota():
|
|
# Now we're tracking usage.
|
|
copy_quota_xml("tracking.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
1,
|
|
"\\N",
|
|
1,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT SUM(x) from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
2,
|
|
"\\N",
|
|
2,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
51,
|
|
"\\N",
|
|
208,
|
|
"\\N",
|
|
100,
|
|
"\\N",
|
|
400,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
|
|
def test_exceed_quota():
|
|
# Change quota, now the limits are tiny so we will exceed the quota.
|
|
copy_quota_xml("tiny_limits.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[["myQuota", 31556952, 0, 1, 1, 1, 1, 1, "\\N", 1, "\\N", "\\N", "\\N", "1"]]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
1,
|
|
0,
|
|
1,
|
|
0,
|
|
1,
|
|
0,
|
|
1,
|
|
0,
|
|
1,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"1",
|
|
]
|
|
]
|
|
)
|
|
|
|
assert re.search(
|
|
"Quota.*has been exceeded",
|
|
instance.query_and_get_error("SELECT * from test_table"),
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
1,
|
|
1,
|
|
1,
|
|
1,
|
|
0,
|
|
1,
|
|
1,
|
|
1,
|
|
0,
|
|
1,
|
|
0,
|
|
"\\N",
|
|
50,
|
|
1,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"1",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Change quota, now the limits are enough to execute queries.
|
|
copy_quota_xml("normal_limits.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
1,
|
|
1000,
|
|
1,
|
|
500,
|
|
0,
|
|
500,
|
|
1,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
50,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
2,
|
|
1000,
|
|
2,
|
|
500,
|
|
0,
|
|
500,
|
|
1,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
"\\N",
|
|
100,
|
|
1000,
|
|
200,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
|
|
def test_add_remove_interval():
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
[31556952],
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Add interval.
|
|
copy_quota_xml("two_intervals.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952,63113904]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
],
|
|
[
|
|
"myQuota",
|
|
63113904,
|
|
1,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
30000,
|
|
"\\N",
|
|
20000,
|
|
120,
|
|
"\\N",
|
|
"\\N",
|
|
],
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
],
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
63113904,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
30000,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
20000,
|
|
120,
|
|
"\\N",
|
|
],
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
1,
|
|
1000,
|
|
1,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
"\\N",
|
|
50,
|
|
1000,
|
|
200,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
],
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
63113904,
|
|
1,
|
|
"\\N",
|
|
1,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
30000,
|
|
50,
|
|
"\\N",
|
|
200,
|
|
20000,
|
|
120,
|
|
"\\N",
|
|
],
|
|
]
|
|
)
|
|
|
|
# Remove interval.
|
|
copy_quota_xml("normal_limits.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
[31556952],
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
1,
|
|
1000,
|
|
1,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
50,
|
|
"\\N",
|
|
200,
|
|
"\\N",
|
|
50,
|
|
1000,
|
|
200,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
2,
|
|
1000,
|
|
2,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
100,
|
|
"\\N",
|
|
400,
|
|
"\\N",
|
|
100,
|
|
1000,
|
|
400,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Remove all intervals.
|
|
copy_quota_xml("simpliest.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits("")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Add one interval back.
|
|
copy_quota_xml("normal_limits.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
[31556952],
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
|
|
def test_add_remove_quota():
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
[31556952],
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quotas_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
1,
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Add quota.
|
|
copy_quota_xml("two_quotas.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
],
|
|
[
|
|
"myQuota2",
|
|
"4590510c-4d13-bf21-ec8a-c2187b092e73",
|
|
"users_xml",
|
|
"['client_key','user_name']",
|
|
"[3600,2629746]",
|
|
0,
|
|
"[]",
|
|
"[]",
|
|
],
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
],
|
|
[
|
|
"myQuota2",
|
|
3600,
|
|
1,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
4000,
|
|
400000,
|
|
4000,
|
|
400000,
|
|
60,
|
|
"\\N",
|
|
"3",
|
|
],
|
|
[
|
|
"myQuota2",
|
|
2629746,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1800,
|
|
"\\N",
|
|
"\\N",
|
|
],
|
|
]
|
|
)
|
|
system_quotas_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
1,
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Drop quota.
|
|
copy_quota_xml("normal_limits.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quotas_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
1,
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
# Drop all quotas.
|
|
copy_quota_xml("no_quotas.xml")
|
|
check_system_quotas("")
|
|
system_quota_limits("")
|
|
system_quotas_usage("")
|
|
|
|
# Add one quota back.
|
|
copy_quota_xml("normal_limits.xml")
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quotas_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
1,
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
|
|
def test_reload_users_xml_by_timer():
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
time.sleep(1) # The modification time of the 'quota.xml' file should be different,
|
|
# because config files are reload by timer only when the modification time is changed.
|
|
copy_quota_xml("tiny_limits.xml", reload_immediately=False)
|
|
assert_eq_with_retry(
|
|
instance,
|
|
"SELECT * FROM system.quotas",
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
["user_name"],
|
|
"[31556952]",
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
],
|
|
)
|
|
assert_eq_with_retry(
|
|
instance,
|
|
"SELECT * FROM system.quota_limits",
|
|
[["myQuota", 31556952, 0, 1, 1, 1, 1, 1, "\\N", 1, "\\N", "\\N", "\\N", "1"]],
|
|
)
|
|
|
|
|
|
def test_dcl_introspection():
|
|
assert instance.query("SHOW QUOTAS") == "myQuota\n"
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA")
|
|
== "CREATE QUOTA myQuota KEYED BY user_name FOR INTERVAL 1 year MAX queries = 1000, query_selects = 500, query_inserts = 500, read_rows = 1000 TO default\n"
|
|
)
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTAS")
|
|
== "CREATE QUOTA myQuota KEYED BY user_name FOR INTERVAL 1 year MAX queries = 1000, query_selects = 500, query_inserts = 500, read_rows = 1000 TO default\n"
|
|
)
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t0\\t1000\\t0\\t500\\t0\\t500\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t1000\\t0\\t\\\\N\\t.*\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t500\\t0\\t500\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t1000\\t200\\t\\\\N\\t.*\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
expected_access = "CREATE QUOTA myQuota KEYED BY user_name FOR INTERVAL 1 year MAX queries = 1000, query_selects = 500, query_inserts = 500, read_rows = 1000 TO default\n"
|
|
assert expected_access in instance.query("SHOW ACCESS")
|
|
|
|
# Add interval.
|
|
copy_quota_xml("two_intervals.xml")
|
|
assert instance.query("SHOW QUOTAS") == "myQuota\n"
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA")
|
|
== "CREATE QUOTA myQuota KEYED BY user_name FOR INTERVAL 1 year MAX queries = 1000, read_rows = 1000, FOR RANDOMIZED INTERVAL 2 year MAX result_bytes = 30000, read_bytes = 20000, execution_time = 120 TO default\n"
|
|
)
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t1000\\t200\\t\\\\N\\t.*\\t\\\\N\n"
|
|
"myQuota\\tdefault\\t.*\\t63113904\\t0\\t\\\\N\t0\\t\\\\N\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t30000\\t0\\t\\\\N\\t0\\t20000\\t.*\\t120",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
# Drop interval, add quota.
|
|
copy_quota_xml("two_quotas.xml")
|
|
assert instance.query("SHOW QUOTAS") == "myQuota\nmyQuota2\n"
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA myQuota")
|
|
== "CREATE QUOTA myQuota KEYED BY user_name FOR INTERVAL 1 year MAX queries = 1000, read_rows = 1000 TO default\n"
|
|
)
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA myQuota2")
|
|
== "CREATE QUOTA myQuota2 KEYED BY client_key, user_name FOR RANDOMIZED INTERVAL 1 hour MAX result_rows = 4000, result_bytes = 400000, read_rows = 4000, read_bytes = 400000, execution_time = 60, failed_sequential_authentications = 3, FOR INTERVAL 1 month MAX execution_time = 1800\n"
|
|
)
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTAS")
|
|
== "CREATE QUOTA myQuota KEYED BY user_name FOR INTERVAL 1 year MAX queries = 1000, read_rows = 1000 TO default\n"
|
|
"CREATE QUOTA myQuota2 KEYED BY client_key, user_name FOR RANDOMIZED INTERVAL 1 hour MAX result_rows = 4000, result_bytes = 400000, read_rows = 4000, read_bytes = 400000, execution_time = 60, failed_sequential_authentications = 3, FOR INTERVAL 1 month MAX execution_time = 1800\n"
|
|
)
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t1000\\t200\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
# Drop all quotas.
|
|
copy_quota_xml("no_quotas.xml")
|
|
assert instance.query("SHOW QUOTAS") == ""
|
|
assert instance.query("SHOW CREATE QUOTA") == ""
|
|
assert instance.query("SHOW QUOTA") == ""
|
|
|
|
|
|
def test_dcl_management():
|
|
copy_quota_xml("no_quotas.xml")
|
|
assert instance.query("SHOW QUOTA") == ""
|
|
|
|
instance.query(
|
|
"CREATE QUOTA qA FOR INTERVAL 15 MONTH MAX QUERIES 123 TO CURRENT_USER"
|
|
)
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA qA")
|
|
== "CREATE QUOTA qA FOR INTERVAL 5 quarter MAX queries = 123 TO default\n"
|
|
)
|
|
assert re.match(
|
|
"qA\\t\\t.*\\t39446190\\t0\\t123\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
assert re.match(
|
|
"qA\\t\\t.*\\t39446190\\t1\\t123\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query(
|
|
"ALTER QUOTA qA FOR INTERVAL 15 MONTH MAX QUERIES 321, MAX ERRORS 10, FOR INTERVAL 0.5 HOUR MAX EXECUTION TIME 0.5"
|
|
)
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA qA")
|
|
== "CREATE QUOTA qA FOR INTERVAL 30 minute MAX execution_time = 0.5, FOR INTERVAL 5 quarter MAX queries = 321, errors = 10 TO default\n"
|
|
)
|
|
assert re.match(
|
|
"qA\\t\\t.*\\t1800\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t.*\\t0.5\\t0\\t\\\\N\\t0\\t\\\\N\n"
|
|
"qA\\t\\t.*\\t39446190\\t1\\t321\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t10\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
assert re.match(
|
|
"qA\\t\\t.*\\t1800\\t1\\t\\\\N\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t.*\\t0.5\\t0\\t\\\\N\\t0\\t\\\\N\n"
|
|
"qA\\t\\t.*\\t39446190\\t2\\t321\\t2\\t\\\\N\\t0\\t\\\\N\\t0\\t10\\t100\\t\\\\N\\t400\\t\\\\N\\t100\\t\\\\N\\t400\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query(
|
|
"ALTER QUOTA qA FOR INTERVAL 15 MONTH NO LIMITS, FOR RANDOMIZED INTERVAL 16 MONTH TRACKING ONLY, FOR INTERVAL 1800 SECOND NO LIMITS"
|
|
)
|
|
assert re.match(
|
|
"qA\\t\\t.*\\t42075936\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t.*\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
assert re.match(
|
|
"qA\\t\\t.*\\t42075936\\t1\\t\\\\N\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("ALTER QUOTA qA RENAME TO qB")
|
|
assert (
|
|
instance.query("SHOW CREATE QUOTA qB")
|
|
== "CREATE QUOTA qB FOR RANDOMIZED INTERVAL 16 month TRACKING ONLY TO default\n"
|
|
)
|
|
assert re.match(
|
|
"qB\\t\\t.*\\t42075936\\t1\\t\\\\N\\t1\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t50\\t\\\\N\\t200\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("SELECT * from test_table")
|
|
assert re.match(
|
|
"qB\\t\\t.*\\t42075936\\t2\\t\\\\N\\t2\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\\t100\\t\\\\N\\t400\\t\\\\N\\t100\\t\\\\N\\t400\\t\\\\N\\t.*\\t\\\\N\\t0\\t\\\\N\\t0\\t\\\\N\n",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
instance.query("DROP QUOTA qB")
|
|
assert instance.query("SHOW QUOTA") == ""
|
|
|
|
|
|
def test_users_xml_is_readonly():
|
|
assert re.search(
|
|
"storage is readonly", instance.query_and_get_error("DROP QUOTA myQuota")
|
|
)
|
|
|
|
|
|
def test_query_inserts():
|
|
check_system_quotas(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"e651da9c-a748-8703-061a-7e5e5096dae7",
|
|
"users_xml",
|
|
"['user_name']",
|
|
[31556952],
|
|
0,
|
|
"['default']",
|
|
"[]",
|
|
]
|
|
]
|
|
)
|
|
system_quota_limits(
|
|
[
|
|
[
|
|
"myQuota",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
500,
|
|
500,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
1000,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
system_quotas_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
1,
|
|
31556952,
|
|
0,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("DROP TABLE IF EXISTS test_table_ins")
|
|
instance.query(
|
|
"CREATE TABLE test_table_ins(x UInt32) ENGINE = MergeTree ORDER BY tuple()"
|
|
)
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
2,
|
|
1000,
|
|
0,
|
|
500,
|
|
0,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
|
|
instance.query("INSERT INTO test_table_ins values(1)")
|
|
system_quota_usage(
|
|
[
|
|
[
|
|
"myQuota",
|
|
"default",
|
|
31556952,
|
|
3,
|
|
1000,
|
|
0,
|
|
500,
|
|
1,
|
|
500,
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
"\\N",
|
|
0,
|
|
1000,
|
|
0,
|
|
"\\N",
|
|
"\\N",
|
|
"\\N",
|
|
]
|
|
]
|
|
)
|
|
instance.query("DROP TABLE test_table_ins")
|
|
|
|
|
|
def test_consumption_of_show_tables():
|
|
assert instance.query("SHOW TABLES") == "test_table\n"
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t500\\t0\\t500\\t0\\t\\\\N\\t1\\t\\\\N.*",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
|
|
def test_consumption_of_show_databases():
|
|
assert (
|
|
instance.query("SHOW DATABASES")
|
|
== "INFORMATION_SCHEMA\ndefault\ninformation_schema\nsystem\n"
|
|
)
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t500\\t0\\t500\\t0\\t\\\\N\\t4\\t\\\\N.*",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
|
|
def test_consumption_of_show_clusters():
|
|
assert len(instance.query("SHOW CLUSTERS")) > 0
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t500\\t0\\t500\\t0\\t\\\\N.*",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
|
|
def test_consumption_of_show_processlist():
|
|
instance.query("SHOW PROCESSLIST")
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t500\\t0\\t500\\t0\\t\\\\N\\t0\\t\\\\N.*",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|
|
|
|
|
|
def test_consumption_of_show_privileges():
|
|
assert len(instance.query("SHOW PRIVILEGES")) > 0
|
|
assert re.match(
|
|
"myQuota\\tdefault\\t.*\\t31556952\\t1\\t1000\\t1\\t500\\t0\\t500\\t0\\t\\\\N.*",
|
|
instance.query("SHOW QUOTA"),
|
|
)
|