refactor tests and add new

tests/unit_tests/test_di.py

@@ -0,0 +1,81 @@
+from typing import Generator
+from argenta import App, DataBridge, Router
+from argenta.di.providers import SystemProvider
+from argenta.orchestrator.argparser import ArgParser, ArgSpace
+from argenta.response import ResponseStatus
+from dishka import Container, make_container
+import pytest
+from argenta.response.entity import Response
+from argenta.di.integration import FromDishka, _get_container_from_response, setup_dishka, _auto_inject_handlers
+
+
+@pytest.fixture
+def argparser() -> ArgParser:
+    return ArgParser(processed_args=[])
+
+@pytest.fixture
+def container(argparser: ArgParser) -> Generator[Container]:
+    container = make_container(SystemProvider(), context={ArgParser: argparser})
+    yield container
+    container.close()
+
+
+def test_get_container_from_response(container: Container):
+    Response.patch_by_container(container)
+    response = Response(ResponseStatus.ALL_FLAGS_VALID)
+    assert _get_container_from_response((response,), {}) == container
+    
+def test_get_container_from_response4(container: Container):
+    Response.patch_by_container(container)
+    response = Response(ResponseStatus.ALL_FLAGS_VALID)
+    assert _get_container_from_response((object(), response,), {}) == container
+    
+def test_get_container_from_response2(container: Container):
+    delattr(Response, '__dishka_container__')
+    response = Response(ResponseStatus.ALL_FLAGS_VALID)
+    with pytest.raises(RuntimeError):
+        _get_container_from_response((response,), {})
+        
+def test_get_container_from_response3(container: Container):
+    Response.patch_by_container(container)
+    with pytest.raises(RuntimeError):
+        assert _get_container_from_response((), {}) == container
+        
+def test_setup_dishka(container: Container):
+    app = App()
+    router = Router()
+    
+    @router.command('command')
+    def handler(res: Response, data_bridge: FromDishka[DataBridge]):
+        print(data_bridge)
+        
+    app.include_router(router)
+    
+    assert setup_dishka(app, container, auto_inject=True) is None
+    
+def test_setup_dishka2(container: Container):
+    app = App()
+    assert setup_dishka(app, container, auto_inject=False) is None
+    
+def test_auto_inject_handlers(container: Container):
+    Response.patch_by_container(container)
+    
+    app = App()
+    router = Router()
+    
+    @router.command('command')
+    def handler(res: Response, data_bridge: FromDishka[DataBridge]):
+        print(data_bridge)
+        
+    app.include_router(router)
+        
+    _auto_inject_handlers(app)
+    _auto_inject_handlers(app) # check idempotency
+    
+def test_get_from_container(container: Container):
+    assert isinstance(container.get(ArgSpace), ArgSpace)
+    
+def test_get_from_container2(container: Container):
+    assert isinstance(container.get(DataBridge), DataBridge)
+    
+