nousresearch--hermes-agent/tests/hermes_cli/test_setup_menu_curses_migration.py

"""Regression tests confirming the setup model/provider/reasoning pickers route
through the shared curses radiolist (ESC + arrow-key handling that works across
terminals, incl. Ghostty) instead of simple_term_menu.

Guards against silently regressing back to simple_term_menu, whose ESC/arrow
handling was unreliable in `hermes setup` (the provider->model sub-menu).
"""
from unittest.mock import patch


def test_prompt_model_selection_uses_curses_radiolist():
    from hermes_cli.auth import _prompt_model_selection

    seen = {}

    def _fake(title, items, *, selected=0, cancel_returns=None, description=None, searchable=False):
        seen["title"] = title
        seen["items"] = items
        return 1  # pick second model

    with patch("hermes_cli.curses_ui.curses_radiolist", side_effect=_fake), \
         patch("builtins.print"):
        result = _prompt_model_selection(["model-a", "model-b"])

    assert result == "model-b"
    assert seen["title"] == "Select default model:"
    # Items are the models plus the custom/skip entries.
    assert seen["items"][:2] == ["model-a", "model-b"]
    assert "Skip (keep current)" in seen["items"]


def test_prompt_model_selection_esc_cancels():
    from hermes_cli.auth import _prompt_model_selection

    # curses_radiolist returns the cancel sentinel (-1) on ESC.
    with patch("hermes_cli.curses_ui.curses_radiolist", return_value=-1), \
         patch("builtins.print"):
        result = _prompt_model_selection(["model-a", "model-b"])

    assert result is None


def test_reasoning_effort_uses_curses_radiolist():
    from hermes_cli.main import _prompt_reasoning_effort_selection

    with patch("hermes_cli.curses_ui.curses_radiolist", return_value=2), \
         patch("builtins.print"):
        result = _prompt_reasoning_effort_selection(["low", "medium", "high"], current_effort="")

    assert result == "high"


def test_reasoning_effort_esc_cancels():
    from hermes_cli.main import _prompt_reasoning_effort_selection

    with patch("hermes_cli.curses_ui.curses_radiolist", return_value=-1), \
         patch("builtins.print"):
        result = _prompt_reasoning_effort_selection(["low", "medium", "high"], current_effort="")

    assert result is None


def test_model_selection_with_pricing_passes_description():
    """When pricing is supplied, the aligned header is passed as the curses
    description (multi-line text above the list), not lost."""
    from hermes_cli.auth import _prompt_model_selection

    seen = {}

    def _fake(title, items, *, selected=0, cancel_returns=None, description=None, searchable=False):
        seen["description"] = description
        return len(items) - 1  # Skip

    pricing = {
        "model-a": {"prompt": "0.000001", "completion": "0.000002"},
        "model-b": {"prompt": "0.000003", "completion": "0.000004"},
    }
    with patch("hermes_cli.curses_ui.curses_radiolist", side_effect=_fake), \
         patch("builtins.print"):
        _prompt_model_selection(["model-a", "model-b"], pricing=pricing)

    # The description should carry the In/Out price header.
    assert seen["description"] is not None
    assert "In" in seen["description"] and "Out" in seen["description"]