Тестирование

В этом разделе описаны практики тестирования приложений на основе Argenta. Примеры основаны на фактическом публичном API.

Модульное тестирование обработчиков

Обработчики в Argenta — обычные функции. Их удобно тестировать как чистые функции, не поднимая весь цикл приложения. Рекомендуются unittest или pytest.

Пример использования:

import io
from contextlib import redirect_stdout

from argenta import Router, Command, Response
from argenta.command import InputCommand


router = Router(title="Demo")


@router.command(Command("PING", description="Ping command"))
def ping(response: Response):
    print("PONG")


def test_ping_prints_pong():
    # Call handler
    with redirect_stdout(io.StringIO()) as stdout:
        router.finds_appropriate_handler(InputCommand.parse("PING"))
    assert "PONG" in stdout.getvalue()

---

Тестирование с внедрением зависимостей (DI)

Если обработчику нужны зависимости, используйте dishka и интеграцию Argenta:

Пример использования:

import io
from contextlib import redirect_stdout

from argenta.command import InputCommand
from dishka import Provider, make_container, Scope

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("HELLO")
def hello(response: Response, service: FromDishka[Service]) -> None:
    print(f"hello {service.hello()}")


class _FakeApp:
    # Minimal stub for setup_dishka; app object is not used in unit tests
    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()

---

Интеграционное тестирование приложения

Для более высокого уровня тестов собирайте App и Router и вызывайте обработчики через парсинг команд, обходя бесконечный цикл ввода. Это даёт близкое к реальности поведение без необходимости симулировать stdin.

Пример использования:

import io
from contextlib import redirect_stdout

from argenta import App, Router, Command, Response
from argenta.command import InputCommand


def test_simple_app() -> None:
    app = App(override_system_messages=True, repeat_command_groups_printing=False)
    router = Router(title="App")

    @router.command(Command("HELP", description="Show help"))
    def help_cmd(response: Response):
        print("Available commands: HELP")

    app.include_router(router)

    with redirect_stdout(io.StringIO()) as stdout:
        router.finds_appropriate_handler(InputCommand.parse("HELP"))

    assert "Available commands:" in stdout.getvalue()

---

E2E-тестирование цикла

Полный запуск цикла start_polling можно покрывать через подпроцесс с передачей строк в stdin. Это тяжелее и обычно не требуется. Если всё же необходимо — пример ниже.

Опасно

Важно: Обязательно передавайте строковый триггер команды выхода последним элементом в списке side_effects при патче input.

Иначе тестируемое приложение будет ожидать ввода следующей команды и не сможет корректно завершиться.

Пример использования:

import sys
from unittest.mock import patch
import pytest
from pytest import CaptureFixture

from argenta import App, Orchestrator, Router, Command, Response


@pytest.fixture(autouse=True)
def patched_argv():
    with patch.object(sys, "argv", ["program.py"]):
        yield


def test_input_incorrect_command(capsys: CaptureFixture[str]):
    router = Router()
    orchestrator = Orchestrator()

    @router.command(Command("test"))
    def test(response: Response) -> None:
        print("test command")

    app = App(override_system_messages=True, printer=print)
    app.include_router(router)
    app.set_unknown_command_handler(lambda command: print(f"Unknown command: {command.trigger}"))

    with patch("builtins.input", side_effect=["help", "q"]):
        orchestrator.start_polling(app)

    output = capsys.readouterr().out
    assert "\nUnknown command: help\n" in output

---

Советы по тестированию

1. Изолируйте тесты: Каждый тест должен быть независимым от других.

2. Моки для внешних интеграций: БД, HTTP-клиенты и т.п. подменяйте заглушками и провайдерами dishka.

3. Покрывайте ошибочные сценарии: Некорректные флаги, неизвестные команды, пустой ввод.

4. Минимизируйте зависимость от форматирования: Сравнивайте ключевые фрагменты вывода, а не весь блок целиком.

5. Измеряйте покрытие: Используйте pytest-cov.