From aaf98782665ba92fc60801f8d9e5dfafd9a7d3be Mon Sep 17 00:00:00 2001
From: AndrianEquestrian <andreymaksiuta@gmail.com>
Date: Sun, 15 Jun 2025 17:50:41 +0300
Subject: [PATCH] fixup! Add support for Fast Stream Depends

---
 .../unit/samples/wiringfastdepends/sample.py  | 17 +++++++----
 tests/unit/wiring/test_fastdepends.py         | 28 ++-----------------
 2 files changed, 13 insertions(+), 32 deletions(-)

diff --git a/tests/unit/samples/wiringfastdepends/sample.py b/tests/unit/samples/wiringfastdepends/sample.py
index 96fe2abf..da58e8f2 100644
--- a/tests/unit/samples/wiringfastdepends/sample.py
+++ b/tests/unit/samples/wiringfastdepends/sample.py
@@ -1,10 +1,19 @@
 import sys
 
+from dependency_injector import containers, providers
 from dependency_injector.wiring import inject, Provide
 from fast_depends import Depends
-from wiringfastdepends.sample import CoefficientService, Container
 
 
+class CoefficientService:
+    @staticmethod
+    def get_coefficient() -> float:
+        return 1.2
+
+
+class Container(containers.DeclarativeContainer):
+    service = providers.Factory(CoefficientService)
+
 
 @inject
 def apply_coefficient(
@@ -15,8 +24,4 @@ def apply_coefficient(
 
 
 container = Container()
-container.wire(modules=[sys.modules[__name__]])
-
-
-def test_wire_positive() -> None:
-    assert apply_coefficient(100) == 120.0
\ No newline at end of file
+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 ee7f2374..32118e59 100644
--- a/tests/unit/wiring/test_fastdepends.py
+++ b/tests/unit/wiring/test_fastdepends.py
@@ -1,31 +1,7 @@
-
-import sys
-
-from dependency_injector import containers, providers
 from dependency_injector.wiring import inject, Provide
-from fast_depends import Depends
 
+from wiringfastdepends import sample
 
-class CoefficientService:
-    @staticmethod
-    def get_coefficient() -> float:
-        return 1.2
-
-
-class Container(containers.DeclarativeContainer):
-    service = providers.Factory(CoefficientService)
-
-
-@inject
-def apply_coefficient(
-    a: int,
-    coefficient_provider: CoefficientService = Depends(Provide[Container.service]),
-) -> float:
-    return a * coefficient_provider.get_coefficient()
-
-
-container = Container()
-container.wire(modules=[sys.modules[__name__]])
 
 def test_apply_coefficient():
-    assert apply_coefficient(100) == 120.0
\ No newline at end of file
+    assert sample.apply_coefficient(100) == 120.0
\ No newline at end of file