import pytest

from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__, name="log_quries_probability")
node1 = cluster.add_instance("node1", with_zookeeper=False)
node2 = cluster.add_instance("node2", with_zookeeper=False)


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

    finally:
        cluster.shutdown()


def test_log_quries_probability_one(start_cluster):
    for i in range(100):
        node1.query("SELECT 12345", settings={"log_queries_probability": 0.5})

    node1.query("SYSTEM FLUSH LOGS")

    assert (
        node1.query(
            "SELECT count() < (2 * 100) FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'"
        )
        == "1\n"
    )
    assert (
        node1.query(
            "SELECT count() > 0         FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'"
        )
        == "1\n"
    )
    assert (
        node1.query(
            "SELECT count() % 2         FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'"
        )
        == "0\n"
    )

    node1.query("TRUNCATE TABLE system.query_log")


def test_log_quries_probability_two(start_cluster):
    for i in range(100):
        node1.query(
            "SELECT 12345 FROM remote('node2', system, one)",
            settings={"log_queries_probability": 0.5},
        )

        node1.query("SYSTEM FLUSH LOGS")
        node2.query("SYSTEM FLUSH LOGS")

        ans1 = node1.query(
            "SELECT count() FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'"
        )
        ans2 = node2.query(
            "SELECT count() FROM system.query_log WHERE query LIKE '%12345%' AND query NOT LIKE '%system.query_log%'"
        )

        assert ans1 == ans2

        node1.query("TRUNCATE TABLE system.query_log")
        node2.query("TRUNCATE TABLE system.query_log")