diff --git a/tests/unit/samples/wiringfastdepends/sample.py b/tests/unit/samples/wiringfastdepends/sample.py index da58e8f2..28310052 100644 --- a/tests/unit/samples/wiringfastdepends/sample.py +++ b/tests/unit/samples/wiringfastdepends/sample.py @@ -3,6 +3,7 @@ import sys from dependency_injector import containers, providers from dependency_injector.wiring import inject, Provide from fast_depends import Depends +from typing_extensions import Annotated class CoefficientService: @@ -23,5 +24,13 @@ def apply_coefficient( return a * coefficient_provider.get_coefficient() +@inject +def apply_coefficient_annotated( + a: int, + coefficient_provider: Annotated[CoefficientService, Depends(Provide[Container.service])], +) -> float: + return a * coefficient_provider.get_coefficient() + + container = Container() container.wire(modules=[sys.modules[__name__]]) \ No newline at end of file diff --git a/tests/unit/wiring/test_fastdepends.py b/tests/unit/wiring/test_fastdepends.py index 32118e59..2c326143 100644 --- a/tests/unit/wiring/test_fastdepends.py +++ b/tests/unit/wiring/test_fastdepends.py @@ -3,5 +3,9 @@ from dependency_injector.wiring import inject, Provide from wiringfastdepends import sample -def test_apply_coefficient(): - assert sample.apply_coefficient(100) == 120.0 \ No newline at end of file +def test_apply_coefficient() -> None: + assert sample.apply_coefficient(100) == 120.0 + + +def test_apply_coefficient_annotated() -> None: + assert sample.apply_coefficient_annotated(100) == 120.0 \ No newline at end of file