From 640d6046be5e8b660d7782bb0d64f8120f7c1108 Mon Sep 17 00:00:00 2001 From: Daniele Varrazzo Date: Sat, 6 Apr 2024 19:29:37 +0000 Subject: [PATCH] test: fix cancel connection status test on Unix socket connection --- tests/pq/test_pgconn.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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 -- 2.47.3