import pytest

from helpers.cluster import ClickHouseCluster
from helpers.test_tools import TSV

cluster = ClickHouseCluster(__file__)
node = cluster.add_instance("node")


@pytest.fixture(scope="module")
def started_cluster():
    try:
        cluster.start()
        yield cluster
    finally:
        cluster.shutdown()


def test_freeze_table(started_cluster):
    node.query(
        """
        CREATE TABLE table_for_freeze
        (
          key UInt64,
          value String
        )
        ENGINE = MergeTree()
        ORDER BY key
        PARTITION BY key % 10;
        """
    )
    node.query(
        """
        INSERT INTO table_for_freeze SELECT number, toString(number) from numbers(10);
        """
    )

    freeze_result = TSV.toMat(
        node.query(
            """
        ALTER TABLE table_for_freeze
        FREEZE WITH NAME 'test_01417'
        FORMAT TSVWithNames
        SETTINGS alter_partition_verbose_result = 1;
        """
        )
    )
    assert 11 == len(freeze_result)
    path_col_ix = freeze_result[0].index("part_backup_path")
    for row in freeze_result[1:]:  # skip header
        part_backup_path = row[path_col_ix]
        node.exec_in_container(["bash", "-c", "test -d {}".format(part_backup_path)])

    freeze_result = TSV.toMat(
        node.query(
            """
        ALTER TABLE table_for_freeze
        FREEZE PARTITION '3'
        WITH NAME
        'test_01417_single_part'
        FORMAT TSVWithNames
        SETTINGS alter_partition_verbose_result = 1;
        """
        )
    )
    assert 2 == len(freeze_result)
    path_col_ix = freeze_result[0].index("part_backup_path")
    for row in freeze_result[1:]:  # skip header
        part_backup_path = row[path_col_ix]
        assert "test_01417_single_part" in part_backup_path
        node.exec_in_container(["bash", "-c", "test -d {}".format(part_backup_path)])