示例#1
0
def test_consume(event_loop, test_consumer, cancelled_future):
    """Test Application._consume."""
    queue = asyncio.Queue(maxsize=1)

    app = Application('testing', consumer=test_consumer)

    asyncio.async(app._consume(queue, cancelled_future))

    event_loop.stop()  # Run the event loop once.
    event_loop.run_forever()

    # The size of the queue won't ever be larger than 1 because of the
    # maxsize argument.
    assert queue.qsize() == 1
示例#2
0
def test_consume(event_loop, test_consumer, ASYNC_QUEUE):
    """Test Application._consume."""
    if ASYNC_QUEUE:
        queue = asyncio.Queue(maxsize=1)
    else:
        queue = sync_queue.Queue(maxsize=1)

    app = Application('testing',
                      settings={"ASYNC_QUEUE": ASYNC_QUEUE},
                      consumer=test_consumer)

    asyncio.ensure_future(app._consume(queue), loop=event_loop)

    event_loop.stop()  # Run the event loop once.
    event_loop.run_forever()

    # The size of the queue won't ever be larger than 1 because of the
    # maxsize argument.
    assert queue.qsize() == 1