def _register_msg_to_handle(self): # pylint: disable=unnecessary-lambda new_instance_builder = lambda: stmgr_pb2.NewInstanceAssignmentMessage() tuple_msg_builder = lambda: stmgr_pb2.TupleMessage() hts2_msg_builder = lambda: tuple_pb2.HeronTupleSet2() self.register_on_message(new_instance_builder) self.register_on_message(tuple_msg_builder) self.register_on_message(hts2_msg_builder)
def _register_msg_to_handle(self): # pylint: disable=unnecessary-lambda new_instance_builder = lambda: stmgr_pb2.NewInstanceAssignmentMessage() hts2_msg_builder = lambda: tuple_pb2.HeronTupleSet2() stateful_start_msg_builder = lambda: ckptmgr_pb2.StartInstanceStatefulProcessing() stateful_restore_msg_builder = lambda: ckptmgr_pb2.RestoreInstanceStateRequest() stateful_initiate_msg_builder = lambda: ckptmgr_pb2.InitiateStatefulCheckpoint() self.register_on_message(new_instance_builder) self.register_on_message(hts2_msg_builder) self.register_on_message(stateful_start_msg_builder) self.register_on_message(stateful_restore_msg_builder) self.register_on_message(stateful_initiate_msg_builder)
def get_mock_assignment_message(pplan=get_mock_pplan()):
"""Returns a mock protobuf NewInstanceAssignmentMessage object from stmgr_pb2"""
# message
mock_message = stmgr_pb2.NewInstanceAssignmentMessage()
mock_message.pplan.MergeFrom(pplan)
return mock_message
