Hermes-agent

2026-06-14 14:30:48 -04:00
commit dac4b88b94
5058 changed files with 1884848 additions and 0 deletions
@@ -0,0 +1,260 @@
+"""Tests for the /credits command — shared view core + gateway handler.
+
+`/credits` is the focused money surface (balance in, top-up out). These tests
+exercise the surface-agnostic `build_credits_view()` core and assert the gateway
+handler renders the block + tappable top-up URL + no-wait copy. The CLI panel is
+a thin wrapper over the same view (interactive prompt_toolkit modal — covered by
+the view-core tests plus manual verification).
+"""
+
+from __future__ import annotations
+
+import asyncio
+
+import pytest
+
+import agent.account_usage as account_usage
+from agent.account_usage import CreditsView, build_credits_view
+from hermes_cli.nous_account import NousPortalAccountInfo, NousPaidServiceAccessInfo
+
+
+def _account(**kwargs) -> NousPortalAccountInfo:
+    kwargs.setdefault("logged_in", True)
+    kwargs.setdefault("source", "account_api")
+    kwargs.setdefault("fresh", True)
+    kwargs.setdefault("portal_base_url", "https://portal.example.test")
+    return NousPortalAccountInfo(**kwargs)
+
+
+@pytest.fixture
+def _logged_in_account(monkeypatch):
+    """Stub the auth token + account fetch so build_credits_view runs offline."""
+    monkeypatch.setattr(
+        "hermes_cli.auth.get_provider_auth_state",
+        lambda provider: {"access_token": "tok", "portal_base_url": "https://portal.example.test"},
+    )
+
+    def _install(account):
+        monkeypatch.setattr(
+            "hermes_cli.nous_account.get_nous_portal_account_info",
+            lambda *a, **kw: account,
+        )
+
+    return _install
+
+
+# ── build_credits_view core ─────────────────────────────────────────────────
+
+
+def test_view_logged_out_when_no_token(monkeypatch):
+    monkeypatch.setattr("hermes_cli.auth.get_provider_auth_state", lambda provider: {})
+    view = build_credits_view()
+    assert view == CreditsView(logged_in=False)
+
+
+def test_view_built_with_org_pinned_url_and_identity(_logged_in_account):
+    _logged_in_account(
+        _account(
+            org_slug="acme",
+            org_name="Acme Inc",
+            email="alice@example.test",
+            paid_service_access=True,
+            paid_service_access_info=NousPaidServiceAccessInfo(
+                purchased_credits_remaining=30.0,
+                total_usable_credits=30.0,
+            ),
+            subscription=None,
+        )
+    )
+
+    view = build_credits_view()
+
+    assert view.logged_in is True
+    assert view.topup_url == "https://portal.example.test/orgs/acme/billing?topup=open"
+    assert view.identity_line == "Topping up as alice@example.test / org Acme Inc"
+    assert view.depleted is False
+    # Balance lines carry the magnitudes but NOT the /usage affordance lines.
+    blob = "\n".join(view.balance_lines)
+    assert "Top-up credits: $30.00" in blob
+    assert "Top up:" not in blob  # the trailing /usage affordance is stripped
+    assert "(or run" not in blob
+
+
+def test_view_depleted_flag(_logged_in_account):
+    _logged_in_account(
+        _account(
+            org_slug="acme",
+            email="alice@example.test",
+            paid_service_access=False,
+            paid_service_access_info=NousPaidServiceAccessInfo(
+                total_usable_credits=0.0,
+            ),
+            subscription=None,
+        )
+    )
+
+    view = build_credits_view()
+    assert view.depleted is True
+
+
+def test_view_falls_back_to_legacy_url_when_slug_null(_logged_in_account):
+    _logged_in_account(
+        _account(
+            org_slug=None,
+            email="alice@example.test",
+            paid_service_access=True,
+            paid_service_access_info=NousPaidServiceAccessInfo(
+                purchased_credits_remaining=5.0,
+                total_usable_credits=5.0,
+            ),
+            subscription=None,
+        )
+    )
+
+    view = build_credits_view()
+    assert view.topup_url == "https://portal.example.test/billing?topup=open"
+    assert "/orgs/" not in view.topup_url
+
+
+def test_view_fetch_failure_is_logged_out(monkeypatch):
+    monkeypatch.setattr(
+        "hermes_cli.auth.get_provider_auth_state",
+        lambda provider: {"access_token": "tok"},
+    )
+
+    def _boom(*a, **kw):
+        raise RuntimeError("portal down")
+
+    monkeypatch.setattr("hermes_cli.nous_account.get_nous_portal_account_info", _boom)
+
+    view = build_credits_view()
+    assert view.logged_in is False
+
+
+# ── gateway _handle_credits_command ─────────────────────────────────────────
+
+
+class _FakeEvent:
+    pass
+
+
+def _make_gateway_stub():
+    """Minimal object exposing the mixin's _handle_credits_command."""
+    from gateway.slash_commands import GatewaySlashCommandsMixin
+
+    class _Stub(GatewaySlashCommandsMixin):
+        def __init__(self):
+            pass
+
+    return _Stub()
+
+
+def test_gateway_credits_renders_block_and_url(monkeypatch):
+    view = CreditsView(
+        logged_in=True,
+        balance_lines=("📈 Nous credits", "Total usable: $52.50"),
+        identity_line="Topping up as alice@example.test / org Acme",
+        topup_url="https://portal.example.test/orgs/acme/billing?topup=open",
+        depleted=False,
+    )
+    monkeypatch.setattr(account_usage, "build_credits_view", lambda *a, **kw: view)
+
+    stub = _make_gateway_stub()
+    out = asyncio.run(stub._handle_credits_command(_FakeEvent()))
+
+    assert "💳" in out
+    assert "Total usable: $52.50" in out
+    assert "Topping up as alice@example.test / org Acme" in out
+    assert "https://portal.example.test/orgs/acme/billing?topup=open" in out
+    assert "credits will appear in /credits shortly" in out
+    # The helper's own 📈 header line is dropped (we render our own 💳 header).
+    assert "📈 Nous credits" not in out
+
+
+def test_gateway_credits_not_logged_in(monkeypatch):
+    monkeypatch.setattr(
+        account_usage, "build_credits_view", lambda *a, **kw: CreditsView(logged_in=False)
+    )
+    stub = _make_gateway_stub()
+    out = asyncio.run(stub._handle_credits_command(_FakeEvent()))
+    assert "Not logged into Nous Portal" in out
+
+
+def test_gateway_credits_fetch_exception_is_not_logged_in(monkeypatch):
+    def _boom(*a, **kw):
+        raise RuntimeError("boom")
+
+    monkeypatch.setattr(account_usage, "build_credits_view", _boom)
+    stub = _make_gateway_stub()
+    out = asyncio.run(stub._handle_credits_command(_FakeEvent()))
+    assert "Not logged into Nous Portal" in out
+
+
+# ── command registry ────────────────────────────────────────────────────────
+
+
+def test_credits_command_registered():
+    from hermes_cli.commands import resolve_command, COMMAND_REGISTRY
+
+    cmd = resolve_command("credits")
+    assert cmd is not None and cmd.name == "credits"
+    # Available on every surface (not cli_only / gateway_only).
+    entry = next(c for c in COMMAND_REGISTRY if c.name == "credits")
+    assert entry.cli_only is False
+    assert entry.gateway_only is False
+
+
+# ── CLI _show_credits non-interactive (TUI slash-worker) path ───────────────
+
+
+def test_cli_show_credits_non_interactive_renders_text_not_modal(monkeypatch, capsys):
+    """In the TUI slash-worker (no self._app), /credits must render the text
+    variant — never invoke the prompt_toolkit modal, which would read the
+    worker's JSON-RPC stdin and crash the command (only the depleted banner
+    would survive). Regression for that exact failure.
+    """
+    import agent.account_usage as account_usage
+    from cli import HermesCLI
+
+    monkeypatch.setattr(
+        account_usage,
+        "build_credits_view",
+        lambda *a, **k: CreditsView(
+            logged_in=True,
+            balance_lines=("📈 Nous credits", "Total usable: $0.00"),
+            identity_line="Topping up as a@b.c / org Acme",
+            topup_url="https://prev.test/orgs/acme/billing?topup=open",
+            depleted=True,
+        ),
+    )
+
+    cli = HermesCLI.__new__(HermesCLI)
+    cli._app = None  # non-interactive, like the slash worker
+
+    # Must NOT call the modal in this context.
+    def _boom_modal(*a, **k):
+        raise AssertionError("modal must not run without a live app")
+
+    monkeypatch.setattr(HermesCLI, "_prompt_text_input_modal", _boom_modal, raising=False)
+
+    cli._show_credits()
+
+    out = capsys.readouterr().out
+    assert "💳 Nous credits" in out
+    assert "Total usable: $0.00" in out
+    assert "Topping up as a@b.c / org Acme" in out
+    assert "https://prev.test/orgs/acme/billing?topup=open" in out
+    assert "credits will appear in /credits shortly" in out
+
+
+def test_cli_show_credits_logged_out(monkeypatch, capsys):
+    import agent.account_usage as account_usage
+    from cli import HermesCLI
+
+    monkeypatch.setattr(
+        account_usage, "build_credits_view", lambda *a, **k: CreditsView(logged_in=False)
+    )
+    cli = HermesCLI.__new__(HermesCLI)
+    cli._app = None
+    cli._show_credits()
+    assert "Not logged into Nous Portal" in capsys.readouterr().out