def test_simple_zmqlet(): args = set_pea_parser().parse_args([ '--host-in', '0.0.0.0', '--host-out', '0.0.0.0', '--port-in', '12346', '--port-out', '12347', '--socket-in', 'PULL_CONNECT', '--socket-out', 'PUSH_CONNECT', '--timeout-ctrl', '-1']) args2 = set_pea_parser().parse_args([ '--host-in', '0.0.0.0', '--host-out', '0.0.0.0', '--port-in', '12347', '--port-out', '12346', '--socket-in', 'PULL_BIND', '--socket-out', 'PUSH_BIND', '--uses', '_logforward', '--timeout-ctrl', '-1' ]) logger = logging.getLogger('zmq-test') with BasePea(args2) as z1, Zmqlet(args, logger) as z: req = jina_pb2.Request() req.request_id = uuid.uuid1().hex d = req.index.docs.add() d.tags['id'] = 2 msg = add_envelope(req, 'tmp', '') z.send_message(msg)
def test_simple_zmqlet(self): args = set_pea_parser().parse_args([ '--host-in', '0.0.0.0', '--host-out', '0.0.0.0', '--port-in', '12346', '--port-out', '12347', '--socket-in', 'PULL_CONNECT', '--socket-out', 'PUSH_CONNECT', '--timeout-ctrl', '-1']) args2 = set_pea_parser().parse_args([ '--host-in', '0.0.0.0', '--host-out', '0.0.0.0', '--port-in', '12347', '--port-out', '12346', '--socket-in', 'PULL_BIND', '--socket-out', 'PUSH_BIND', '--uses', '_logforward', '--timeout-ctrl', '-1' ]) with BasePea(args2) as z1, Zmqlet(args) as z: req = jina_pb2.Request() req.request_id = 1 d = req.index.docs.add() d.id = 2 msg = add_envelope(req, 'tmp', '') z.send_message(msg)
def test_add_route(): r = jina_pb2.Request() r.control.command = jina_pb2.Request.ControlRequest.IDLE msg = ProtoMessage(None, r, pod_name='test1', identity='sda') msg.add_route('name', 'identity') assert len(msg.envelope.routes) == 2 assert msg.envelope.routes[1].pod == 'name' assert msg.envelope.routes[1].pod_id == 'identity'
def test_not_read_zmqlet(): with MockBasePeaNotRead(args3), Zmqlet(args1, default_logger) as z: req = jina_pb2.Request() req.request_id = uuid.uuid1().hex d = req.index.docs.add() d.tags['id'] = 2 msg = ProtoMessage(None, req, 'tmp', '') z.send_message(msg)
def get_prev_reqs(): # three requests, each with the SAME doc, but diff evaluations result = [] for j in range(3): r = jina_pb2.Request() d = r.index.docs.add() d.id = 'SAME DOC THEY ARE' # same doc id ev1 = d.evaluations.add() ev1.value = j # diff eval ev1.op_name = f'op{j}' # diff eval result.append(r.index) return result