From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Wed, 8 Jun 2022 01:17:25 +0000 (+0200)
Subject: test(crdb): add "suicide" tests
X-Git-Tag: 3.1~49^2~10
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=d01276d31a02fa3c26e873eed9b6be8657798b3d;p=thirdparty%2Fpsycopg.git

test(crdb): add "suicide" tests
---

diff --git a/tests/crdb/test_connection.py b/tests/crdb/test_connection.py
index e748ca330..be5201932 100644
--- a/tests/crdb/test_connection.py
+++ b/tests/crdb/test_connection.py
@@ -41,6 +41,13 @@ def test_tpc_recover(dsn):
             conn.tpc_recover()
 
 
+def test_broken_connection(conn):
+    cur = conn.cursor()
+    with pytest.raises(psycopg.DatabaseError):
+        cur.execute("cancel session (select session_id from [show session_id])")
+    assert conn.closed
+
+
 def test_broken(conn):
     (session_id,) = conn.execute("show session_id").fetchone()
     with pytest.raises(psycopg.OperationalError):
diff --git a/tests/crdb/test_connection_async.py b/tests/crdb/test_connection_async.py
index b400619fd..69e6412a6 100644
--- a/tests/crdb/test_connection_async.py
+++ b/tests/crdb/test_connection_async.py
@@ -39,6 +39,13 @@ async def test_tpc_recover(dsn):
             await conn.tpc_recover()
 
 
+async def test_broken_connection(aconn):
+    cur = aconn.cursor()
+    with pytest.raises(psycopg.DatabaseError):
+        await cur.execute("cancel session (select session_id from [show session_id])")
+    assert aconn.closed
+
+
 async def test_broken(aconn):
     cur = await aconn.execute("show session_id")
     (session_id,) = await cur.fetchone()