From 72bc921cbf0ce6c263473ce10da2f769c031f6f6 Mon Sep 17 00:00:00 2001 From: Denis Laxalde Date: Thu, 4 Nov 2021 15:58:50 +0100 Subject: [PATCH] Add type parameter to AsyncConnection in dns tests --- tests/test_dns.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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) -- 2.47.3