def test_should_build_IPv6Packet_with_ICMP_payload_from_well_know_values_when_to_bytes_method_is_called(self):
# GIVEN
ipv6_packet = IPv6Packet(IPv6Header(source_address="fd00:1234:4555::ff:fe00:1800",
destination_address="ff03::1"),
ICMPv6(ICMPv6Header(128, 0),
ICMPv6EchoBody(0, 2, bytearray([0x80, 0x00, 0xc7, 0xbf, 0x00, 0x00, 0x00, 0x01,
0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41,
0x41, 0x41]))),
[HopByHop(options=[
HopByHopOption(HopByHopOptionHeader(_type=0x6d),
MPLOption(S=1, M=0, V=0, sequence=2, seed_id=bytearray([0x00, 0x18])))
])])
# WHEN
ipv6_packet_bytes = ipv6_packet.to_bytes()
# THEN
expected_ipv6_packet_bytes = bytearray([0x60, 0x00, 0x00, 0x00, 0x00, 0x22, 0x00, 0x40,
0xfd, 0x00, 0x12, 0x34, 0x45, 0x55, 0x00, 0x00,
0x00, 0x00, 0x00, 0xff, 0xfe, 0x00, 0x18, 0x00,
0xff, 0x03, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
0x3a, 0x00, 0x6d, 0x04, 0x40, 0x02, 0x00, 0x18,
0x80, 0x00, 0x87, 0x12, 0x00, 0x00, 0x00, 0x02,
0x80, 0x00, 0xc7, 0xbf, 0x00, 0x00, 0x00, 0x01,
0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41,
0x41, 0x41])
self.assertEqual(expected_ipv6_packet_bytes, ipv6_packet_bytes)
def test_convert_ICMPv6_echo_body_to_data_when_to_bytes_method_is_called(self):
# GIVEN
identifier = any_identifier()
sequence_number = any_sequence_number()
data = any_data()
body = ICMPv6EchoBody(identifier, sequence_number, data)
# WHEN
actual = body.to_bytes()
# THEN
expected = bytearray([identifier >> 8, identifier & 0xFF, sequence_number >> 8, sequence_number & 0xFF]) + data
self.assertEqual(expected, actual)
def test_should_build_ICMPv6EchoBody_from_well_know_values_when_to_bytes_method_is_called(self):
# GIVEN
body = ICMPv6EchoBody(0, 2, bytearray([0x80, 0x00, 0xc7, 0xbf, 0x00, 0x00, 0x00, 0x01,
0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41, 0x41,
0x41, 0x41]))
# WHEN
actual = body.to_bytes()
# THEN
expected = bytearray([0x00, 0x00, 0x00, 0x02, 0x80, 0x00, 0xc7, 0xbf,
0x00, 0x00, 0x00, 0x01, 0x41, 0x41, 0x41, 0x41,
0x41, 0x41, 0x41, 0x41, 0x41, 0x41])
self.assertEqual(expected, actual)
def test_should_create_ICMPv6EchoBody_from_data_when_from_bytes_classmethod_is_called(
self):
# GIVEN
identifier = any_identifier()
sequence_number = any_sequence_number()
body_data = any_data()
data = bytearray([(identifier >> 8), (identifier & 0xFF),
(sequence_number >> 8), (sequence_number & 0xFF)])
data += body_data
# WHEN
actual = ICMPv6EchoBody.from_bytes(io.BytesIO(data))
# THEN
self.assertEqual(identifier, actual.identifier)
self.assertEqual(sequence_number, actual.sequence_number)
self.assertEqual(body_data, actual.data)
def test_should_create_ICMPv6EchoBody_from_data_when_from_bytes_classmethod_is_called(self):
# GIVEN
identifier = any_identifier()
sequence_number = any_sequence_number()
body_data = any_data()
data = bytearray([(identifier >> 8),
(identifier & 0xFF),
(sequence_number >> 8),
(sequence_number & 0xFF)])
data += body_data
# WHEN
actual = ICMPv6EchoBody.from_bytes(io.BytesIO(data))
# THEN
self.assertEqual(identifier, actual.identifier)
self.assertEqual(sequence_number, actual.sequence_number)
self.assertEqual(body_data, actual.data)