def test_not_read_zmqlet(): with MockBasePeaNotRead(args3), Zmqlet(args1, default_logger) as z: req = Request() req.request_id = random_identity() d = req.data.docs.add() d.tags['id'] = 2 msg = Message(None, req, 'tmp', '') z.send_message(msg)
def test_recv_message_zmqlet(mocker): zmqlet1 = Zmqlet(args1, default_logger) zmqlet2 = Zmqlet(args2, default_logger) req = Request() req.request_id = random_identity() doc = req.data.docs.add() doc.tags['id'] = 2 msg = Message(None, req, 'tmp', '') def callback(msg_): assert msg_.request.docs[0].tags['id'] == msg.request.data.docs[0].tags['id'] mock = mocker.Mock() zmqlet1.send_message(msg) time.sleep(1) zmqlet2.recv_message(mock) validate_callback(mock, callback)
def test_not_decompressed_zmqlet(mocker): with MockPea(args2) as pea, Zmqlet(args1, default_logger) as z: req = Request() req.request_id = random_identity() d = req.data.docs.add() d.tags['id'] = 2 msg = Message(None, req, 'tmp', '') mock = mocker.Mock() z.send_message(msg) time.sleep(1) z.recv_message(mock) def callback(msg_): pass validate_callback(mock, callback) print(f' joining pea') pea.join() print(f' joined pea')