async def test_add_msg_w_response(loop): app = BaseApplication('app', loop=loop, CONFIG='config.tests') qp = AsyncQueueProducer(app) qp.add_msg_to_queue = Mock( side_effect=coroutine(lambda q, m: "24d8f597g8utr")) qp.get_message_from_queue = Mock( side_effect=coroutine(lambda cid: "response")) await qp.add_msg_to_queue_with_response('queue1', "qweqwe") assert qp.add_msg_to_queue.called assert qp.get_message_from_queue.called
async def test_add_msg_w_response(loop): app = BaseApplication('app', loop=loop, CONFIG='config.tests') qp = AsyncQueueProducer(app) qp.add_msg_to_queue = Mock(side_effect=coroutine( lambda q, m: "24d8f597g8utr" )) qp.get_message_from_queue = Mock(side_effect=coroutine( lambda cid: "response" )) await qp.add_msg_to_queue_with_response('queue1', "qweqwe") assert qp.add_msg_to_queue.called assert qp.get_message_from_queue.called