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, print_func=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