From 7f2a3ca8a8e1ea17fe7749cdb9de5e7bdb9a963d Mon Sep 17 00:00:00 2001 From: Justin Riley Date: Fri, 12 Aug 2016 11:03:55 -0400 Subject: [PATCH] add kwargs for server, port, and tcp to dnsquery --- onedns/tests/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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)