from __future__ import print_function

import pytest

import helpers
from helpers.client import QueryRuntimeException
from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__)
node = cluster.add_instance(
    "node", main_configs=["configs/zookeeper_config.xml"], with_zookeeper=True
)


@pytest.fixture(scope="module")
def start_cluster():
    try:
        cluster.start()

        yield cluster
    finally:
        cluster.shutdown()


def test_fetch_partition_with_outdated_parts(start_cluster):
    node.query(
        "CREATE TABLE simple (date Date, id UInt32) ENGINE = ReplicatedMergeTree('/clickhouse/tables/0/simple', 'node') ORDER BY tuple() PARTITION BY date;"
    )
    node.query("INSERT INTO simple VALUES ('2020-08-27', 1)")

    node.query(
        "CREATE TABLE simple2 (date Date, id UInt32) ENGINE = ReplicatedMergeTree('/clickhouse/tables/1/simple', 'node') ORDER BY tuple() PARTITION BY date;"
    )
    node.query("INSERT INTO simple2 VALUES ('2020-08-27', 2)")
    node.query("INSERT INTO simple2 VALUES ('2020-08-27', 3)")
    node.query("OPTIMIZE TABLE simple2 FINAL")

    # until now both tables will have the same part

    node.query(
        "ALTER TABLE simple2 FETCH PARTITION '2020-08-27' FROM 'zookeeper2:/clickhouse/tables/0/simple';"
    )

    node.query("ALTER TABLE simple2 ATTACH PARTITION '2020-08-27';")

    assert node.query("SELECT id FROM simple2 order by id").strip() == "1\n2\n3"