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 retries to rests test_async_query_sending/test_async_connect

Browse Source
This commit is contained in:
avogar 2023-09-18 19:35:08 +00:00
parent c3a5d7dbf0
commit 838668e9df
1 changed files with 26 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
tests/integration/test_hedged_requests/test.py
View File

@ -133,7 +133,7 @@ def check_if_query_sending_was_suspended():
"SELECT value FROM system.events WHERE event='SuspendSendingQueryToShard'"
)
assert int(result) >= 1
return len(result) != 0 and int(result) >= 1
def check_if_query_sending_was_not_suspended():
@ -380,11 +380,19 @@ def test_async_connect(started_cluster):
# Restart server to reset connection pool state
NODES["node"].restart_clickhouse()
NODES["node"].query(
"SELECT hostName(), id FROM distributed_connect ORDER BY id LIMIT 1 SETTINGS prefer_localhost_replica = 0, connect_timeout_with_failover_ms=5000, async_query_sending_for_remote=1, max_threads=1, max_distributed_connections=1"
)
check_changing_replica_events(2)
check_if_query_sending_was_suspended()
attempt = 0
while attempt < 1000:
NODES["node"].query(
"SELECT hostName(), id FROM distributed_connect ORDER BY id LIMIT 1 SETTINGS prefer_localhost_replica = 0, connect_timeout_with_failover_ms=5000, async_query_sending_for_remote=1, max_threads=1, max_distributed_connections=1"
)
check_changing_replica_events(2)
if check_if_query_sending_was_suspended():
break
attempt += 1
assert attempt < 1000
NODES["node"].query("DROP TABLE distributed_connect")
@ -418,10 +426,17 @@ def test_async_query_sending(started_cluster):
)
check_if_query_sending_was_not_suspended()
NODES["node"].query(
"SELECT hostName(), id FROM distributed_query_sending ORDER BY id LIMIT 1 SETTINGS"
" prefer_localhost_replica = 0, async_query_sending_for_remote=1, max_threads = 1, max_distributed_connections=1"
)
check_if_query_sending_was_suspended()
attempt = 0
while attempt < 1000:
NODES["node"].query(
"SELECT hostName(), id FROM distributed_query_sending ORDER BY id LIMIT 1 SETTINGS"
" prefer_localhost_replica = 0, async_query_sending_for_remote=1, max_threads = 1, max_distributed_connections=1"
)
if check_if_query_sending_was_suspended():
break
attempt += 1
assert attempt < 1000
NODES["node"].query("DROP TABLE distributed_query_sending")