fix: Allow to inject async functions into functions wrapped by `conte…

…xtlib.asynccontextmanager`

fast_depends/use.py

@@ -184,7 +184,7 @@ def injected_wrapper(*args: P.args, **kwargs: P.kwargs) -> T:
 
 
 class solve_async_gen:
-    _iter: Optional[AsyncIterator[Any]]
+    _iter: Optional[AsyncIterator[Any]] = None
 
     def __init__(
         self,
@@ -231,7 +231,7 @@ async def __anext__(self) -> Any:
 
 
 class solve_gen:
-    _iter: Optional[Iterator[Any]]
+    _iter: Optional[Iterator[Any]] = None
 
     def __init__(
         self,

tests/async/test_depends.py

@@ -1,4 +1,5 @@
 import logging
+from contextlib import asynccontextmanager
 from dataclasses import dataclass
 from functools import partial
 from unittest.mock import Mock
@@ -469,3 +470,17 @@ async def func(a=Depends(dep)):
         return a
 
     assert await func() == "a"
+
+
+@pytest.mark.anyio
+async def test_asynccontextmanager():
+    async def dep(a: str = "a"):
+        return a
+
+    @asynccontextmanager
+    @inject
+    async def func(a: str, b: str = Depends(dep)):
+        yield a == b
+
+    async with func("a") as is_equal:
+        assert is_equal

tests/sync/test_depends.py

@@ -1,4 +1,5 @@
 import logging
+from contextlib import contextmanager
 from dataclasses import dataclass
 from functools import partial
 from unittest.mock import Mock
@@ -352,3 +353,16 @@ def func(a=Depends(dep)):
         return a
 
     assert func() == "a"
+
+
+def test_contextmanager():
+    def dep(a: str = "a"):
+        return a
+
+    @contextmanager
+    @inject
+    def func(a: str, b: str = Depends(dep)):
+        yield a == b
+
+    with func("a") as is_equal:
+        assert is_equal