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)
