async def test_internal_queue_get_delay():
queue = InternalQueue()
task = asyncio.create_task(queue.get())
await asyncio.sleep(0.001)
queue.put_nowait(True)
await asyncio.sleep(0.001)
assert task.done()
assert task.result() is True
async def test_internal_queue_put_delay():
queue = InternalQueue(maxsize=1)
queue.put_nowait(False)
task = asyncio.create_task(queue.put(True))
await asyncio.sleep(0.001)
assert queue.get_nowait() is False
await asyncio.sleep(0.001)
assert task.done()
assert queue.get_nowait() is True
def test_queue_mock_context_sync():
queue = InternalQueue()
with QueueMockContext(queue) as m:
queue.put_nowait(1)
queue.put_nowait(2)
queue.get_nowait()
assert m.put_items == [1, 2]
assert m.got_items == [1]
async def test_internal_queue_put_nowait_full():
queue = InternalQueue(maxsize=1)
queue.put_nowait(True)
with pytest.raises(QueueFull):
queue.put_nowait(True)
async def test_internal_queue_put_nowait():
queue = InternalQueue()
queue.put_nowait(True)