async def test_connect(loop): app = BaseApplication('app', loop=loop, CONFIG='config.tests') qp = AsyncQueueProducer(app) qp.make_connection = Mock(side_effect=coroutine( lambda queue_name: Mock(channel=Mock(side_effect=coroutine(lambda: None))) )) qp.create_queue_and_consume = Mock(side_effect=coroutine( lambda: ("queue_name", "consumer_tag") )) qp.create_queue = Mock(side_effect=coroutine( lambda queue_name: {"queue": queue_name, "message_count": 0, "consumer_count": 0} )) await qp.connect() assert qp.make_connection.called assert qp.create_queue_and_consume.called assert qp.create_queue.called
async def test_connect(loop): app = BaseApplication('app', loop=loop, CONFIG='config.tests') qp = AsyncQueueProducer(app) qp.make_connection = Mock(side_effect=coroutine(lambda queue_name: Mock( channel=Mock(side_effect=coroutine(lambda: None))))) qp.create_queue_and_consume = Mock( side_effect=coroutine(lambda: ("queue_name", "consumer_tag"))) qp.create_queue = Mock( side_effect=coroutine(lambda queue_name: { "queue": queue_name, "message_count": 0, "consumer_count": 0 })) await qp.connect() assert qp.make_connection.called assert qp.create_queue_and_consume.called assert qp.create_queue.called