add get method

aiomisc/iterator_wrapper.py

@@ -59,6 +59,8 @@ def __exit__(
         self.close()
 
     def put(self, item: Any) -> None:
+        if self.is_closed:
+            raise ChannelClosed
         self.queue.append(item)
 
     def __await__(self) -> Any:
@@ -70,6 +72,9 @@ def __await__(self) -> Any:
 
         return self.queue.popleft()
 
+    async def get(self) -> Any:
+        return await self
+
 
 class IteratorWrapperStatistic(Statistic):
     started: int
@@ -167,7 +172,7 @@ def __aiter__(self) -> AsyncIterator[Any]:
 
     async def __anext__(self) -> Awaitable[T]:
         try:
-            item, is_exc = await self.__channel
+            item, is_exc = await self.__channel.get()
         except ChannelClosed:
             await self.wait_closed()
             raise StopAsyncIteration

tests/test_thread_pool.py

@@ -41,8 +41,8 @@ def executor(loop: asyncio.AbstractEventLoop):
         thread_pool.shutdown(wait=True)
 
 
-async def test_from_thread_channel(loop, threaded_decorator):
-    channel = FromThreadChannel(maxsize=2, loop=loop)
+async def test_from_thread_channel(threaded_decorator):
+    channel = FromThreadChannel(maxsize=2)
 
     @threaded_decorator
     def in_thread():
@@ -60,7 +60,7 @@ def in_thread():
 
 
 async def test_from_thread_channel_wait_before(loop, threaded_decorator):
-    channel = FromThreadChannel(maxsize=1, loop=loop)
+    channel = FromThreadChannel(maxsize=1)
 
     @threaded_decorator
     def in_thread():
@@ -79,14 +79,14 @@ def in_thread():
 
 
 async def test_from_thread_channel_close(loop):
-    channel = FromThreadChannel(maxsize=1, loop=loop)
+    channel = FromThreadChannel(maxsize=1)
     with channel:
         channel.put(1)
 
     with pytest.raises(ChannelClosed):
         channel.put(2)
 
-    channel = FromThreadChannel(maxsize=1, loop=loop)
+    channel = FromThreadChannel(maxsize=1)
     task = loop.create_task(channel.get())
 
     with pytest.raises(asyncio.TimeoutError):