From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Sat, 6 Apr 2024 19:29:37 +0000 (+0000)
Subject: test: fix cancel connection status test on Unix socket connection
X-Git-Tag: 3.2.0~59
X-Git-Url: http://git.ipfire.org/gitweb/gitweb.cgi?a=commitdiff_plain;h=640d6046be5e8b660d7782bb0d64f8120f7c1108;p=thirdparty%2Fpsycopg.git

test: fix cancel connection status test on Unix socket connection
---

diff --git a/tests/pq/test_pgconn.py b/tests/pq/test_pgconn.py
index 901058ec4..cbe33a2dc 100644
--- a/tests/pq/test_pgconn.py
+++ b/tests/pq/test_pgconn.py
@@ -451,7 +451,9 @@ def test_cancel_conn_nonblocking(pgconn):
         cancel_conn = pgconn.cancel_conn()
         assert cancel_conn.status == pq.ConnStatus.ALLOCATED
         cancel_conn.start()
-        assert cancel_conn.status == pq.ConnStatus.STARTED
+        # On network sockets, connection starts with STARTED.
+        # On Unix sockets, connection starts with MADE.
+        assert cancel_conn.status in (pq.ConnStatus.STARTED, pq.ConnStatus.MADE)
         wait_cancel(cancel_conn)
         assert cancel_conn.status == pq.ConnStatus.OK