onedns/onedns/tests/utils.py

12 lines
333 B
Python

import dnslib
from onedns.tests import conftest
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(server, port, tcp=tcp)
return dnslib.DNSRecord.parse(a_pkt)