docs

docs/code_snippets/testing/di_handler_unittest.py

@@ -1,59 +1,42 @@
 import io
-import unittest
 from contextlib import redirect_stdout
-from unittest.mock import Mock
 
-from dishka import Provider, provide, make_container, Scope  # pyright: ignore[reportUnknownVariableType]
+from argenta.command import InputCommand
+from dishka import Provider, make_container, Scope
 
-from argenta import Router, Command
-from argenta.di.integration import inject, setup_dishka, FromDishka
-from argenta.response import Response
-from argenta.response.status import ResponseStatus
+from argenta import Router, Response
+from argenta.di.integration import setup_dishka, FromDishka
 
 
 class Service:
     def hello(self) -> str:
         return "world"
-
+        
+def get_service() -> Service:
+    return Service()
+        
 
 router = Router(title="DI")
 
-
-@router.command(Command("HELLO"))
-@inject  # Auto-inject dependencies from the Response container
+@router.command("HELLO")
 def hello(response: Response, service: FromDishka[Service]) -> None:
     print(f"hello {service.hello()}")
-
-
-class TestDIHandler(unittest.TestCase):
-    def test_hello_uses_service(self):
-        # Prepare DI container with a stub
-        fake = Mock(spec=Service)
-        fake.hello.return_value = "mocked"
-
-        class TestProvider(Provider):
-            scope = Scope.APP
-
-            @provide(scope=Scope.APP)
-            def service(self) -> Service:
-                return fake
-
-        container = make_container(TestProvider())
-
-        # Bind container to Response via integration
-        setup_dishka(app=_FakeApp(), container=container, auto_inject=False)  # type: ignore[arg-type]
-
-        # Create Response bound to the container
-        r = Response(ResponseStatus.ALL_FLAGS_VALID)
-        r._dishka_container = container  # direct assignment is acceptable in tests  # pyright: ignore[reportPrivateUsage]
-
-        # Call handler directly
-        with redirect_stdout(io.StringIO()) as stdout:
-            hello(r)
-
-        self.assertIn("hello mocked", stdout.getvalue())
-
-
+    
+    
 class _FakeApp:
     # Minimal stub for setup_dishka; app object is not used in unit tests
-    registered_routers = []
+    registered_routers = [router]
+
+
+def test_hello_uses_service():
+    provider = Provider(scope=Scope.APP)
+    provider.provide(get_service)
+    
+    container = make_container(provider)
+    setup_dishka(app=_FakeApp(), container=container, auto_inject=True)
+
+    # Call handler
+    with redirect_stdout(io.StringIO()) as stdout:
+        router.finds_appropriate_handler(InputCommand.parse('HELLO'))
+
+    assert "hello world" in stdout.getvalue()