def test_query_value_invalid_dict(self):
value_dict = {
'A': '127.0.0.1',
'AAAAA': [ '::1', "::2" ],
'nonexistant' : [ '::1' ]
}
with self.assertRaises(RuntimeError):
a = DNSAnswerConfig(value_dict)
def test_ipv4_and_ipv6_list_generation(self):
a = DNSAnswerConfig(['::1', '127.0.0.1', '::2', '127.0.0.2'])
self.assertEqual(True, 'AAAA' in a.value_dict)
self.assertEqual(2, len(a.value_dict['AAAA']))
self.assertEqual('::1', a.value_dict['AAAA'][0])
self.assertEqual('::2', a.value_dict['AAAA'][1])
self.assertEqual(True, 'A' in a.value_dict)
self.assertEqual(2, len(a.value_dict['A']))
self.assertEqual('127.0.0.1', a.value_dict['A'][0])
self.assertEqual('127.0.0.2', a.value_dict['A'][1])
def test_is_valid_query_type(self):
a = DNSAnswerConfig(None)
self.assertEqual(True, a.isValidQueryType('A'))
self.assertEqual(True, a.isValidQueryType('AAAA'))
self.assertEqual(True, a.isValidQueryType('NS'))
self.assertEqual(True, a.isValidQueryType('MX'))
self.assertEqual(False, a.isValidQueryType('Foobar'))
def test_query_value_valid_dict(self):
value_dict = {
'A': '127.0.0.1',
'AAAA': [ '::1', "::2" ],
'NS' : [ '::1' ]
}
result_dict = {
'A': [ '127.0.0.1' ],
'AAAA': [ '::1', "::2" ],
'NS' : [ '::1' ]
}
a = DNSAnswerConfig(value_dict)
self.assertEqual(result_dict, a.value_dict)
def test_is_ipv6_address(self):
a = DNSAnswerConfig(None)
self.assertEqual(False, a.isIPv6Address('127.0.0.1'))
self.assertEqual(False, a.isIPv6Address('invalid'))
self.assertEqual(True, a.isIPv6Address('::1'))
def test_ip_list_invalid_address(self):
with self.assertRaises(RuntimeError):
a = DNSAnswerConfig('invalid_address')
def test_invalid_list_generation(self):
with self.assertRaises(RuntimeError):
a = DNSAnswerConfig(['::1', 'invalid_address'])
def test_ipv6_generation(self):
a = DNSAnswerConfig('::1')
self.assertEqual(True, 'AAAA' in a.value_dict)
self.assertEqual(['::1'], a.value_dict['AAAA'])
def test_ip_generation(self):
a = DNSAnswerConfig('127.0.0.1')
self.assertEqual(True, 'A' in a.value_dict)
self.assertEqual(['127.0.0.1'], a.value_dict['A'])