def get_a_mock_message_list_and_builder(): """Get a list of IncomingPackets, the corresponding mock messages, builder and typename""" pkt_list = [] raw_list = mock_protobuf.get_many_mock_pplans() for msg in raw_list: reqid = REQID.generate_zero() pkt = convert_to_incoming_packet(reqid, msg) pkt_list.append(pkt) builder, typename = mock_protobuf.get_pplan_builder_and_typename() return pkt_list, raw_list, builder, typename
def get_a_mock_message_list_and_builder(): """Get a list of IncomingPackets, the corresponding mock messages, builder and typename""" pkt_list = [] raw_list = mock_protobuf.get_many_mock_pplans() for msg in raw_list: reqid = REQID.generate_zero() pkt = convert_to_incoming_packet(reqid, msg) pkt_list.append(pkt) builder, typename = mock_protobuf.get_pplan_builder_and_typename() return pkt_list, raw_list, builder, typename
def test_register_on_message(self): # try registering with PhysicalPlan as a type of message to receive builder, typename = mock_protobuf.get_pplan_builder_and_typename() self.mock_client.register_on_message(builder) self.assertTrue(typename in self.mock_client.registered_message_map)
def test_register_on_message(self): # try registering with PhysicalPlan as a type of message to receive builder, typename = mock_protobuf.get_pplan_builder_and_typename() self.mock_client.register_on_message(builder) self.assertTrue(typename in self.mock_client.registered_message_map)