ClickHouse/tests/integration/test_tmp_policy/test.py

# pylint: disable=unused-argument
# pylint: disable=redefined-outer-name

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()
        yield cluster
    finally:
        cluster.shutdown()


def test_different_versions(start_cluster):
    query = 'SELECT count(ignore(*)) FROM (SELECT * FROM system.numbers LIMIT 1e7) GROUP BY number'
    settings = {
        'max_bytes_before_external_group_by': 1 << 20,
        'max_bytes_before_external_sort': 1 << 20,
    }

    assert node.contains_in_log('Setting up /disk1/ to store temporary data in it')
    assert node.contains_in_log('Setting up /disk2/ to store temporary data in it')

    node.query(query, settings=settings)
    assert node.contains_in_log('Writing part of aggregation data into temporary file /disk1/')
    assert node.contains_in_log('Writing part of aggregation data into temporary file /disk2/')