tests: move dns query code to onedns.tests.utils
parent
e6a1724251
commit
9ba49ee677
|
@ -3,6 +3,7 @@ import dnslib
|
||||||
|
|
||||||
from IPy import IP
|
from IPy import IP
|
||||||
|
|
||||||
|
from onedns.tests import utils
|
||||||
from onedns.tests import conftest
|
from onedns.tests import conftest
|
||||||
|
|
||||||
|
|
||||||
|
@ -23,9 +24,7 @@ def test_lookup(dns, qname, qtype, output):
|
||||||
dns.clear()
|
dns.clear()
|
||||||
dns.add_host(HOST, HOST_IP)
|
dns.add_host(HOST, HOST_IP)
|
||||||
try:
|
try:
|
||||||
q = dnslib.DNSRecord(q=dnslib.DNSQuestion(qname, qtype))
|
a = utils.dnsquery(qname, qtype)
|
||||||
a_pkt = q.send(conftest.INTERFACE, conftest.PORT, tcp=False)
|
|
||||||
a = dnslib.DNSRecord.parse(a_pkt)
|
|
||||||
assert a.short() == output
|
assert a.short() == output
|
||||||
finally:
|
finally:
|
||||||
dns.close()
|
dns.close()
|
||||||
|
@ -34,10 +33,7 @@ def test_lookup(dns, qname, qtype, output):
|
||||||
def test_nxdomain(dns):
|
def test_nxdomain(dns):
|
||||||
dns.clear()
|
dns.clear()
|
||||||
try:
|
try:
|
||||||
q = dnslib.DNSRecord(q=dnslib.DNSQuestion(
|
a = utils.dnsquery('unknownhost', dnslib.QTYPE.A)
|
||||||
'unknownhost', dnslib.QTYPE.A))
|
|
||||||
a_pkt = q.send(conftest.INTERFACE, conftest.PORT, tcp=False)
|
|
||||||
a = dnslib.DNSRecord.parse(a_pkt)
|
|
||||||
assert dnslib.RCODE.get(a.header.rcode) == 'NXDOMAIN'
|
assert dnslib.RCODE.get(a.header.rcode) == 'NXDOMAIN'
|
||||||
finally:
|
finally:
|
||||||
dns.close()
|
dns.close()
|
||||||
|
|
|
@ -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)
|
Loading…
Reference in New Issue