ClickHouse/tests/integration/test_dictionaries_wait_for_load/test.py
2024-09-27 10:19:49 +00:00

49 lines
1.2 KiB
Python

import pytest
from helpers.cluster import ClickHouseCluster
from helpers.test_tools import TSV
DICTIONARY_FILES = [
"dictionaries/long_loading_dictionary.xml",
]
cluster = ClickHouseCluster(__file__)
node1 = cluster.add_instance(
"node1",
main_configs=["configs/no_dictionaries_lazy_load.xml"],
dictionaries=DICTIONARY_FILES,
)
node0 = cluster.add_instance(
"node0",
dictionaries=DICTIONARY_FILES,
)
@pytest.fixture(scope="module", autouse=True)
def started_cluster():
try:
cluster.start()
yield cluster
finally:
cluster.shutdown()
def get_status(instance, dictionary_name):
return instance.query(
"SELECT status FROM system.dictionaries WHERE name='" + dictionary_name + "'"
).rstrip("\n")
def test_wait_for_dictionaries_load():
assert get_status(node1, "long_loading_dictionary") == "LOADED"
assert node1.query("SELECT * FROM dictionary(long_loading_dictionary)") == TSV(
[[1, "aa"], [2, "bb"]]
)
assert get_status(node0, "long_loading_dictionary") == "NOT_LOADED"
assert node0.query("SELECT * FROM dictionary(long_loading_dictionary)") == TSV(
[[1, "aa"], [2, "bb"]]
)
assert get_status(node0, "long_loading_dictionary") == "LOADED"