53f598e7a2
Pure, refactor-independent helpers for type-to-filter search in the curses single-/radio-select menus: subsequence matching, filtered-index mapping, cursor reconciliation, scroll clamping, and an active-search key handler, plus unit tests. Salvaged from #22758 (the curses event loop was since refactored into a shared driver on main, so the integration is rebuilt in a follow-up commit; these pure helpers and their tests carry over unchanged).
69 lines
1.8 KiB
Python
69 lines
1.8 KiB
Python
from hermes_cli.curses_ui import (
|
|
_SearchState,
|
|
_filter_indices,
|
|
_handle_active_search_key,
|
|
_move_filtered_cursor,
|
|
_reconcile_cursor,
|
|
)
|
|
|
|
|
|
class _FakeCurses:
|
|
KEY_BACKSPACE = 263
|
|
KEY_DOWN = 258
|
|
KEY_ENTER = 343
|
|
|
|
|
|
def test_filter_indices_keeps_all_items_for_blank_query():
|
|
assert _filter_indices(["Anthropic", "OpenAI"], "") == [0, 1]
|
|
assert _filter_indices(["Anthropic", "OpenAI"], " ") == [0, 1]
|
|
|
|
|
|
def test_filter_indices_matches_subsequences():
|
|
items = ["claude-opus-4-7", "gpt-5.4-codex", "deepseek-v4"]
|
|
|
|
assert _filter_indices(items, "co47") == [0]
|
|
assert _filter_indices(items, "gpt5") == [1]
|
|
|
|
|
|
def test_filter_indices_requires_all_tokens():
|
|
items = ["OpenAI Codex", "OpenAI Chat Completions", "Anthropic Claude"]
|
|
|
|
assert _filter_indices(items, "open cod") == [0]
|
|
|
|
|
|
def test_reconcile_cursor_moves_to_first_visible_match():
|
|
assert _reconcile_cursor([2, 4], 0) == (2, 0)
|
|
assert _reconcile_cursor([2, 4], 4) == (4, 1)
|
|
|
|
|
|
def test_move_filtered_cursor_wraps_within_matches():
|
|
filtered = [2, 4, 7]
|
|
|
|
assert _move_filtered_cursor(filtered, 2, 0, -1) == 7
|
|
assert _move_filtered_cursor(filtered, 7, 2, 1) == 2
|
|
|
|
|
|
def test_active_search_allows_navigation_keys_to_reach_menu_loop():
|
|
search = _SearchState(active=True, query="opus")
|
|
|
|
assert _handle_active_search_key(_FakeCurses, _FakeCurses.KEY_DOWN, search) == (
|
|
False,
|
|
False,
|
|
False,
|
|
)
|
|
assert search.active is True
|
|
assert search.query == "opus"
|
|
|
|
|
|
def test_active_search_consumes_query_editing_and_confirm_keys():
|
|
search = _SearchState(active=True, query="op")
|
|
|
|
assert _handle_active_search_key(_FakeCurses, ord("u"), search) == (True, False, True)
|
|
assert search.query == "opu"
|
|
|
|
assert _handle_active_search_key(_FakeCurses, _FakeCurses.KEY_ENTER, search) == (
|
|
True,
|
|
True,
|
|
False,
|
|
)
|