def test_dzt_query_no_result(monkeypatch): rdataset = dns.rdataset.from_text(1, 1, 28800, ip_192_0_2_1) class Result(object): def __init__(self): self.answer = [] def udp_mock(query_message, nameserver, timeout=0): return Result() monkeypatch.setattr(dns.query, 'udp', udp_mock) record = Record('example.com', rdataset) record.query('192.0.2.2', False) assert record.rdataset_query is None
def test_record_query_no_recursion(monkeypatch): rdataset = dns.rdataset.from_text(1, 1, 28800, ip_192_0_2_1) class Answer(object): def to_rdataset(self): return rdataset class Result(object): def __init__(self): self.answer = [Answer()] def udp_mock(query_message, nameserver, timeout=0): return Result() monkeypatch.setattr(dns.query, 'udp', udp_mock) record = Record('example.com', rdataset) record.query('192.0.2.2', True) assert record.query_msg.flags == 0
def test_dzt_query_no_result(monkeypatch): rdataset = dns.rdataset.from_text(1, 1, 28800, ip_192_0_2_1) class Result(object): def __init__(self): self.answer = [] def udp_mock(query_message, nameserver, timeout=0): return Result() monkeypatch.setattr( dns.query, 'udp', udp_mock ) record = Record('example.com', rdataset) record.query('192.0.2.2', False) assert record.rdataset_query is None
def test_record_query_no_recursion(monkeypatch): rdataset = dns.rdataset.from_text(1, 1, 28800, ip_192_0_2_1) class Answer(object): def to_rdataset(self): return rdataset class Result(object): def __init__(self): self.answer = [Answer()] def udp_mock(query_message, nameserver, timeout=0): return Result() monkeypatch.setattr( dns.query, 'udp', udp_mock ) record = Record('example.com', rdataset) record.query('192.0.2.2', True) assert record.query_msg.flags == 0
def test_record(rdataset_file, rdataset_query, rdataset_match, ttl_match): record = Record('www.example.com', dns.rdataset.from_text(*rdataset_file)) record.rdataset_query = dns.rdataset.from_text(*rdataset_query) assert record.rdataset_match is rdataset_match assert record.ttl_match is ttl_match