added Try
All checks were successful
CI / build (push) Successful in 18s

This commit is contained in:
2024-11-03 22:39:08 +08:00
parent e0e763170f
commit 6ef7f8107e
10 changed files with 338 additions and 188 deletions

View File

@@ -1,6 +1,7 @@
import unittest
from src.pwo import retry, async_retry, async_test
from pwo import retry, async_retry, async_test, AsyncQueueIterator, aenumerate
from asyncio import Queue
class PrivateTest(unittest.TestCase):
@@ -70,5 +71,25 @@ class PrivateTest(unittest.TestCase):
await bar()
self.assertEqual(max_attempts, attempt)
if __name__ == '__main__':
unittest.main()
@async_test
async def test_async_queue_iterator(self):
queue = Queue()
queue_size = 10
objects = [object() for _ in range(queue_size)]
async def poll() -> int:
completed = 0
async for i, obj in aenumerate(AsyncQueueIterator(queue)):
self.assertIs(objects[i], obj)
completed += 1
return completed
handle = poll()
for o in objects:
queue.put_nowait(o)
queue.put_nowait(None)
processed = await handle
self.assertEqual(queue_size, processed)