test(crdb): add "suicide" tests

diff --git a/tests/crdb/test_connection.py b/tests/crdb/test_connection.py
index e748ca330223417b61be023b977def124b43de46..be5201932b9daa5336a63db6f6fbf62c681c2e81 100644 (file)
--- 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 b400619fd176d246cb797de9ed4438dd364e23f7..69e6412a6c332dbbdc1c59a9976ee4fd895725cf 100644 (file)
--- 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()