Create table with the same load_balancing policy in test_insert_distributed_load_balancing

This commit is contained in:
Azat Khuzhin 2020-05-14 03:22:21 +03:00
parent 52d73c7f45
commit dac0b7d1f5

View File

@ -19,8 +19,7 @@ def start_cluster():
finally: finally:
cluster.shutdown() cluster.shutdown()
@pytest.fixture(scope='function', autouse=True) def create_tables(**dist_settings):
def create_tables():
n1.query('DROP TABLE IF EXISTS data') n1.query('DROP TABLE IF EXISTS data')
n2.query('DROP TABLE IF EXISTS data') n2.query('DROP TABLE IF EXISTS data')
n1.query('DROP TABLE IF EXISTS dist') n1.query('DROP TABLE IF EXISTS dist')
@ -35,9 +34,10 @@ def create_tables():
data, data,
rand() rand()
) )
""") """, settings=dist_settings)
def insert_data(**settings): def insert_data(**settings):
create_tables(**settings)
n1.query('INSERT INTO dist SELECT * FROM numbers(10)', settings=settings) n1.query('INSERT INTO dist SELECT * FROM numbers(10)', settings=settings)
n1.query('SYSTEM FLUSH DISTRIBUTED dist') n1.query('SYSTEM FLUSH DISTRIBUTED dist')