Retry a brittle test

diff --git a/tests/test_waiting.py b/tests/test_waiting.py
index 64f75286b3792bf219809cd40a42b2ab36e10406..deb35ea6594302bb0453043ef8c6fc9a25ea215b 100644 (file)
--- a/tests/test_waiting.py
+++ b/tests/test_waiting.py
@@ -22,10 +22,12 @@ timeouts = [
 
 
 @pytest.mark.parametrize("timeout", timeouts)
-def test_wait_conn(dsn, timeout):
-    gen = generators.connect(dsn)
-    conn = waiting.wait_conn(gen, **timeout)
-    assert conn.status == ConnStatus.OK
+def test_wait_conn(dsn, timeout, retries):
+    for retry in retries:
+        with retry:
+            gen = generators.connect(dsn)
+            conn = waiting.wait_conn(gen, **timeout)
+            assert conn.status == ConnStatus.OK
 
 
 def test_wait_conn_bad(dsn):