import time

import pytest
from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__)
node1 = cluster.add_instance(
    "node1",
    main_configs=["configs/remote_servers.xml"],
    with_zookeeper=True,
    stay_alive=True,
)


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

        node1.query(
            """
            CREATE DATABASE test;
            CREATE TABLE test_table(date Date, id UInt32)
            ENGINE = ReplicatedMergeTree('/clickhouse/tables/test/replicated', 'node1') ORDER BY id PARTITION BY toYYYYMM(date);
            """
        )

        yield cluster

    except Exception as ex:
        print(ex)

    finally:
        cluster.shutdown()


def drop_zk(zk):
    zk.delete(path="/clickhouse", recursive=True)


def test_startup_without_zookeeper(start_cluster):
    node1.query(
        "INSERT INTO test_table VALUES ('2018-10-01', 1), ('2018-10-02', 2), ('2018-10-03', 3)"
    )
    assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
    assert (
        node1.query("SELECT is_readonly from system.replicas where table='test_table'")
        == "0\n"
    )

    cluster.run_kazoo_commands_with_retries(drop_zk)

    time.sleep(5)
    assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
    with pytest.raises(Exception):
        node1.query(
            "INSERT INTO test_table VALUES ('2018-10-01', 1), ('2018-10-02', 2), ('2018-10-03', 3)"
        )

    node1.restart_clickhouse()

    assert node1.query("SELECT COUNT(*) from test_table") == "3\n"
    assert (
        node1.query("SELECT is_readonly from system.replicas where table='test_table'")
        == "1\n"
    )