add kwargs for server, port, and tcp to dnsquery
parent
6fecaed95a
commit
7f2a3ca8a8
|
@ -3,7 +3,9 @@ import dnslib
|
||||||
from onedns.tests import conftest
|
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))
|
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)
|
return dnslib.DNSRecord.parse(a_pkt)
|
||||||
|
|
Loading…
Reference in New Issue