Commit 9e493493 authored by Marek Vavruša's avatar Marek Vavruša

tests: fixed exception with canceled queries

parent 7ced7282
......@@ -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):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment