add kwargs for server, port, and tcp to dnsquery
parent
6fecaed95a
commit
7f2a3ca8a8
|
@ -3,7 +3,9 @@ import dnslib
|
|||
from onedns.tests import conftest
|
||||
|
||||
|
||||
def dnsquery(qname, qtype):
|
||||
def dnsquery(qname, qtype, server=None, port=None, tcp=False):
|
||||
server = server or conftest.INTERFACE
|
||||
port = port or conftest.PORT
|
||||
q = dnslib.DNSRecord(q=dnslib.DNSQuestion(qname, qtype))
|
||||
a_pkt = q.send(conftest.INTERFACE, conftest.PORT, tcp=False)
|
||||
a_pkt = q.send(server, port, tcp=tcp)
|
||||
return dnslib.DNSRecord.parse(a_pkt)
|
||||
|
|
Loading…
Reference in New Issue