blob: 06f1ac1c1219d67c646522bf1fd0c0ef8e1af4be (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
|
# Copyright (C) Dnspython Contributors, see LICENSE for text of ISC license
import socket
import unittest
import dns.name
import dns.rdataclass
import dns.rdatatype
import dns.resolver
# Some tests require the internet to be available to run, so let's
# skip those if it's not there.
_network_available = True
try:
socket.gethostbyname('dnspython.org')
except socket.gaierror:
_network_available = False
@unittest.skipIf(not _network_available, "Internet not reachable")
class OverrideSystemResolverTestCase(unittest.TestCase):
def test_override(self):
res = dns.resolver.Resolver()
res.nameservers = ['8.8.8.8']
res.cache = dns.resolver.LRUCache()
dns.resolver.override_system_resolver(res)
self.assertTrue(socket.getaddrinfo is
dns.resolver._getaddrinfo)
socket.gethostbyname('www.dnspython.org')
answer = res.cache.get((dns.name.from_text('www.dnspython.org.'),
dns.rdatatype.A, dns.rdataclass.IN))
self.assertTrue(answer is not None)
res.cache.flush()
socket.gethostbyname_ex('www.dnspython.org')
answer = res.cache.get((dns.name.from_text('www.dnspython.org.'),
dns.rdatatype.A, dns.rdataclass.IN))
self.assertTrue(answer is not None)
res.cache.flush()
socket.getfqdn('8.8.8.8')
answer = res.cache.get((dns.name.from_text('8.8.8.8.in-addr.arpa.'),
dns.rdatatype.PTR, dns.rdataclass.IN))
self.assertTrue(answer is not None)
res.cache.flush()
socket.gethostbyaddr('8.8.8.8')
answer = res.cache.get((dns.name.from_text('8.8.8.8.in-addr.arpa.'),
dns.rdatatype.PTR, dns.rdataclass.IN))
self.assertTrue(answer is not None)
dns.resolver.restore_system_resolver()
self.assertTrue(socket.getaddrinfo is
dns.resolver._original_getaddrinfo)
|
