def test_should_create_tlv_when_parse_method_is_called(self): # GIVEN class DummyNetworkDataTlvsFactory: def parse(self, data, message_info): return bytearray(data.read()) tlv = any_tlv_data() factory = network_layer.NetworkLayerTlvsFactory( {tlv[0]: DummyNetworkDataTlvsFactory()}) # WHEN actual_tlvs = factory.parse(io.BytesIO(tlv), None) # THEN self.assertEqual(tlv[2:], actual_tlvs[0])
def create_default_network_tlvs_factory(): return network_layer.NetworkLayerTlvsFactory( tlvs_factories=create_deafult_network_tlvs_factories())