示例#1
0
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
示例#2
0
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