mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-14 11:33:46 +00:00
b75963d370
This PR formats all the `*.py` files found under the `tests/integration` folder. It also reorders the imports and cleans up a bunch of unused imports. The formatting also takes care of other things like wrapping lines and fixing spaces and indents such that the tests look more readable.
81 lines
2.4 KiB
Python
81 lines
2.4 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',
|
|
main_configs=["configs/config.d/storage_configuration.xml"],
|
|
tmpfs=['/disk1:size=100M', '/disk2:size=100M'])
|
|
|
|
|
|
@pytest.fixture(scope='module')
|
|
def start_cluster():
|
|
try:
|
|
cluster.start()
|
|
node.query('CREATE DATABASE test ENGINE=Ordinary')
|
|
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/test/{table}/default@127%2E0%2E0%2E2:9000 -maxdepth 1 -type f 2>/dev/null | wc -l'.format(
|
|
root=root, table=table)
|
|
]).split('\n')[0])
|
|
|
|
|
|
def test_insert(start_cluster):
|
|
node.query('CREATE TABLE test.foo (key Int) Engine=Memory()')
|
|
node.query("""
|
|
CREATE TABLE test.dist_foo (key Int)
|
|
Engine=Distributed(
|
|
test_cluster_two_shards,
|
|
test,
|
|
foo,
|
|
key%2,
|
|
'default'
|
|
)
|
|
""")
|
|
# manual only (but only for remote node)
|
|
node.query('SYSTEM STOP DISTRIBUTED SENDS test.dist_foo')
|
|
|
|
node.query('INSERT INTO test.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 test.dist_foo') == '100\n'
|
|
node.query('SYSTEM FLUSH DISTRIBUTED test.dist_foo')
|
|
assert node.query('SELECT count() FROM test.dist_foo') == '200\n'
|
|
|
|
#
|
|
# RENAME
|
|
#
|
|
node.query('RENAME TABLE test.dist_foo TO test.dist2_foo')
|
|
|
|
node.query('INSERT INTO test.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 test.dist2_foo') == '300\n'
|
|
node.query('SYSTEM FLUSH DISTRIBUTED test.dist2_foo')
|
|
assert node.query('SELECT count() FROM test.dist2_foo') == '400\n'
|
|
|
|
#
|
|
# DROP
|
|
#
|
|
node.query('DROP TABLE test.dist2_foo')
|
|
for disk in ['disk1', 'disk2']:
|
|
node.exec_in_container([
|
|
'bash', '-c',
|
|
'test ! -e /{}/data/test/dist2_foo'.format(disk)
|
|
])
|