diff --git a/onedns/tests/utils.py b/onedns/tests/utils.py index f2a8d60..5865e26 100644 --- a/onedns/tests/utils.py +++ b/onedns/tests/utils.py @@ -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)