hermes-agent/tests/hermes_cli/test_cli_status_bar_goal.py

"""Status-bar goal segment (⊙ goal N/M) — active-goal-only rendering.

The segment mirrors the desktop composer goal indicator: it appears only
while a /goal is ACTIVE, shows turns used vs the turn budget, and stays out
of the bar entirely for paused/done/absent goals (those already print their
own glyph lines in the conversation thread).
"""

from datetime import datetime, timedelta
from types import SimpleNamespace

from cli import HermesCLI

def _make_cli(model: str = "anthropic/claude-sonnet-4-20250514"):
    cli_obj = HermesCLI.__new__(HermesCLI)
    cli_obj.model = model
    cli_obj.session_start = datetime.now() - timedelta(minutes=14, seconds=32)
    cli_obj.conversation_history = [{"role": "user", "content": "hi"}]
    cli_obj.agent = None
    return cli_obj

def _attach_goal(cli_obj, *, active: bool, turns_used: int = 3, max_turns: int = 20):
    """Bind a fake GoalManager the way _get_goal_manager caches one."""
    cli_obj.session_id = "sess-goal-test"
    cli_obj._goal_manager = SimpleNamespace(
        session_id="sess-goal-test",
        is_active=lambda: active,
        state=SimpleNamespace(turns_used=turns_used, max_turns=max_turns),
    )
    return cli_obj

class TestStatusBarGoalSegment:
    def test_goal_segment_composition(self):
        cli_obj = _attach_goal(_make_cli(), active=True, turns_used=3, max_turns=20)

        snapshot = cli_obj._get_status_bar_snapshot()

        assert snapshot["goal_active"] is True
        assert snapshot["goal_turns_used"] == 3
        assert snapshot["goal_max_turns"] == 20
        assert cli_obj._status_bar_goal_segment(snapshot) == "⊙ goal 3/20"

    def test_goal_segment_absent_when_paused(self):
        # Paused goals must NOT occupy the status bar (active-only contract).
        cli_obj = _attach_goal(_make_cli(), active=False)

        snapshot = cli_obj._get_status_bar_snapshot()

        assert snapshot["goal_active"] is False
        assert cli_obj._status_bar_goal_segment(snapshot) == ""

    def test_active_goal_rendered_in_wide_status_bar(self):
        cli_obj = _attach_goal(_make_cli(), active=True, turns_used=5, max_turns=20)

        text = cli_obj._build_status_bar_text(width=120)

        assert "⊙ goal 5/20" in text