import pytest
from helpers.cluster import ClickHouseCluster

cluster = ClickHouseCluster(__file__)

node = cluster.add_instance("node", main_configs=[])


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


def check_read_rows(query_id, read_rows):
    node.query("SYSTEM FLUSH LOGS")
    real_read_rows = node.query(
        f"""SELECT read_rows
            FROM system.query_log
            WHERE
                type = 'QueryFinish' and
                query_id = '{query_id}'
            ORDER BY initial_query_start_time_microseconds DESC
            LIMIT 1"""
    )
    assert real_read_rows == str(read_rows) + "\n"


def test_simple_range(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number=1 FORMAT Values",
        query_id="test_equal",
    )
    assert response == "(1)"
    check_read_rows("test_equal", 1)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number > 1 and number < 6 FORMAT Values",
        query_id="test_single_range",
    )
    assert response == "(2),(3),(4),(5)"
    check_read_rows("test_single_range", 4)


def test_between(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number between 1 and 6 FORMAT Values",
        query_id="test_between",
    )
    assert response == "(1),(2),(3),(4),(5),(6)"
    check_read_rows("test_between", 6)


def test_blank_range(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number < 1 and number > 6 FORMAT Values",
        query_id="test_blank_range",
    )
    assert response == ""
    check_read_rows("test_blank_range", 0)


def test_in(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number in (2, 3) FORMAT Values",
        query_id="test_in_simple",
    )
    assert response == "(2),(3)"
    check_read_rows("test_in_simple", 2)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number in (2, 3, 3) FORMAT Values",
        query_id="test_in_with_duplicated_values",
    )
    assert response == "(2),(3)"
    check_read_rows("test_in_with_duplicated_values", 2)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number in (2, 3, 1) FORMAT Values",
        query_id="test_in_with_unordered_values",
    )
    assert response == "(1),(2),(3)"
    check_read_rows("test_in_with_unordered_values", 3)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number in (1, 2, 5) FORMAT Values",
        query_id="test_in_multiple_ranges",
    )
    assert response == "(1),(2),(5)"
    check_read_rows("test_in_multiple_ranges", 3)


def test_not_in(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number not in (2, 3) limit 3 FORMAT Values",
        query_id="test_not_in",
    )
    assert response == "(0),(1),(4)"
    check_read_rows("test_not_in", 3)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number not in (2, 4, 5) limit 4 FORMAT Values",
        query_id="test_not_in_multiple_ranges",
    )
    assert response == "(0),(1),(3),(6)"
    check_read_rows("test_not_in_multiple_ranges", 4)


def test_and(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number in (2, 4) and number > 2 FORMAT Values",
        query_id="test_and",
    )
    assert response == "(4)"
    check_read_rows("test_and", 1)


def test_or(started_cluster):
    response = node.query(
        """SELECT
                *
            FROM
                system.numbers
            WHERE
                (number > 1 and number < 3) or (number in (4, 6)) or (number > 7 and number < 9)
            FORMAT Values""",
        query_id="test_simple_or",
    )
    assert response == "(2),(4),(6),(8)"
    check_read_rows("test_simple_or", 4)

    response = node.query(
        "SELECT * FROM system.numbers WHERE (number > 1 and number < 3) or (number < 6) FORMAT Values",
        query_id="test_or_with_overlapping_ranges",
    )
    assert response == "(0),(1),(2),(3),(4),(5)"
    check_read_rows("test_or_with_overlapping_ranges", 6)


def test_not(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE not (number > 1 and number < 3) limit 5 FORMAT Values",
        query_id="test_not",
    )
    assert response == "(0),(1),(3),(4),(5)"
    check_read_rows("test_not", 5)


def test_true_or_false(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number < 3 and 1 limit 5 FORMAT Values",
        query_id="test_true",
    )
    assert response == "(0),(1),(2)"
    check_read_rows("test_true", 3)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number < 3 and 0 FORMAT Values",
        query_id="test_false",
    )
    assert response == ""
    check_read_rows("test_false", 0)


def test_limit(started_cluster):
    response = node.query(
        "SELECT * FROM system.numbers WHERE number > 2 limit 1 FORMAT Values",
        query_id="test_simple_limit",
    )
    assert response == "(3)"
    check_read_rows("test_simple_limit", 1)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number not in (2, 3) limit 1 FORMAT Values",
        query_id="test_limit_with_multi_ranges",
    )
    assert response == "(0)"
    check_read_rows("test_limit_with_multi_ranges", 1)

    response = node.query(
        "SELECT * FROM system.numbers WHERE number not in (2, 3) limit 1, 2 FORMAT Values",
        query_id="test_limit_with_offset",
    )
    assert response == "(1),(4)"
    check_read_rows("test_limit_with_offset", 3)


def test_subquery(started_cluster):
    response = node.query(
        """SELECT
                *
            FROM
                (select * FROM system.numbers WHERE number < 2) AS n
            FORMAT Values""",
        query_id="test_subquery",
    )
    assert response == "(0),(1)"
    check_read_rows("test_subquery", 2)


def test_multi_streams(started_cluster):
    response = node.query(
        """SELECT
                *
            FROM
                system.numbers_mt
            WHERE
                number > 1 and number < 7
            ORDER BY
                number
            FORMAT Values
            settings max_block_size=2""",
        query_id="test_multi_streams",
    )
    assert response == "(2),(3),(4),(5),(6)"
    check_read_rows("test_multi_streams", 5)

    response = node.query(
        """SELECT
                *
            FROM
                system.numbers_mt
            WHERE
                (number > 1 and number < 3) or (number in (4, 6)) or (number > 7 and number < 10)
            ORDER BY
                number
            FORMAT Values
            settings max_block_size=2""",
        query_id="test_multi_streams_with_multi_ranges",
    )
    assert response == "(2),(4),(6),(8),(9)"
    check_read_rows("test_multi_streams", 5)


def test_overflow(started_cluster):
    response = node.query(
        "SELECT number FROM numbers(18446744073709551614, 5) FORMAT Values",
        query_id="test_overflow",
    )
    assert response == "(18446744073709551614),(18446744073709551615),(0),(1),(2)"
    check_read_rows("test_overflow", 5)


def test_non_number_filter(started_cluster):
    response = node.query(
        "SELECT toString(number) as a FROM numbers(3) WHERE a = '1' FORMAT Values",
        query_id="test_non_number_filter",
    )
    assert response == "('1')"
    check_read_rows("test_non_number_filter", 3)

    response = node.query(
        "SELECT toString(number) as a FROM numbers(1, 4) WHERE a = '1' FORMAT Values SETTINGS max_block_size = 3",
        query_id="test_non_number_filter2",
    )
    assert response == "('1')"
    check_read_rows("test_non_number_filter2", 4)