From: Denis Laxalde <denis.laxalde@dalibo.com>
Date: Thu, 4 Nov 2021 14:58:50 +0000 (+0100)
Subject: Add type parameter to AsyncConnection in dns tests
X-Git-Tag: 3.0.3~3^2~27
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=72bc921cbf0ce6c263473ce10da2f769c031f6f6;p=thirdparty%2Fpsycopg.git

Add type parameter to AsyncConnection in dns tests
---

diff --git a/tests/test_dns.py b/tests/test_dns.py
index 2165cc981..974ac6056 100644
--- 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)