From a8067fb3dd6facbc6f0e2d5ab6fff6099157ee59 Mon Sep 17 00:00:00 2001 From: kssenii Date: Tue, 29 Jun 2021 20:29:21 +0000 Subject: [PATCH] Update test --- tests/integration/test_odbc_interaction/test.py | 8 ++------ 1 file changed, 2 insertions(+), 6 deletions(-) diff --git a/tests/integration/test_odbc_interaction/test.py b/tests/integration/test_odbc_interaction/test.py index 403d6f58972..39a283448f5 100644 --- a/tests/integration/test_odbc_interaction/test.py +++ b/tests/integration/test_odbc_interaction/test.py @@ -554,9 +554,7 @@ def test_concurrent_queries(started_cluster): busy_pool = Pool(5) p = busy_pool.map_async(node_insert, range(5)) p.wait() - result = node1.query("SELECT count() FROM test_pg_table", user='default') - logging.debug(result) - assert(int(result) == 5 * 5 * 1000) + assert_eq_with_retry(node1, "SELECT count() FROM test_pg_table", str(5*5*1000)) def node_insert_select(_): for i in range(5): @@ -566,9 +564,7 @@ def test_concurrent_queries(started_cluster): busy_pool = Pool(5) p = busy_pool.map_async(node_insert_select, range(5)) p.wait() - result = node1.query("SELECT count() FROM test_pg_table", user='default') - logging.debug(result) - assert(int(result) == 5 * 5 * 1000 * 2) + assert_eq_with_retry(node1, "SELECT count() FROM test_pg_table", str(5*5*1000*2)) node1.query('DROP TABLE test_pg_table;') cursor.execute('DROP TABLE clickhouse.test_pg_table;')