diff --git a/dns/_asyncio_backend.py b/dns/_asyncio_backend.py index 6a5633269..17bd0f788 100644 --- a/dns/_asyncio_backend.py +++ b/dns/_asyncio_backend.py @@ -79,7 +79,7 @@ async def getsockname(self): return self.transport.get_extra_info('sockname') -class StreamSocket(dns._asyncbackend.DatagramSocket): +class StreamSocket(dns._asyncbackend.StreamSocket): def __init__(self, af, reader, writer): self.family = af self.reader = reader diff --git a/dns/_curio_backend.py b/dns/_curio_backend.py index 894e87c8f..6fa7b3a17 100644 --- a/dns/_curio_backend.py +++ b/dns/_curio_backend.py @@ -49,7 +49,7 @@ async def getsockname(self): return self.socket.getsockname() -class StreamSocket(dns._asyncbackend.DatagramSocket): +class StreamSocket(dns._asyncbackend.StreamSocket): def __init__(self, socket): self.socket = socket self.family = socket.family diff --git a/dns/_trio_backend.py b/dns/_trio_backend.py index 82512cdbc..a00d4a4e5 100644 --- a/dns/_trio_backend.py +++ b/dns/_trio_backend.py @@ -49,7 +49,7 @@ async def getsockname(self): return self.socket.getsockname() -class StreamSocket(dns._asyncbackend.DatagramSocket): +class StreamSocket(dns._asyncbackend.StreamSocket): def __init__(self, family, stream, tls=False): self.family = family self.stream = stream