From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Wed, 1 Feb 2023 00:52:57 +0000 (+0100)
Subject: test: add TypeInfo.fetch() test with autocommit transaction
X-Git-Tag: pool-3.1.7~11^2~3
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=20905fa311725b5535c46e0526c6619e2415d179;p=thirdparty%2Fpsycopg.git

test: add TypeInfo.fetch() test with autocommit transaction
---

diff --git a/tests/test_typeinfo.py b/tests/test_typeinfo.py
index 063c57762..1d74ef2bf 100644
--- a/tests/test_typeinfo.py
+++ b/tests/test_typeinfo.py
@@ -13,7 +13,7 @@ from .fix_crdb import crdb_encoding
 
 
 @pytest.mark.parametrize("name", ["text", sql.Identifier("text")])
-@pytest.mark.parametrize("status", ["IDLE", "INTRANS"])
+@pytest.mark.parametrize("status", ["IDLE", "INTRANS", None])
 @pytest.mark.parametrize(
     "encoding", ["utf8", crdb_encoding("latin1"), crdb_encoding("sql_ascii")]
 )
@@ -21,9 +21,13 @@ def test_fetch(conn, name, status, encoding):
     with conn.transaction():
         conn.execute("select set_config('client_encoding', %s, false)", [encoding])
 
-    status = getattr(TransactionStatus, status)
-    if status == TransactionStatus.INTRANS:
-        conn.execute("select 1")
+    if status:
+        status = getattr(TransactionStatus, status)
+        if status == TransactionStatus.INTRANS:
+            conn.execute("select 1")
+    else:
+        conn.autocommit = True
+        status = TransactionStatus.IDLE
 
     assert conn.info.transaction_status == status
     info = TypeInfo.fetch(conn, name)
@@ -40,7 +44,7 @@ def test_fetch(conn, name, status, encoding):
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize("name", ["text", sql.Identifier("text")])
-@pytest.mark.parametrize("status", ["IDLE", "INTRANS"])
+@pytest.mark.parametrize("status", ["IDLE", "INTRANS", None])
 @pytest.mark.parametrize(
     "encoding", ["utf8", crdb_encoding("latin1"), crdb_encoding("sql_ascii")]
 )
@@ -50,9 +54,13 @@ async def test_fetch_async(aconn, name, status, encoding):
             "select set_config('client_encoding', %s, false)", [encoding]
         )
 
-    status = getattr(TransactionStatus, status)
-    if status == TransactionStatus.INTRANS:
-        await aconn.execute("select 1")
+    if status:
+        status = getattr(TransactionStatus, status)
+        if status == TransactionStatus.INTRANS:
+            await aconn.execute("select 1")
+    else:
+        await aconn.set_autocommit(True)
+        status = TransactionStatus.IDLE
 
     assert aconn.info.transaction_status == status
     info = await TypeInfo.fetch(aconn, name)