def _get_bolt(self): """Returns Bolt protobuf message""" bolt = topology_pb2.Bolt() bolt.comp.CopyFrom(self._get_base_component()) # Add streams self._add_in_streams(bolt) self._add_out_streams(bolt) return bolt
def create_mock_bolt(self, bolt_name, input_streams, output_streams, bolt_parallelism): bolt = protoTopology.Bolt() bolt.comp.name = bolt_name kv = bolt.comp.config.kvs.add() kv.key = constants.TOPOLOGY_COMPONENT_PARALLELISM kv.value = str(bolt_parallelism) for stream in input_streams: bolt.inputs.add().stream.CopyFrom(stream) for stream in output_streams: bolt.outputs.add().stream.CopyFrom(stream) return bolt
def get_mock_bolt(component=get_mock_component(), inputs=[], outputs=[]): """Returns a mock protobuf Bolt object from topology_pb2""" bolt = topology_pb2.Bolt() bolt.comp.CopyFrom(component) for i in inputs: added = bolt.inputs.add() added.CopyFrom(i) for o in outputs: added = bolt.outputs.add() added.CopyFrom(o) return bolt