From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Thu, 28 Oct 2021 21:20:32 +0000 (+0200)
Subject: Retry a brittle test
X-Git-Tag: 3.0.2~18
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=6e79b7e502db873a39cdffecec5b0f9a8bb5e44d;p=thirdparty%2Fpsycopg.git

Retry a brittle test
---

diff --git a/tests/test_waiting.py b/tests/test_waiting.py
index 64f75286b..deb35ea65 100644
--- 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):