def setUp(self):
"""
Setups everything needed for the tests.
"""
self.dns_lookup = Dns()
self.subject = "8.8.8.8"
def setUp(self):
"""
Setups everything needed for the tests.
"""
self.dns_lookup = Dns()
self.subject = "www.twitter.com"
def test_custom_lifetime(self):
"""
Tests the initiation of a custom lifetime.
"""
dns_lookup = Dns(lifetime=5)
expected = 2.0
actual = dns_lookup.resolver.timeout
self.assertEqual(expected, actual)
expected = 5.0
actual = dns_lookup.resolver.lifetime
self.assertEqual(expected, actual)
def test_custom_nameservers(self):
"""
Tests the initiation of a custom nameserver.
"""
dns_lookup = Dns(lifetime=5, dns_server="8.8.8.8")
expected = ["8.8.8.8"]
actual = dns_lookup.resolver.nameservers
self.assertEqual(expected, actual)
self.dns_lookup.update_nameserver("8.8.8.8")
actual = dns_lookup.resolver.nameservers
self.assertEqual(expected, actual)
def test_custom_lifetime_not_int_nor_fload(self):
"""
Tests the initiation of a custom lifetime which
is not an integer nor a float.
"""
dns_lookup = Dns(lifetime="Hello, World!")
expected = 2.0
actual = dns_lookup.resolver.timeout
self.assertEqual(expected, actual)
expected = 4.0
actual = dns_lookup.resolver.lifetime
self.assertEqual(expected, actual)
def test_custom_lifetime_same_as_timeout(self):
"""
Tests the initiation of a custom lifetime which
is equal to the default timeout.
"""
dns_lookup = Dns(lifetime=2)
expected = 2.0
actual = dns_lookup.resolver.timeout
self.assertEqual(expected, actual)
expected = 4.0
actual = dns_lookup.resolver.lifetime
self.assertEqual(expected, actual)