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
