def test_positive_1(self):
number = 16
expected = b'\x00\x10'
actual = _encode_number(number)
self.assertEqual(expected, actual)
def test_zero(self):
number = 0
expected = b'\x00\x00'
actual = _encode_number(number)
self.assertEqual(expected, actual)
def test_standard_A_recursive_query(self):
query = Query('github.com', RRType.A, is_recursion_desired=True)
id_bytes = _encode_number(query.header.identifier)
expected = id_bytes + b'\x01\x00\x00\x01\x00\x00\x00\x00\x00\00' \
b'\x06github\x03com\x00\x00\x01\x00\x01'
actual = query.to_bytes()
self.assertEqual(expected, actual)
def test_standard_A_query(self):
query = Query('google.com', RRType.A)
id_bytes = _encode_number(query.header.identifier)
expected = id_bytes + b'\x01\x00\x00\x01\x00\x00\x00\x00\x00\00' \
b'\x06google\x03com\x00\x00\x01\x00\x01'
actual = query.to_bytes()
self.assertEqual(expected, actual)