def test_check_query(self):
res = Resolver(5, False, 3600)
hostname, alias, ips = res.gethostbyname("nickstracke.xyz")
ip1 = ips[0].rdata.address
ip2 = socket.gethostbyname("nickstracke.xyz")
self.assertEqual(ip1, ip2)
res.shutdown()
class TestResolverCache(TestCase):
"""Resolver tests with cache enabled"""
def setUp(self):
self.RC = RecordCache(100,
"dns/ResolverTestCache.cache") #never written to
self.res = Resolver(5, True, -1)
def test_resolver_caching(self):
hostname = "google.com"
t = time.time()
self.res.gethostbyname(hostname)
d1 = time.time() - t
t = time.time()
self.res.gethostbyname(hostname)
d2 = time.time() - t
self.assertTrue(d2 < d1)
def tearDown(self):
self.res.shutdown()
