mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-09-20 08:40:50 +00:00
88 lines
3.4 KiB
Python
88 lines
3.4 KiB
Python
import pytest
|
|
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
uuids = []
|
|
|
|
@pytest.fixture(scope="module")
|
|
def cluster():
|
|
try:
|
|
cluster = ClickHouseCluster(__file__)
|
|
cluster.add_instance("node1", main_configs=["configs/storage_conf.xml"], with_nginx=True)
|
|
cluster.add_instance("node2", main_configs=["configs/storage_conf_web.xml"], with_nginx=True)
|
|
cluster.add_instance("node3", main_configs=["configs/storage_conf_web.xml"], with_nginx=True)
|
|
cluster.start()
|
|
|
|
node1 = cluster.instances["node1"]
|
|
expected = ""
|
|
global uuids
|
|
for i in range(3):
|
|
node1.query(""" CREATE TABLE data{} (id Int32) ENGINE = MergeTree() ORDER BY id SETTINGS storage_policy = 'def';""".format(i))
|
|
node1.query("INSERT INTO data{} SELECT number FROM numbers(500000 * {})".format(i, i + 1))
|
|
expected = node1.query("SELECT * FROM data{} ORDER BY id".format(i))
|
|
|
|
metadata_path = node1.query("SELECT data_paths FROM system.tables WHERE name='data{}'".format(i))
|
|
metadata_path = metadata_path[metadata_path.find('/'):metadata_path.rfind('/')+1]
|
|
print(f'Metadata: {metadata_path}')
|
|
|
|
node1.exec_in_container(['bash', '-c',
|
|
'/usr/bin/clickhouse static-files-disk-uploader --test-mode --url http://nginx:80/test1 --metadata-path {}'.format(metadata_path)], user='root')
|
|
parts = metadata_path.split('/')
|
|
uuids.append(parts[3])
|
|
print(f'UUID: {parts[3]}')
|
|
|
|
yield cluster
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
@pytest.mark.parametrize("node_name", ["node2"])
|
|
def test_usage(cluster, node_name):
|
|
node1 = cluster.instances["node1"]
|
|
node2 = cluster.instances[node_name]
|
|
global uuids
|
|
assert(len(uuids) == 3)
|
|
for i in range(3):
|
|
node2.query("""
|
|
ATTACH TABLE test{} UUID '{}'
|
|
(id Int32) ENGINE = MergeTree() ORDER BY id
|
|
SETTINGS storage_policy = 'web';
|
|
""".format(i, uuids[i], i, i))
|
|
|
|
result = node2.query("SELECT * FROM test{} settings max_threads=20".format(i))
|
|
|
|
result = node2.query("SELECT count() FROM test{}".format(i))
|
|
assert(int(result) == 500000 * (i+1))
|
|
|
|
result = node2.query("SELECT id FROM test{} WHERE id % 56 = 3 ORDER BY id".format(i))
|
|
assert(result == node1.query("SELECT id FROM data{} WHERE id % 56 = 3 ORDER BY id".format(i)))
|
|
|
|
result = node2.query("SELECT id FROM test{} WHERE id > 789999 AND id < 999999 ORDER BY id".format(i))
|
|
assert(result == node1.query("SELECT id FROM data{} WHERE id > 789999 AND id < 999999 ORDER BY id".format(i)))
|
|
|
|
node2.query("DROP TABLE test{}".format(i))
|
|
print(f"Ok {i}")
|
|
|
|
|
|
def test_incorrect_usage(cluster):
|
|
node1 = cluster.instances["node1"]
|
|
node2 = cluster.instances["node3"]
|
|
global uuids
|
|
node2.query("""
|
|
ATTACH TABLE test0 UUID '{}'
|
|
(id Int32) ENGINE = MergeTree() ORDER BY id
|
|
SETTINGS storage_policy = 'web';
|
|
""".format(uuids[0]))
|
|
|
|
result = node2.query("SELECT count() FROM test0")
|
|
assert(int(result) == 500000)
|
|
|
|
result = node2.query_and_get_error("ALTER TABLE test0 ADD COLUMN col1 Int32 first")
|
|
assert("Table is read-only" in result)
|
|
|
|
result = node2.query_and_get_error("TRUNCATE TABLE test0")
|
|
assert("Table is read-only" in result)
|
|
|
|
node2.query("DROP TABLE test0")
|