From 9e493493e927570140a3d130bdad3b1f947290a6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Marek=20Vavru=C5=A1a?= <marek.vavrusa@nic.cz>
Date: Thu, 1 Oct 2015 10:39:15 +0200
Subject: [PATCH] tests: fixed exception with canceled queries
---
tests/pydnstest/testserver.py | 5 ++++-
1 file changed, 4 insertions(+), 1 deletion(-)
diff --git a/tests/pydnstest/testserver.py b/tests/pydnstest/testserver.py
index f96c5bb39..c0b854316 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):
--
GitLab