import pytest

from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__)

node1 = cluster.add_instance(
    "node1",
    main_configs=["configs/remote_servers.xml"],
    user_configs=["configs/users_config.xml"],
    with_zookeeper=True,
)
node2 = cluster.add_instance(
    "node2",
    main_configs=["configs/remote_servers.xml"],
    user_configs=["configs/users_config.xml"],
    with_zookeeper=True,
)
node3 = cluster.add_instance(
    "node3",
    main_configs=["configs/remote_servers.xml"],
    user_configs=["configs/users_config.xml"],
    with_zookeeper=True,
)
node4 = cluster.add_instance(
    "node4",
    main_configs=["configs/remote_servers.xml"],
    user_configs=["configs/users_config.xml"],
    with_zookeeper=True,
)


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

        yield cluster
    finally:
        cluster.shutdown()


def test_long_query(started_cluster):
    node1.query(
        "CREATE TABLE cluster_table (key UInt64, value String) ENGINE = ReplicatedMergeTree('/test/1/cluster_table', '1') ORDER BY tuple()"
    )
    node2.query(
        "CREATE TABLE cluster_table (key UInt64, value String) ENGINE = ReplicatedMergeTree('/test/1/cluster_table', '2') ORDER BY tuple()"
    )

    node1.query(
        "INSERT INTO cluster_table SELECT number, toString(number) FROM numbers(20)"
    )
    node2.query("SYSTEM SYNC REPLICA cluster_table")

    node3.query(
        "CREATE TABLE cluster_table (key UInt64, value String) ENGINE = ReplicatedMergeTree('/test/2/cluster_table', '1') ORDER BY tuple()"
    )

    node4.query(
        "CREATE TABLE cluster_table (key UInt64, value String) ENGINE = ReplicatedMergeTree('/test/2/cluster_table', '2') ORDER BY tuple()"
    )
    node3.query(
        "INSERT INTO cluster_table SELECT number, toString(number) FROM numbers(20)"
    )
    node4.query("SYSTEM SYNC REPLICA cluster_table")

    node1.query(
        "ALTER TABLE cluster_table ON CLUSTER 'test_cluster' UPDATE key = 1 WHERE sleepEachRow(1) == 0",
        settings={"mutations_sync": "2"},
    )

    assert node1.query("SELECT SUM(key) FROM cluster_table") == "20\n"
    assert node2.query("SELECT SUM(key) FROM cluster_table") == "20\n"
    assert node3.query("SELECT SUM(key) FROM cluster_table") == "20\n"
    assert node4.query("SELECT SUM(key) FROM cluster_table") == "20\n"