vitalif
/
onedns
1
0
Fork 0
Code Issues 1 Pull Requests Releases Wiki Activity

tests: move dns query code to onedns.tests.utils

master
Justin Riley 2016-08-03 12:01:41 -04:00
parent e6a1724251
commit 9ba49ee677
2 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
onedns/tests/test_dynamicresolver.py
View File

@ -3,6 +3,7 @@ import dnslib
from IPy import IP
from onedns.tests import utils
from onedns.tests import conftest
@ -23,9 +24,7 @@ def test_lookup(dns, qname, qtype, output):
dns.clear()
dns.add_host(HOST, HOST_IP)
try:
q = dnslib.DNSRecord(q=dnslib.DNSQuestion(qname, qtype))
a_pkt = q.send(conftest.INTERFACE, conftest.PORT, tcp=False)
a = dnslib.DNSRecord.parse(a_pkt)
a = utils.dnsquery(qname, qtype)
assert a.short() == output
finally:
dns.close()
@ -34,10 +33,7 @@ def test_lookup(dns, qname, qtype, output):
def test_nxdomain(dns):
dns.clear()
try:
q = dnslib.DNSRecord(q=dnslib.DNSQuestion(
'unknownhost', dnslib.QTYPE.A))
a_pkt = q.send(conftest.INTERFACE, conftest.PORT, tcp=False)
a = dnslib.DNSRecord.parse(a_pkt)
a = utils.dnsquery('unknownhost', dnslib.QTYPE.A)
assert dnslib.RCODE.get(a.header.rcode) == 'NXDOMAIN'
finally:
dns.close()

9
onedns/tests/utils.py Normal file
View File

@ -0,0 +1,9 @@
import dnslib
from onedns.tests import conftest
def dnsquery(qname, qtype):
q = dnslib.DNSRecord(q=dnslib.DNSQuestion(qname, qtype))
a_pkt = q.send(conftest.INTERFACE, conftest.PORT, tcp=False)
return dnslib.DNSRecord.parse(a_pkt)