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
