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
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