Fix connection delay bias in pool test queue

The connection time was mistakenly counted in the execution time, which
made the test more brittle than already is.

diff --git a/tests/pool/test_pool.py b/tests/pool/test_pool.py
index 6c0f9374d839a37507c581f60319f05bd6d57e25..6abd895536561a1da5c0803e5612cf186271dea9 100644 (file)
--- a/tests/pool/test_pool.py
+++ b/tests/pool/test_pool.py
@@ -292,6 +292,7 @@ def test_queue(dsn, retries):
         with retry:
             results = []
             with pool.ConnectionPool(dsn, min_size=2) as p:
+                p.wait()
                 ts = [Thread(target=worker, args=(i,)) for i in range(6)]
                 [t.start() for t in ts]
                 [t.join() for t in ts]

diff --git a/tests/pool/test_pool_async.py b/tests/pool/test_pool_async.py
index efeb73fc8af2019d428776bb8466d891e2d3dcb0..e8593946ff521b09a235cdd8f1f571dcb6be308b 100644 (file)
--- a/tests/pool/test_pool_async.py
+++ b/tests/pool/test_pool_async.py
@@ -313,6 +313,7 @@ async def test_queue(dsn, retries):
         with retry:
             results = []
             async with pool.AsyncConnectionPool(dsn, min_size=2) as p:
+                await p.wait()
                 ts = [create_task(worker(i)) for i in range(6)]
                 await asyncio.gather(*ts)