def test_should_convert_HopByHopOptionHeader_to_bytes_when_to_bytes_method_is_called(self): # GIVEN _type = any_type() length = any_length() hop_by_hop_option_header = HopByHopOptionHeader(_type, length) # WHEN data = hop_by_hop_option_header.to_bytes() # THEN expected_data = bytearray([_type, length]) self.assertEqual(expected_data, data)
def test_should_convert_HopByHopOptionHeader_to_bytes_when_to_bytes_method_is_called(self): # GIVEN _type = any_type() length = any_length() hop_by_hop_option_header = HopByHopOptionHeader(_type, length) # WHEN data = hop_by_hop_option_header.to_bytes() # THEN expected_data = bytearray([_type, length]) self.assertEqual(expected_data, data)
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 any_hop_by_hop_bytes_option_header(length=4): _type = any_type() # 0 or 1 means padding, so type have to be higher than 1 while _type <= 1: _type = any_type() return HopByHopOptionHeader(_type, length)
def test_should_create_HopByHopOptionHeader_when_to_bytes_method_is_called_with_data(self): # GIVEN _type = any_type() length = any_length() data = bytearray([_type, length]) # WHEN option_header = HopByHopOptionHeader.from_bytes(io.BytesIO(data)) # THEN self.assertEqual(_type, option_header.type) self.assertEqual(length, option_header.length)
def test_should_return_proper_length_when_len_is_called_with_HopByHopOptionHeader_object(self): # GIVEN _type = any_type() length = any_length() option_header = HopByHopOptionHeader(_type, length) # WHEN option_header_length = len(option_header) # THEN expected_option_header_length = 2 self.assertEqual(expected_option_header_length, option_header_length)
def test_should_create_HopByHopOptionHeader_when_to_bytes_method_is_called_with_data(self): # GIVEN _type = any_type() length = any_length() data = bytearray([_type, length]) # WHEN option_header = HopByHopOptionHeader.from_bytes(io.BytesIO(data)) # THEN self.assertEqual(_type, option_header.type) self.assertEqual(length, option_header.length)
def any_hop_by_hop_bytes_option_header(length=4): return HopByHopOptionHeader(any_type(), length)