external/mailinabox
1
0
Fork 0
mirror of https://github.com/mail-in-a-box/mailinabox.git synced 2024-12-22 07:17:05 +00:00
Code Issues Releases Wiki Activity

test_dns: more error handling

Browse Source
This commit is contained in:
Joshua Tauberer 2014-06-04 19:31:55 -04:00
parent 7fa4862f1a
commit 3961e1aec3
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_dns.py
View File

@ -49,8 +49,13 @@ def test2(tests, server, description):
# do the query and format the result as a string
try:
response = dns.resolver.query(qname, rtype)
except dns.resolver.NXDOMAIN:
except dns.resolver.NoNameservers:
# host did not have an answer for this query
print("Could not connect to %s for DNS query." % server)
sys.exit(1)
except (dns.resolver.NXDOMAIN, dns.resolver.NoAnswer):
# host did not have an answer for this query; not sure what the
# difference is between the two exceptions
response = ["[no value]"]
response = ";".join(str(r) for r in response)
response = re.sub(r"(\"p=).*(\")", r"\1__KEY__\2", response) # normalize DKIM key