Make db_test._perform_query compatible with Psycopg 2.9

Remove context handler for connection. Complements psycopg/psycopg2#941.
Museum-Barberini · Aug 17, 2021 · dc69f8b · dc69f8b
1 parent d402361
commit dc69f8b
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/_utils/db_test.py b/tests/_utils/db_test.py
@@ -55,10 +55,9 @@ def _perform_query(query):
         database='postgres'
     )
     try:
-        with connection:
-            connection.autocommit = True  # required for meta queries
-            with connection.cursor() as cursor:
-                return cursor.execute(query)
+        connection.autocommit = True  # required for meta queries
+        with connection.cursor() as cursor:
+            return cursor.execute(query)
     finally:
         connection.close()