mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-14 18:32:29 +00:00
245 lines
7.2 KiB
Python
245 lines
7.2 KiB
Python
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)
|