ClickHouse/dbms/tests/integration/test_distributed_storage_configuration/test.py

66 lines
2.0 KiB
Python

# pylint: disable=unused-argument
# pylint: disable=redefined-outer-name
# pylint: disable=line-too-long
import pytest
from helpers.cluster import ClickHouseCluster
cluster = ClickHouseCluster(__file__)
node = cluster.add_instance('node',
config_dir='configs',
tmpfs=['/disk1:size=100M', '/disk2:size=100M'])
@pytest.fixture(scope='module')
def start_cluster():
try:
cluster.start()
yield cluster
finally:
cluster.shutdown()
def _files_in_dist_mon(node, root, table):
return int(node.exec_in_container([
'bash',
'-c',
# `-maxdepth 1` to avoid /tmp/ subdirectory
'find /{root}/data/default/{table}/default@127%2E0%2E0%2E2:9000 -maxdepth 1 -type f | wc -l'.format(root=root, table=table)
]).split('\n')[0])
def test_different_versions(start_cluster):
node.query('CREATE TABLE foo (key Int) Engine=Memory()')
node.query("""
CREATE TABLE dist_foo (key Int)
Engine=Distributed(
test_cluster_two_shards,
currentDatabase(),
foo,
key%2,
'default'
)
""")
# manual only
node.query('SYSTEM STOP DISTRIBUTED SENDS dist_foo')
node.query('INSERT INTO dist_foo SELECT * FROM numbers(100)')
assert _files_in_dist_mon(node, 'disk1', 'dist_foo') == 1
assert _files_in_dist_mon(node, 'disk2', 'dist_foo') == 0
assert node.query('SELECT count() FROM dist_foo') == '100\n'
node.query('SYSTEM FLUSH DISTRIBUTED dist_foo')
assert node.query('SELECT count() FROM dist_foo') == '200\n'
#
# RENAME
#
node.query('RENAME TABLE dist_foo TO dist2_foo')
node.query('INSERT INTO dist2_foo SELECT * FROM numbers(100)')
assert _files_in_dist_mon(node, 'disk1', 'dist2_foo') == 0
assert _files_in_dist_mon(node, 'disk2', 'dist2_foo') == 1
assert node.query('SELECT count() FROM dist2_foo') == '300\n'
node.query('SYSTEM FLUSH DISTRIBUTED dist2_foo')
assert node.query('SELECT count() FROM dist2_foo') == '400\n'