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()