refactor tests and add new

tests/unit_tests/test_argparser.py

@@ -245,3 +245,11 @@ def test_argparser_parse_args_populates_argspace(
     assert debug_arg is not None
     assert debug_arg.value is True
     assert debug_arg.founder_class is BooleanArgument
+    
+def test_str_input_argument():
+    arg = InputArgument('host', value='192.168.0.0', founder_class=ValueArgument)
+    assert str(arg) == 'InputArgument(host=192.168.0.0)'
+    
+def test_repr_input_argument():
+    arg = InputArgument('host', value='192.168.0.0', founder_class=ValueArgument)
+    assert repr(arg) == "InputArgument<name=host, value=192.168.0.0, founder_class=ValueArgument>" 

tests/unit_tests/test_command.py

@@ -23,6 +23,18 @@ def test_parse_raw_command_without_flag_name_with_value():
 def test_parse_raw_command_with_repeated_flag_name():
     with pytest.raises(RepeatedInputFlagsException):
         InputCommand.parse('ssh --host 192.168.0.3 --host 172.198.0.43')
+        
+        
+def test_parse_raw_command_with_triple_prefix():
+    assert InputCommand.parse(
+        'ssh ---host 192.168.0.0'
+    ).input_flags.get_flag_by_name('host') == \
+    InputFlag('host', input_value='192.168.0.0', prefix='---')
+    
+    
+def test_parse_raw_command_with_unprocessed_entity():
+    with pytest.raises(UnprocessedInputFlagException):
+        InputCommand.parse('ssh --host 192.168.0.3 9977')
 
 
 def test_parse_empty_raw_command():

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)
+    
+    

tests/unit_tests/test_flag.py

@@ -1,7 +1,9 @@
 import re
+from sys import flags
 
 from argenta.command.flag import Flag, InputFlag, PossibleValues
 from argenta.command.flag.flags import Flags, InputFlags
+import pytest
 
 
 def test_get_string_entity():
@@ -114,3 +116,112 @@ def test_add_flags():
     flags = Flags()
     flags.add_flags([Flag('test'), Flag('test2')])
     assert len(flags.flags) == 2
+    
+def test_eq_flags():
+    flags = Flags([Flag('some')])
+    flags2 = Flags([Flag('some')])
+    assert flags == flags2
+    
+def test_contains_flags():
+    flags = Flags([Flag('some')])
+    flag = Flag('some')
+    assert flag in flags
+    
+def test_eq_flags2():
+    flags = Flags([Flag('some')])
+    flags2 = Flags([Flag('other')])
+    assert flags != flags2
+    
+def test_eq_flags3():
+    flags = Flags([Flag('some')])
+    flags2 = Flags([Flag('some'), Flag('other')])
+    assert flags != flags2
+    
+def test_eq_flags4():
+    flags = Flags([Flag('some')])
+    not_flags = object()
+    assert flags != not_flags
+    
+def test_contains_flags2():
+    flags = Flags([Flag('some')])
+    flag = Flag('nonexists')
+    assert flag not in flags
+    
+def test_contains_flags3():
+    flags = Flags([Flag('some')])
+    not_flag = object
+    with pytest.raises(TypeError):
+        not_flag in flags  # pyright: ignore[reportUnusedExpression]
+        
+def test_get_flag_by_name():
+    flags = Flags([Flag('some')])
+    assert flags.get_flag_by_name('some') == Flag('some')
+    
+def test_eq_input_flags3():
+    flags = InputFlags([InputFlag('some', input_value='')])
+    flags2 = InputFlags([
+        InputFlag('some', input_value=''), 
+        InputFlag('some2', input_value='')
+    ])
+    assert flags != flags2
+    
+def test_eq_input_flags4():
+    flags = InputFlags([InputFlag('some', input_value='')])
+    not_flags = object()
+    assert flags != not_flags
+    
+def test_contains_input_flags2():
+    flags = InputFlags([InputFlag('some', input_value='')])
+    flag = InputFlag('nonexists', input_value='')
+    assert flag not in flags
+    
+def test_contains_input_flags3():
+    flags = InputFlags([InputFlag('some', input_value='')])
+    not_flag = object
+    with pytest.raises(TypeError):
+        not_flag in flags  # pyright: ignore[reportUnusedExpression]
+
+def test_len_flags():
+    flags = Flags([Flag('one'), Flag('two')])
+    assert len(flags) == 2
+    
+def test_bool_flags():
+    flags = Flags([Flag('one'), Flag('two')])
+    assert bool(flags)
+    
+def test_bool_flags2():
+    flags = Flags([])
+    assert not bool(flags)
+    
+def test_getitem_flags():
+    flags = Flags([Flag('one'), Flag('two')])
+    assert flags[1] == Flag('two')
+    
+def test_str_flag():
+    flag = Flag('two')
+    assert str(flag) == '--two'
+    
+def test_repr_flag():
+    flag = Flag('two')
+    assert repr(flag) == 'Flag<name=two, prefix=-->'
+    
+def test_eq_flag():
+    flag = Flag('two')
+    not_flag = object()
+    with pytest.raises(NotImplementedError):
+        flag == not_flag  # pyright: ignore[reportUnusedExpression]
+        
+def test_str_input_flag():
+    flag = InputFlag('two', input_value='value')
+    assert str(flag) == '--two value'
+    
+def test_repr_input_flag():
+    flag = InputFlag('two', input_value='some_value')
+    assert repr(flag) == 'InputFlag<name=two, prefix=--, value=some_value, status=None>'
+    
+def test_eq_input_flag():
+    flag = InputFlag('two', input_value='')
+    not_flag = object()
+    with pytest.raises(NotImplementedError):
+        flag == not_flag  # pyright: ignore[reportUnusedExpression]
+    

tests/unit_tests/test_router.py

@@ -1,7 +1,7 @@
 import re
 import pytest
 
-from argenta.command import Command
+from argenta.command import Command, InputCommand
 from argenta.command.flag import Flag, InputFlag
 from argenta.command.flag.flags import Flags, InputFlags
 from argenta.command.flag.models import PossibleValues, ValidationStatus
@@ -102,3 +102,51 @@ def test_get_router_aliases3():
     def handler(response: Response): 
         pass
     assert router.aliases == set()
+    
+def test_find_appropiate_handler(capsys: pytest.CaptureFixture[str]):
+    router = Router()
+    
+    @router.command(Command('hello', aliases={'hi'}))
+    def handler(res: Response):
+        print("Hello World!")
+        
+    router.finds_appropriate_handler(InputCommand('hi'))
+    
+    output = capsys.readouterr()
+    
+    assert "Hello World!" in output.out
+    
+def test_find_appropiate_handler2(capsys: CaptureFixture[str]):
+    router = Router()
+    
+    @router.command(Command('hello', flags=Flag('flag'), aliases={'hi'}))
+    def handler(res: Response):
+        print("Hello World!")
+        
+    router.finds_appropriate_handler(InputCommand('hi'))
+    
+    output = capsys.readouterr()
+    
+    assert "Hello World!" in output.out
+    
+def test_wrong_typehint(capsys: pytest.CaptureFixture[str]):
+    class NotResponse: pass
+    
+    def func(response: NotResponse): pass
+    
+    _validate_func_args(func)
+    
+    output = capsys.readouterr()
+    
+    assert "WARNING" in output.out
+    
+def test_missing_typehint(capsys: pytest.CaptureFixture[str]):
+    def func(response): pass  # pyright: ignore[reportMissingParameterType, reportUnknownParameterType]
+    
+    _validate_func_args(func)  # pyright: ignore[reportUnknownArgumentType]
+    
+    output = capsys.readouterr()
+    
+    assert output.out == ''
+        
+