def _get_spout(self):
"""Returns Spout protobuf message"""
spout = topology_pb2.Spout()
spout.comp.CopyFrom(self._get_base_component())
# Add output streams
self._add_out_streams(spout)
return spout
def create_mock_spout(self, spout_name, output_streams, spout_parallelism):
spout = protoTopology.Spout()
spout.comp.name = spout_name
kv = spout.comp.config.kvs.add()
kv.key = constants.TOPOLOGY_COMPONENT_PARALLELISM
kv.value = str(spout_parallelism)
for stream in output_streams:
spout.outputs.add().stream.CopyFrom(stream)
return spout
def get_mock_spout(component=get_mock_component(), outputs=[]):
"""Returns a mock protobuf Spout object from topology_pb2"""
spout = topology_pb2.Spout()
spout.comp.CopyFrom(component)
for out in outputs:
added = spout.outputs.add()
added.CopyFrom(out)
return spout