def create_mock_simple_physical_plan(self, spout_parallelism=1, bolt_parallelism=1): pplan = protoPPlan.PhysicalPlan() pplan.topology.CopyFrom( self.create_mock_simple_topology(spout_parallelism, bolt_parallelism)) return pplan
def create_mock_simple_physical_plan( self, spout_parallelism=1, bolt_parallelism=1, instances_num=1): pplan = protoPPlan.PhysicalPlan() pplan.topology.CopyFrom(self.create_mock_simple_topology( spout_parallelism, bolt_parallelism)) pplan.stmgrs.extend([self.create_mock_stream_manager()]) instances = [] for i in range(instances_num): instances.append(self.create_mock_instance(i+1)) pplan.instances.extend(instances) return pplan
def get_mock_pplan(topology=get_mock_topology(), stmgrs=[], instances=[]): """Returns a mock protobuf PhysicalPlan object from physical_plan_pb2""" pplan = physical_plan_pb2.PhysicalPlan() pplan.topology.MergeFrom(topology) if len(stmgrs) == 0: stmgrs.append(get_mock_stmgr()) if len(instances) == 0: instances.append(get_mock_instance()) for stmgr in stmgrs: added = pplan.stmgrs.add() added.CopyFrom(stmgr) for instance in instances: added = pplan.instances.add() added.CopyFrom(instance) return pplan
def get_pplan_builder_and_typename(): """Returns a PhysicalPlan builder callable and typename 'PhysicalPlan'""" # pylint: disable=unnecessary-lambda builder = lambda: physical_plan_pb2.PhysicalPlan() typename = builder().DESCRIPTOR.full_name return builder, typename