Add type parameter to AsyncConnection in dns tests

diff --git a/tests/test_dns.py b/tests/test_dns.py
index 2165cc981e5296021b45fd96c04283f3421e054b..974ac605696318e9173164a28fd7923707b009c8 100644 (file)
--- a/tests/test_dns.py
+++ b/tests/test_dns.py
@@ -2,6 +2,7 @@ import pytest
 
 import psycopg
 from psycopg.conninfo import conninfo_to_dict
+from psycopg.rows import Row
 
 pytestmark = [pytest.mark.dns]
 
@@ -131,7 +132,7 @@ async def test_resolve_hostaddr_conn(monkeypatch, fake_resolve):
     )
 
     # TODO: not enabled by default, but should be usable to make a subclass
-    class AsyncDnsConnection(psycopg.AsyncConnection):
+    class AsyncDnsConnection(psycopg.AsyncConnection[Row]):
         @classmethod
         async def _get_connection_params(cls, conninfo, **kwargs):
             params = await super()._get_connection_params(conninfo, **kwargs)