def test_one_A(self):
header = _Header(0, MessageType.RESPONSE, 1)
question = [_Question('google.com')]
a_resource_data = _AResourceData(b'\xac\xd9\x0en')
answers = [
_ResourceRecord('google.com',
type_=RRType.A,
length=4,
data=a_resource_data,
ttl=0)
]
authorities = []
additions = []
actual = Answer(header, question, answers, authorities, additions)
self.assertEqual(0, actual.header.identifier)
self.assertEqual(MessageType.RESPONSE, actual.header.message_type)
self.assertEqual(QueryType.STANDARD, actual.header.query_type)
self.assertEqual(False, actual.header.is_authority_answer)
self.assertEqual(False, actual.header.is_truncated)
self.assertEqual(False, actual.header.is_recursion_desired)
self.assertEqual(False, actual.header.is_recursion_available)
self.assertEqual(ResponseType.NO_ERROR, actual.header.response_type)
self.assertEqual('google.com', actual.questions[0].name)
self.assertEqual(RRType.A, actual.questions[0].type_)
self.assertEqual(RRClass.IN, actual.questions[0].class_)
self.assertEqual('google.com', actual.answers[0].name)
self.assertEqual(RRType.A, actual.answers[0].type_)
self.assertEqual(RRClass.IN, actual.answers[0].class_)
self.assertEqual(0, actual.answers[0].ttl)
self.assertEqual(4, actual.answers[0].length)
self.assertEqual(a_resource_data.ip, actual.answers[0].data.ip)
def test_resolve_ip4_recursive(mock_get_answer):
args = Namespace(hostname='vk.com',
protocol='udp',
server='8.8.8.8',
port=53,
timeout=10,
inverse=False,
dump=False,
debug=False,
ipv6=False)
header = _Header(1823,
MessageType.RESPONSE,
question_count=1,
answer_count=1)
questions = [_Question('vk.com')]
answers = [
_ResourceRecord('vk.com',
type_=RRType.A,
length=4,
data=_AResourceData(b'\x57\xf0\xb6\xe0'))
]
mock_get_answer.return_value = Answer(header, questions, answers, [], [])
assert resolver.resolve(args).answers[0].data.ip == '87.240.182.224'
def test_get_primary_name_server_1(mock_find_name_servers, mock_get_answer):
mock_find_name_servers.return_value = [
'ns4.vkontakte.ru', 'ns2.vkontakte.ru'
]
header = _Header(9633,
MessageType.RESPONSE,
question_count=1,
answer_count=4)
questions = [_Question('vk.com', type_=RRType.NS)]
answers = [
_ResourceRecord(
'vk.com',
type_=RRType.NS,
length=18,
data=_SOAResourceData(
b'\xc0\x0c\x00\x06\x00\x01\x00\x00\x02\x0b\x00\x44'
b'\x03ns1\x09vkontakte\x02ru\x00'
b'\x03\x00\x00\x00\x00\x00\x00\x00\x00\x00'
b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00',
offset=0))
]
mock_get_answer.return_value = Answer(header, questions, answers, [], [])
assert resolver.get_primary_name_server('vk.com',
protocol='udp',
port=53,
timeout=10) == 'ns1.vkontakte.ru'
def test_NS_question_2(self):
question = _Question('python.org', type_=RRType.NS)
expected = b'\x06python\x03org\x00\x00\x02\x00\x01'
actual = question.to_bytes()
self.assertEqual(expected, actual)
def test_A_question_2(self):
question = _Question('python.org')
expected = b'\x06python\x03org\x00\x00\x01\x00\x01'
actual = question.to_bytes()
self.assertEqual(expected, actual)
def test_NS_question(self):
in_bytes = b'\x01\x00\x80\x00\x00\x01\x00\x01\x00\x00\x00\x00' \
b'\x06python\x03org\x00\x00\x02\x00\x01'
expected = _Question('python.org', RRType.NS)
actual = _Question.from_bytes(in_bytes, 12).question
self.equal_questions(expected, actual)
def test_get_primary_name_server_2(mock_find_name_servers, mock_get_answer):
mock_find_name_servers.return_value = [
'ns4.vkontakte.ru', 'ns2.vkontakte.ru'
]
header = _Header(9633,
MessageType.RESPONSE,
question_count=1,
answer_count=0)
questions = [_Question('vk.com', type_=RRType.NS)]
mock_get_answer.return_value = Answer(header, questions, [], [], [])
assert resolver.get_primary_name_server(
'vk.com', protocol='udp', port=53, timeout=10) is None
def test_get_zone_dump(mock_find_name_servers, mock_get_answer):
mock_find_name_servers.return_value = ['ns1.vk.ru', 'ns4.vk.ru']
f_header = _Header(2932,
MessageType.RESPONSE,
1,
response_type=ResponseType.REFUSED)
s_header = _Header(2928, MessageType.RESPONSE, 1, answer_count=1)
questions = [_Question('vk.com', RRType.AXFR)]
mock_get_answer.side_effect = [
Answer(f_header, questions, [], [], []),
Answer(s_header, questions, [], [], [])
]
assert not resolver.get_zone_dump('vk.com', port=53, timeout=10).answers
def test_resolve_reverse_lookup(mock_get_answer):
args = Namespace(hostname='87.240.182.224',
protocol='udp',
server='8.8.8.8',
port=53,
timeout=10,
inverse=True,
dump=False,
debug=False,
ipv6=False)
f_header = _Header(1823,
MessageType.RESPONSE,
question_count=1,
answer_count=0,
authority_count=1)
s_header = _Header(2938,
MessageType.RESPONSE,
question_count=1,
answer_count=1)
questions = [_Question('vk.com')]
authorities = [
_ResourceRecord('vk.com',
type_=RRType.NS,
length=18,
data=_NSResourceData(
b'\xc0\x0c\x00\x02\x00\x01\x00\x00\x02\x0b\x00\x12'
b'\x03ns4\x09vkontakte\x02ru\x00',
offset=0))
]
answers = [
_ResourceRecord('vk.com',
type_=RRType.PTR,
length=8,
data=_PTRResourceData(b'\x02vk\x03com\x00'))
]
first = Answer(f_header, questions, [], authorities, [])
second = Answer(s_header, questions, answers, [], [])
mock_get_answer.side_effect = [first, second]
assert (
resolver.resolve_reverse_lookup(args).answers[0].data.name == 'vk.com')
def test_find_name_servers(mock_get_answer, mock_get_root_servers):
mock_get_root_servers.return_value = {'198.41.0.4'}
header = _Header(9633,
MessageType.RESPONSE,
question_count=1,
answer_count=4)
questions = [_Question('vk.com', type_=RRType.NS)]
answers = [
_ResourceRecord('vk.com',
type_=RRType.NS,
length=18,
data=_NSResourceData(
b'\xc0\x0c\x00\x02\x00\x01\x00\x00\x02\x0b\x00\x12'
b'\x03ns4\x09vkontakte\x02ru\x00',
offset=0))
]
mock_get_answer.return_value = Answer(header, questions, answers, [], [])
assert resolver.find_name_servers('vk.com',
protocol='udp',
port=53,
timeout=10) == ['ns4.vkontakte.ru']
def test_NS_question(self):
question = _Question('google.com', RRType.NS)
self.assertEqual(question.name, 'google.com')
self.assertEqual(question.type_, RRType.NS)
self.assertEqual(question.class_, RRClass.IN)
def test_A_question(self):
question = _Question('yandex.com')
self.assertEqual(question.name, 'yandex.com')
self.assertEqual(question.type_, RRType.A)
self.assertEqual(question.class_, RRClass.IN)
