Add type variable to *Cursor in connection tests

diff --git a/tests/test_connection.py b/tests/test_connection.py
index 42c5c849ab5a40698eff7f9026d54dfeffeb1879..09cf1df9291bb51068cd96a9bfdadef472747c8c 100644 (file)
--- a/tests/test_connection.py
+++ b/tests/test_connection.py
@@ -504,7 +504,7 @@ def test_fileno(conn):
 def test_cursor_factory(conn):
     assert conn.cursor_factory is psycopg.Cursor
 
-    class MyCursor(psycopg.Cursor):
+    class MyCursor(psycopg.Cursor[psycopg.rows.Row]):
         pass
 
     conn.cursor_factory = MyCursor
@@ -518,7 +518,7 @@ def test_cursor_factory(conn):
 def test_server_cursor_factory(conn):
     assert conn.server_cursor_factory is psycopg.ServerCursor
 
-    class MyServerCursor(psycopg.ServerCursor):
+    class MyServerCursor(psycopg.ServerCursor[psycopg.rows.Row]):
         pass
 
     conn.server_cursor_factory = MyServerCursor

diff --git a/tests/test_connection_async.py b/tests/test_connection_async.py
index 650e5f4ea05a7d83c49cb81d9d2335d66b549a70..fe608f2053b3c65981d81ed55cc039e9d81bf814 100644 (file)
--- a/tests/test_connection_async.py
+++ b/tests/test_connection_async.py
@@ -513,7 +513,7 @@ async def test_fileno(aconn):
 async def test_cursor_factory(aconn):
     assert aconn.cursor_factory is psycopg.AsyncCursor
 
-    class MyCursor(psycopg.AsyncCursor):
+    class MyCursor(psycopg.AsyncCursor[psycopg.rows.Row]):
         pass
 
     aconn.cursor_factory = MyCursor
@@ -527,7 +527,7 @@ async def test_cursor_factory(aconn):
 async def test_server_cursor_factory(aconn):
     assert aconn.server_cursor_factory is psycopg.AsyncServerCursor
 
-    class MyServerCursor(psycopg.AsyncServerCursor):
+    class MyServerCursor(psycopg.AsyncServerCursor[psycopg.rows.Row]):
         pass
 
     aconn.server_cursor_factory = MyServerCursor