import pytest
from helpers.client import QueryRuntimeException
from helpers.cluster import ClickHouseCluster
from helpers.test_tools import TSV

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_part_should_reset_mutation(start_cluster):
    node.query(
        "CREATE TABLE test (i Int64, s String) ENGINE = ReplicatedMergeTree('/clickhouse/tables/test', 'node') ORDER BY i;"
    )
    node.query("INSERT INTO test SELECT 1, 'a'")
    node.query("optimize table test final")
    node.query("optimize table test final")

    expected = TSV("""all_0_0_2\t1\ta""")
    assert TSV(node.query("SELECT _part, * FROM test")) == expected

    node.query(
        "ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
    )
    node.query(
        "ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
    )
    node.query(
        "ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
    )
    node.query(
        "ALTER TABLE test UPDATE s='xxx' WHERE 1", settings={"mutations_sync": "2"}
    )

    expected = TSV("""all_0_0_2_4\t1\txxx""")
    assert TSV(node.query("SELECT _part, * FROM test")) == expected

    node.query(
        "CREATE TABLE restore (i Int64, s String) ENGINE = ReplicatedMergeTree('/clickhouse/tables/restore', 'node') ORDER BY i;"
    )
    node.query(
        "ALTER TABLE restore FETCH PARTITION tuple() FROM '/clickhouse/tables/test/'"
    )
    node.query("ALTER TABLE restore ATTACH PART 'all_0_0_2_4'")
    node.query("INSERT INTO restore select 2, 'a'")

    print(TSV(node.query("SELECT _part, * FROM restore")))
    expected = TSV("""all_0_0_0\t1\txxx\nall_1_1_0\t2\ta""")
    assert TSV(node.query("SELECT _part, * FROM restore ORDER BY i")) == expected

    node.query(
        "ALTER TABLE restore UPDATE s='yyy' WHERE 1", settings={"mutations_sync": "2"}
    )

    expected = TSV("""all_0_0_0_2\t1\tyyy\nall_1_1_0_2\t2\tyyy""")
    assert TSV(node.query("SELECT _part, * FROM restore ORDER BY i")) == expected

    node.query("ALTER TABLE restore DELETE WHERE 1", settings={"mutations_sync": "2"})

    assert node.query("SELECT count() FROM restore").strip() == "0"