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 901058ec4bb4bb7fb22a99e2d8e0e74a502419b2..cbe33a2dcf447569c2e917dfa825b26a7dd09b47 100644 (file)
--- 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