def test_timeout_failure():
resolver = mock_resolver(['1.1.1.1'], {})
udp_sequence = [
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'raise': dns.exception.Timeout(timeout=1),
},
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'raise': dns.exception.Timeout(timeout=2),
},
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'raise': dns.exception.Timeout(timeout=3),
},
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'raise': dns.exception.Timeout(timeout=4),
},
]
with patch('dns.resolver.get_default_resolver', resolver):
with patch('dns.resolver.Resolver', resolver):
with patch('dns.query.udp', mock_query_udp(udp_sequence)):
with pytest.raises(dns.exception.Timeout) as exc:
resolver = ResolveDirectlyFromNameServers()
resolver.resolve_nameservers('example.com')
assert exc.value.kwargs['timeout'] == 4
def test_error_servfail():
resolver = mock_resolver(['1.1.1.1'], {})
udp_sequence = [
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result': create_mock_response(dns.rcode.SERVFAIL),
},
]
with patch('dns.resolver.get_default_resolver', resolver):
with patch('dns.resolver.Resolver', resolver):
with patch('dns.query.udp', mock_query_udp(udp_sequence)):
with pytest.raises(ResolverError) as exc:
resolver = ResolveDirectlyFromNameServers()
resolver.resolve_nameservers('example.com')
assert exc.value.args[
0] == 'Error SERVFAIL while querying 1.1.1.1 with query get NS for "com."'
def test_error_nxdomain():
resolver = mock_resolver(['1.1.1.1'], {})
udp_sequence = [
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result': create_mock_response(dns.rcode.NXDOMAIN),
},
]
with patch('dns.resolver.get_default_resolver', resolver):
with patch('dns.resolver.Resolver', resolver):
with patch('dns.query.udp', mock_query_udp(udp_sequence)):
with pytest.raises(dns.resolver.NXDOMAIN) as exc:
resolver = ResolveDirectlyFromNameServers()
resolver.resolve_nameservers('example.com')
assert exc.value.kwargs['qnames'] == [
dns.name.from_unicode(u'com')
]
def test_resolver_non_default():
resolver = mock_resolver(
['1.1.1.1'], {
('1.1.1.1', ): [
{
'target':
'ns.com',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '2.2.2.2'),
)),
},
{
'target':
'ns.example.com',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.example.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '3.3.3.3'),
)),
},
{
'target':
'ns.org',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '2.2.3.3'),
)),
},
{
'target':
'ns.example.org',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.example.org',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '4.4.4.4'),
)),
},
],
('3.3.3.3', ): [
{
'target':
dns.name.from_unicode(u'example.org'),
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'example.org',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '1.2.3.4'),
)),
},
],
('4.4.4.4', ): [
{
'target':
dns.name.from_unicode(u'example.org'),
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'example.org',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '1.2.3.4'),
)),
},
],
})
udp_sequence = [
{
'query_target':
dns.name.from_unicode(u'com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'com',
3600,
dns.rdata.from_text(
dns.rdataclass.IN,
dns.rdatatype.NS, 'ns.com'),
)
]),
},
{
'query_target':
dns.name.from_unicode(u'example.com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'2.2.2.2',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'example.com',
3600,
dns.rdata.from_text(
dns.rdataclass.IN,
dns.rdatatype.NS,
'ns.example.com'),
)
]),
},
{
'query_target':
dns.name.from_unicode(u'www.example.com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'3.3.3.3',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(
dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'www.example.com',
3600,
dns.rdata.from_text(
dns.rdataclass.IN, dns.rdatatype.SOA,
'ns.example.com. ns.example.com. 12345 7200 120 2419200 10800'
),
),
dns.rrset.from_rdata(
'www.example.com', 3600,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.CNAME,
'example.org'))
]),
},
{
'query_target':
dns.name.from_unicode(u'org'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'org',
3600,
dns.rdata.from_text(
dns.rdataclass.IN,
dns.rdatatype.NS, 'ns.org'),
)
]),
},
{
'query_target':
dns.name.from_unicode(u'example.org'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'2.2.3.3',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(
dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'example.org',
3600,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.NS,
'ns.example.org'),
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.NS,
'ns.example.com'),
)
]),
},
]
with patch('dns.resolver.get_default_resolver', resolver):
with patch('dns.resolver.Resolver', resolver):
with patch('dns.query.udp', mock_query_udp(udp_sequence)):
resolver = ResolveDirectlyFromNameServers(
always_ask_default_resolver=False)
assert resolver.resolve_nameservers('example.com') == [
'ns.example.com'
]
# www.example.com is a CNAME for example.org
rrset_dict = resolver.resolve('www.example.com')
assert sorted(
rrset_dict.keys()) == ['ns.example.com', 'ns.example.org']
rrset = rrset_dict['ns.example.com']
assert len(rrset) == 1
assert rrset.name == dns.name.from_unicode(u'example.org',
origin=None)
assert rrset.rdtype == dns.rdatatype.A
assert rrset[0].to_text() == u'1.2.3.4'
rrset = rrset_dict['ns.example.org']
assert len(rrset) == 1
assert rrset.name == dns.name.from_unicode(u'example.org',
origin=None)
assert rrset.rdtype == dns.rdatatype.A
assert rrset[0].to_text() == u'1.2.3.4'
# The following results should be cached:
assert resolver.resolve_nameservers('com') == ['ns.com']
print(
resolver.resolve_nameservers('example.com',
resolve_addresses=True))
assert resolver.resolve_nameservers(
'example.com', resolve_addresses=True) == ['3.3.3.3']
print(
resolver.resolve_nameservers('example.org',
resolve_addresses=True))
assert resolver.resolve_nameservers(
'example.org',
resolve_addresses=True) == ['3.3.3.3', '4.4.4.4']
def test_no_response():
fake_query = MagicMock()
fake_query.question = 'Doctor Who?'
resolver = mock_resolver(
['1.1.1.1'], {
('1.1.1.1', ): [
{
'target':
'ns.example.com',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.example.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '3.3.3.3'),
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '5.5.5.5'),
)),
},
{
'target':
'ns2.example.com',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.example.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '4.4.4.4'),
)),
},
],
('3.3.3.3', '5.5.5.5'): [
{
'target': dns.name.from_unicode(u'example.com'),
'lifetime': 10,
'result': create_mock_answer(),
},
],
('4.4.4.4', ): [
{
'target': dns.name.from_unicode(u'example.com'),
'lifetime': 10,
'raise': dns.resolver.NoAnswer(response=fake_query),
},
],
})
udp_sequence = [
{
'query_target':
dns.name.from_unicode(u'com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'com',
3600,
dns.rdata.from_text(
dns.rdataclass.IN,
dns.rdatatype.NS, 'ns.com'),
)
]),
},
{
'query_target':
dns.name.from_unicode(u'example.com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(
dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'example.com',
3600,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.NS,
'ns.example.com'),
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.NS,
'ns2.example.com'),
)
]),
},
]
with patch('dns.resolver.get_default_resolver', resolver):
with patch('dns.resolver.Resolver', resolver):
with patch('dns.query.udp', mock_query_udp(udp_sequence)):
resolver = ResolveDirectlyFromNameServers()
rrset_dict = resolver.resolve('example.com')
assert sorted(rrset_dict.keys()) == [
'ns.example.com', 'ns2.example.com'
]
assert rrset_dict['ns.example.com'] is None
assert rrset_dict['ns2.example.com'] is None
# Verify nameserver IPs
assert resolver.resolve_nameservers('example.com') == [
'ns.example.com', 'ns2.example.com'
]
assert resolver.resolve_nameservers(
'example.com', resolve_addresses=True) == [
'3.3.3.3', '4.4.4.4', '5.5.5.5'
]
def test_timeout_handling():
resolver = mock_resolver(
['1.1.1.1'], {
('1.1.1.1', ): [
{
'target': 'ns.example.com',
'lifetime': 10,
'raise': dns.exception.Timeout(timeout=10),
},
{
'target':
'ns.example.com',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.example.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '3.3.3.3'),
)),
},
{
'target': 'ns.com',
'lifetime': 10,
'raise': dns.exception.Timeout(timeout=10),
},
{
'target': 'ns.com',
'lifetime': 10,
'raise': dns.exception.Timeout(timeout=10),
},
{
'target':
'ns.com',
'lifetime':
10,
'result':
create_mock_answer(
dns.rrset.from_rdata(
'ns.com',
300,
dns.rdata.from_text(dns.rdataclass.IN,
dns.rdatatype.A, '2.2.2.2'),
)),
},
],
})
udp_sequence = [
{
'query_target': dns.name.from_unicode(u'com'),
'query_type': dns.rdatatype.NS,
'nameserver': '1.1.1.1',
'kwargs': {
'timeout': 10,
},
'raise': dns.exception.Timeout(timeout=10),
},
{
'query_target':
dns.name.from_unicode(u'com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(dns.rcode.NOERROR,
answer=[
dns.rrset.from_rdata(
'com',
3600,
dns.rdata.from_text(
dns.rdataclass.IN,
dns.rdatatype.NS, 'ns.com'),
)
]),
},
{
'query_target':
dns.name.from_unicode(u'example.com'),
'query_type':
dns.rdatatype.NS,
'nameserver':
'1.1.1.1',
'kwargs': {
'timeout': 10,
},
'result':
create_mock_response(dns.rcode.NOERROR,
authority=[
dns.rrset.from_rdata(
'example.com',
3600,
dns.rdata.from_text(
dns.rdataclass.IN,
dns.rdatatype.NS,
'ns.example.com'),
)
]),
},
]
with patch('dns.resolver.get_default_resolver', resolver):
with patch('dns.resolver.Resolver', resolver):
with patch('dns.query.udp', mock_query_udp(udp_sequence)):
resolver = ResolveDirectlyFromNameServers()
assert resolver.resolve_nameservers(
'example.com', resolve_addresses=True) == ['3.3.3.3']
# The following results should be cached:
assert resolver.resolve_nameservers('com') == ['ns.com']
assert resolver.resolve_nameservers(
'com', resolve_addresses=True) == ['2.2.2.2']
assert resolver.resolve_nameservers('example.com') == [
'ns.example.com'
]
assert resolver.resolve_nameservers(
'example.com', resolve_addresses=True) == ['3.3.3.3']