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)