mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-11-17 13:13:36 +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.
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
import time
|
|
|
|
import pytest
|
|
from helpers.cluster import ClickHouseCluster
|
|
|
|
cluster = ClickHouseCluster(__file__)
|
|
node1 = cluster.add_instance('node1', main_configs=['configs/remote_servers.xml'], with_zookeeper=True, stay_alive=True)
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def start_cluster():
|
|
try:
|
|
cluster.start()
|
|
|
|
node1.query(
|
|
'''
|
|
CREATE DATABASE test;
|
|
CREATE TABLE test_table(date Date, id UInt32)
|
|
ENGINE = ReplicatedMergeTree('/clickhouse/tables/test/replicated', 'node1') ORDER BY id PARTITION BY toYYYYMM(date);
|
|
'''
|
|
)
|
|
|
|
yield cluster
|
|
|
|
except Exception as ex:
|
|
print ex
|
|
|
|
finally:
|
|
cluster.shutdown()
|
|
|
|
|
|
def drop_zk(zk):
|
|
zk.delete(path="/clickhouse", recursive=True)
|
|
|
|
|
|
def test_startup_without_zookeeper(start_cluster):
|
|
node1.query("INSERT INTO test_table VALUES ('2018-10-01', 1), ('2018-10-02', 2), ('2018-10-03', 3)")
|
|
assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
|
|
assert node1.query("SELECT is_readonly from system.replicas where table='test_table'") == "0\n"
|
|
|
|
cluster.run_kazoo_commands_with_retries(drop_zk)
|
|
|
|
time.sleep(5)
|
|
assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
|
|
with pytest.raises(Exception):
|
|
node1.query("INSERT INTO test_table VALUES ('2018-10-01', 1), ('2018-10-02', 2), ('2018-10-03', 3)")
|
|
|
|
node1.restart_clickhouse()
|
|
|
|
assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
|
|
assert node1.query("SELECT is_readonly from system.replicas where table='test_table'") == "1\n"
|