add kwargs for server, port, and tcp to dnsquery

master
Justin Riley 2016-08-12 11:03:55 -04:00
parent 6fecaed95a
commit 7f2a3ca8a8
1 changed files with 4 additions and 2 deletions

View File

@ -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)