mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-15 02:41:59 +00:00
208 lines
12 KiB
Python
208 lines
12 KiB
Python
import contextlib
|
|
import os
|
|
import urllib.request, urllib.parse, urllib.error
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
|
|
class SimpleCluster:
|
|
def close(self):
|
|
self.cluster.shutdown()
|
|
|
|
def __init__(self, cluster, name, config_dir):
|
|
self.cluster = cluster
|
|
self.instance = self.add_instance(name, config_dir)
|
|
cluster.start()
|
|
|
|
def add_instance(self, name, config_dir):
|
|
script_path = os.path.dirname(os.path.realpath(__file__))
|
|
return self.cluster.add_instance(name, main_configs=[os.path.join(script_path, config_dir, 'config.xml')])
|
|
|
|
|
|
def test_dynamic_query_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "dynamic_handler", "test_dynamic_handler")) as cluster:
|
|
test_query = urllib.parse.quote_plus('SELECT * FROM system.settings WHERE name = \'max_threads\'')
|
|
|
|
assert 404 == cluster.instance.http_request('?max_threads=1', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_dynamic_handler_get?max_threads=1', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_dynamic_handler_get?max_threads=1', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 400 == cluster.instance.http_request('test_dynamic_handler_get?max_threads=1', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 200 == cluster.instance.http_request(
|
|
'test_dynamic_handler_get?max_threads=1&get_dynamic_handler_query=' + test_query,
|
|
method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
|
|
def test_predefined_query_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "predefined_handler", "test_predefined_handler")) as cluster:
|
|
assert 404 == cluster.instance.http_request('?max_threads=1', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_predefined_handler_get?max_threads=1', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_predefined_handler_get?max_threads=1', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 500 == cluster.instance.http_request('test_predefined_handler_get?max_threads=1', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert b'max_threads\t1\n' == cluster.instance.http_request(
|
|
'test_predefined_handler_get?max_threads=1&setting_name=max_threads', method='GET',
|
|
headers={'XXX': 'xxx'}).content
|
|
|
|
assert b'max_final_threads\t1\nmax_threads\t1\n' == cluster.instance.http_request(
|
|
'query_param_with_url/max_threads?max_threads=1&max_final_threads=1',
|
|
headers={'XXX': 'max_final_threads'}).content
|
|
|
|
|
|
def test_fixed_static_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "static_handler", "test_static_handler")) as cluster:
|
|
assert 404 == cluster.instance.http_request('', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_fixed_static_handler', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_fixed_static_handler', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 402 == cluster.instance.http_request('test_get_fixed_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
assert 'text/html; charset=UTF-8' == \
|
|
cluster.instance.http_request('test_get_fixed_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).headers['Content-Type']
|
|
assert b'Test get static handler and fix content' == cluster.instance.http_request(
|
|
'test_get_fixed_static_handler', method='GET', headers={'XXX': 'xxx'}).content
|
|
|
|
|
|
def test_config_static_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "static_handler", "test_static_handler")) as cluster:
|
|
assert 404 == cluster.instance.http_request('', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_config_static_handler', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_config_static_handler', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
# check default status code
|
|
assert 200 == cluster.instance.http_request('test_get_config_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
assert 'text/plain; charset=UTF-8' == \
|
|
cluster.instance.http_request('test_get_config_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).headers['Content-Type']
|
|
assert b'Test get static handler and config content' == cluster.instance.http_request(
|
|
'test_get_config_static_handler', method='GET', headers={'XXX': 'xxx'}).content
|
|
|
|
|
|
def test_absolute_path_static_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "static_handler", "test_static_handler")) as cluster:
|
|
cluster.instance.exec_in_container(
|
|
['bash', '-c',
|
|
'echo "<html><body>Absolute Path File</body></html>" > /var/lib/clickhouse/user_files/absolute_path_file.html'],
|
|
privileged=True, user='root')
|
|
|
|
assert 404 == cluster.instance.http_request('', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_absolute_path_static_handler', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_absolute_path_static_handler', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
# check default status code
|
|
assert 200 == cluster.instance.http_request('test_get_absolute_path_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
assert 'text/html; charset=UTF-8' == \
|
|
cluster.instance.http_request('test_get_absolute_path_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).headers['Content-Type']
|
|
assert b'<html><body>Absolute Path File</body></html>\n' == cluster.instance.http_request(
|
|
'test_get_absolute_path_static_handler', method='GET', headers={'XXX': 'xxx'}).content
|
|
|
|
|
|
def test_relative_path_static_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "static_handler", "test_static_handler")) as cluster:
|
|
cluster.instance.exec_in_container(
|
|
['bash', '-c',
|
|
'echo "<html><body>Relative Path File</body></html>" > /var/lib/clickhouse/user_files/relative_path_file.html'],
|
|
privileged=True, user='root')
|
|
|
|
assert 404 == cluster.instance.http_request('', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_relative_path_static_handler', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_get_relative_path_static_handler', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
# check default status code
|
|
assert 200 == cluster.instance.http_request('test_get_relative_path_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
assert 'text/html; charset=UTF-8' == \
|
|
cluster.instance.http_request('test_get_relative_path_static_handler', method='GET',
|
|
headers={'XXX': 'xxx'}).headers['Content-Type']
|
|
assert b'<html><body>Relative Path File</body></html>\n' == cluster.instance.http_request(
|
|
'test_get_relative_path_static_handler', method='GET', headers={'XXX': 'xxx'}).content
|
|
|
|
|
|
def test_defaults_http_handlers():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "defaults_handlers", "test_defaults_handlers")) as cluster:
|
|
assert 200 == cluster.instance.http_request('', method='GET').status_code
|
|
assert b'Default server response' == cluster.instance.http_request('', method='GET').content
|
|
|
|
assert 200 == cluster.instance.http_request('ping', method='GET').status_code
|
|
assert b'Ok.\n' == cluster.instance.http_request('ping', method='GET').content
|
|
|
|
assert 200 == cluster.instance.http_request('replicas_status', method='get').status_code
|
|
assert b'Ok.\n' == cluster.instance.http_request('replicas_status', method='get').content
|
|
|
|
assert 200 == cluster.instance.http_request('replicas_status?verbose=1', method='get').status_code
|
|
assert b'' == cluster.instance.http_request('replicas_status?verbose=1', method='get').content
|
|
|
|
assert 200 == cluster.instance.http_request('?query=SELECT+1', method='GET').status_code
|
|
assert b'1\n' == cluster.instance.http_request('?query=SELECT+1', method='GET').content
|
|
|
|
|
|
def test_prometheus_handler():
|
|
with contextlib.closing(
|
|
SimpleCluster(ClickHouseCluster(__file__), "prometheus_handler", "test_prometheus_handler")) as cluster:
|
|
assert 404 == cluster.instance.http_request('', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_prometheus', method='GET', headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_prometheus', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 200 == cluster.instance.http_request('test_prometheus', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
assert b'ClickHouseProfileEvents_Query' in cluster.instance.http_request('test_prometheus', method='GET',
|
|
headers={'XXX': 'xxx'}).content
|
|
|
|
|
|
def test_replicas_status_handler():
|
|
with contextlib.closing(SimpleCluster(ClickHouseCluster(__file__), "replicas_status_handler",
|
|
"test_replicas_status_handler")) as cluster:
|
|
assert 404 == cluster.instance.http_request('', method='GET', headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_replicas_status', method='GET',
|
|
headers={'XXX': 'bad'}).status_code
|
|
|
|
assert 404 == cluster.instance.http_request('test_replicas_status', method='POST',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
|
|
assert 200 == cluster.instance.http_request('test_replicas_status', method='GET',
|
|
headers={'XXX': 'xxx'}).status_code
|
|
assert b'Ok.\n' == cluster.instance.http_request('test_replicas_status', method='GET',
|
|
headers={'XXX': 'xxx'}).content
|