Skip to content

Commit cb419e4

Browse files
fix flaky test issue (#1392)
1 parent 4aba1b6 commit cb419e4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/integ/test_async.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -245,10 +245,10 @@ def test_close_connection_with_completed_async_queries(conn_cnx):
245245
cur.execute_async("select 2")
246246
qid2 = cur.sfqid
247247
while con.is_still_running(
248-
con._get_query_status(qid1)
248+
con._get_query_status(qid1)[0]
249249
): # use _get_query_status to avoid caching
250250
time.sleep(1)
251-
while con.is_still_running(con._get_query_status(qid2)):
251+
while con.is_still_running(con._get_query_status(qid2)[0]):
252252
time.sleep(1)
253253
assert con._all_async_queries_finished()
254254
assert len(con._done_async_sfqids) == 2 and con.rest is None

0 commit comments

Comments
 (0)