fix: Don't reset state of generator on each invocation

Author: Zhenay

fast_depends/use.py

@@ -199,7 +199,6 @@ def __init__(
         self.overrides = overrides
 
     def __aiter__(self) -> "solve_async_gen":
-        self._iter = None
         self.stack = AsyncExitStack()
         return self
 
@@ -246,7 +245,6 @@ def __init__(
         self.overrides = overrides
 
     def __iter__(self) -> "solve_gen":
-        self._iter = None
         self.stack = ExitStack()
         return self
 

tests/async/test_depends.py

@@ -2,6 +2,7 @@
 from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from functools import partial
+from typing import AsyncGenerator
 from unittest.mock import Mock
 
 import pytest
@@ -484,3 +485,21 @@ async def func(a: str, b: str = Depends(dep)):
 
     async with func("a") as is_equal:
         assert is_equal
+
+
+@pytest.mark.anyio
+async def test_asyncgenerator_iter():
+    # ref: https://github.com/Lancetnik/FastDepends/issues/165
+
+    async def simple_dependency(a: int, b: int = 3):
+        return a + b
+
+    @inject
+    async def method(a: int, d: int = Depends(simple_dependency)) -> AsyncGenerator[int, None]:
+        for i in range(a + d):
+            yield i
+
+    iterator = method(5)
+
+    assert len([v async for v in iterator]) == 13
+    assert len([v async for v in iterator]) == 0

tests/sync/test_depends.py

@@ -2,6 +2,7 @@
 from contextlib import contextmanager
 from dataclasses import dataclass
 from functools import partial
+from typing import Generator
 from unittest.mock import Mock
 
 import pytest
@@ -366,3 +367,19 @@ def func(a: str, b: str = Depends(dep)):
 
     with func("a") as is_equal:
         assert is_equal
+
+
+def test_generator_iter():
+    # ref: https://github.com/Lancetnik/FastDepends/issues/165
+
+    def simple_dependency(a: int, b: int = 3):
+        return a + b
+
+    @inject
+    def method(a: int, d: int = Depends(simple_dependency)) -> Generator[int, None, None]:
+        yield from range(a + d)
+
+    iterator = method(5)
+
+    assert len(list(iterator)) == 13
+    assert len(list(iterator)) == 0