thevar1able/ClickHouse
1
0
Fork 0
mirror of https://github.com/ClickHouse/ClickHouse.git synced 2024-09-20 00:30:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing asserts

Browse Source
This commit is contained in:
János Benjamin Antal 2024-01-04 10:18:04 +00:00
parent 2517d23e27
commit 1a1fa07128
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/integration/test_storage_kafka/test.py
View File

@ -892,12 +892,14 @@ def test_kafka_formats(kafka_cluster):
"""
expected_rows_count = raw_expected.count("\n")
instance.query_with_retry(
result_checker = lambda res: res.count("\n") == expected_rows_count
res = instance.query_with_retry(
f"SELECT * FROM test.kafka_{list(all_formats.keys())[-1]}_mv;",
retry_count=30,
sleep_time=1,
check_callback=lambda res: res.count("\n") == expected_rows_count,
check_callback=result_checker,
)
assert result_checker(res)
for format_name, format_opts in list(all_formats.items()):
logging.debug(("Checking {}".format(format_name)))
@ -3808,12 +3810,14 @@ def test_kafka_formats_with_broken_message(kafka_cluster):
"""
expected_rows_count = raw_expected.count("\n")
instance.query_with_retry(
result_checker = lambda res: res.count("\n") == expected_rows_count
res = instance.query_with_retry(
f"SELECT * FROM test.kafka_data_{list(all_formats.keys())[-1]}_mv;",
retry_count=30,
sleep_time=1,
check_callback=lambda res: res.count("\n") == expected_rows_count,
check_callback=result_checker,
)
assert result_checker(res)
for format_name, format_opts in list(all_formats.items()):
logging.debug(f"Checking {format_name}")