mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-24 00:22:29 +00:00
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
import pytest
|
|
from helpers.cluster import ClickHouseCluster, get_docker_compose_path, run_and_check
|
|
from time import sleep
|
|
import os
|
|
|
|
DOCKER_COMPOSE_PATH = get_docker_compose_path()
|
|
SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
|
|
ch_server = cluster.add_instance(
|
|
"clickhouse-server",
|
|
with_coredns=True,
|
|
main_configs=[
|
|
"configs/config.xml",
|
|
"configs/reverse_dns_function.xml",
|
|
"configs/listen_host.xml",
|
|
],
|
|
)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def started_cluster():
|
|
global cluster
|
|
try:
|
|
cluster.start()
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def setup_ch_server(dns_server_ip):
|
|
ch_server.exec_in_container(
|
|
(["bash", "-c", f"echo 'nameserver {dns_server_ip}' > /etc/resolv.conf"])
|
|
)
|
|
ch_server.exec_in_container(
|
|
(["bash", "-c", "echo 'options ndots:0' >> /etc/resolv.conf"])
|
|
)
|
|
ch_server.query("SYSTEM DROP DNS CACHE")
|
|
|
|
|
|
def test_reverse_dns_query(started_cluster):
|
|
dns_server_ip = cluster.get_instance_ip(cluster.coredns_host)
|
|
|
|
setup_ch_server(dns_server_ip)
|
|
|
|
for _ in range(0, 200):
|
|
response = ch_server.query("select reverseDNSQuery('2001:4860:4860::8888')")
|
|
assert response == "['dns.google']\n"
|