diff --git a/tests/pydnstest/testserver.py b/tests/pydnstest/testserver.py index f96c5bb39de36f918ebf45b4d768d80ec9fa0cc8..c0b854316966de37161d0b4d4deecb9a945b5e3d 100644 --- a/tests/pydnstest/testserver.py +++ b/tests/pydnstest/testserver.py @@ -9,7 +9,10 @@ import itertools def recvfrom_msg(stream): """ Receive DNS/UDP message. """ - data, addr = stream.recvfrom(4096) + try: + data, addr = stream.recvfrom(4096) + except: + return None, None return dns.message.from_wire(data), addr def sendto_msg(stream, message, addr):