Merge branch 'release/1.1.1'

Author: s3rius

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "taskiq-dependencies"
-version = "1.1.0"
+version = "1.1.1"
 description = "FastAPI like dependency injection implementation"
 authors = ["Pavel Kirilin <win10@list.ru>"]
 readme = "README.md"

taskiq_dependencies/graph.py

@@ -123,13 +123,13 @@ def _build_graph(self) -> None:  # noqa: C901, WPS210
 
                 # We check, that default value is an instance of
                 # TaskiqDepends.
-                if not isinstance(param.default, Dependency):
+                if not isinstance(default_value, Dependency):
                     continue
 
                 # If user haven't set the dependency,
                 # using TaskiqDepends constructor,
                 # we need to find variable's type hint.
-                if param.default.dependency is None:
+                if default_value.dependency is None:
                     if hints.get(param_name) is None:
                         # In this case, we don't know anything
                         # about this dependency. And it cannot be resolved.
@@ -150,14 +150,14 @@ def _build_graph(self) -> None:  # noqa: C901, WPS210
                 else:
                     # We can get dependency by simply using
                     # user supplied function.
-                    dependency_func = param.default.dependency
+                    dependency_func = default_value.dependency
 
                 # Now we construct new TaskiqDepends instance
                 # with correct dependency function and cache.
                 dep_obj = Dependency(
                     dependency_func,
-                    use_cache=param.default.use_cache,
-                    kwargs=param.default.kwargs,
+                    use_cache=default_value.use_cache,
+                    kwargs=default_value.kwargs,
                 )
                 # Also we set the parameter name,
                 # it will help us in future when

tests/test_fastapi.py

@@ -0,0 +1,31 @@
+from typing import Any
+from unittest.mock import patch
+
+from taskiq_dependencies import DependencyGraph
+
+
+class MyFastapiDepends:
+    def __init__(self, dependency: Any, use_cache: bool = False) -> None:
+        self.dependency = dependency
+        self.use_cache = use_cache
+
+
+def test_dependency_swap() -> None:
+    """
+    Test that dependency classes are swapped.
+
+    This test checks that if function depends on FastAPI depends, it will
+    be swapped and resolved.
+    """
+    with patch("taskiq_dependencies.graph.FastapiDepends", MyFastapiDepends):
+
+        def func_a() -> int:
+            return 1
+
+        def func_b(dep_a: int = MyFastapiDepends(func_a)) -> int:  # type: ignore
+            return dep_a
+
+        with DependencyGraph(func_b).sync_ctx() as ctx:
+            kwargs = ctx.resolve_kwargs()
+
+        assert kwargs == {"dep_a": 1}