mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-22 10:32:00 +00:00
1986666c12
- CommandDef("billing", subcommands=buy|auto-reload|limit), added to
_SLACK_VIA_HERMES_ONLY so it routes via /hermes on Slack (keeps the 50-cap
parity test green, same as /credits).
- cli.py::_show_billing + screen helpers: all 5 screens (overview, buy→confirm→
poll, auto-reload, monthly-limit read-only). Reuses _prompt_text_input_modal /
_prompt_text_input (D-C). Non-interactive (_app is None) renders text + portal
deep-link, never prompts (R7). Decimal money end-to-end. 2s/5-min cancellable
poll loop; 429/503 = retry not failure; settled = ledger truth. Lazy step-up on
403 insufficient_scope. no_payment_method treated as mainline funnel-to-portal.
- 6 CLI tests; 156 command tests (incl. Slack/Telegram parity) green.
113 lines
4.3 KiB
Python
113 lines
4.3 KiB
Python
"""Tests for the /billing CLI handler (cli.py::_show_billing).
|
|
|
|
Focus on the non-interactive (no live prompt_toolkit app) path — the same
|
|
discipline as the /credits non-interactive test: it must render text, never
|
|
invoke the modal (which would read the slash-worker's JSON-RPC stdin and hang).
|
|
Plus role/kill-switch gating and logged-out handling.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from decimal import Decimal
|
|
|
|
import pytest
|
|
|
|
import agent.billing_view as bv
|
|
from agent.billing_view import BillingState, CardInfo, MonthlyCap
|
|
from cli import HermesCLI
|
|
|
|
|
|
@pytest.fixture
|
|
def cli():
|
|
obj = HermesCLI.__new__(HermesCLI) # bypass __init__ (no full app needed)
|
|
obj._app = None # non-interactive: forces the text path
|
|
return obj
|
|
|
|
|
|
def _boom_modal(*a, **kw):
|
|
raise AssertionError("modal must NOT be called in non-interactive mode")
|
|
|
|
|
|
def test_billing_logged_out(cli, monkeypatch, capsys):
|
|
monkeypatch.setattr(bv, "build_billing_state", lambda *a, **kw: BillingState(logged_in=False))
|
|
cli._show_billing("/billing")
|
|
out = capsys.readouterr().out
|
|
assert "Not logged into Nous Portal" in out
|
|
assert "hermes portal" in out
|
|
|
|
|
|
def test_billing_overview_non_interactive_renders_text_not_modal(cli, monkeypatch, capsys):
|
|
monkeypatch.setattr(HermesCLI, "_prompt_text_input_modal", _boom_modal, raising=False)
|
|
state = BillingState(
|
|
logged_in=True,
|
|
org_name="Acme",
|
|
role="OWNER",
|
|
balance_usd=Decimal("142.5"),
|
|
cli_billing_enabled=True,
|
|
charge_presets=(Decimal("100"),),
|
|
monthly_cap=MonthlyCap(limit_usd=Decimal("1000"), spent_this_month_usd=Decimal("180"),
|
|
is_default_ceiling=True),
|
|
portal_url="https://portal/billing?topup=open",
|
|
)
|
|
monkeypatch.setattr(bv, "build_billing_state", lambda *a, **kw: state)
|
|
cli._show_billing("/billing")
|
|
out = capsys.readouterr().out
|
|
assert "Usage credits" in out
|
|
assert "$142.50" in out
|
|
assert "$180 of $1000 used (default ceiling)" in out
|
|
# Non-interactive lists subcommands as text.
|
|
assert "/billing buy" in out
|
|
assert "Manage on portal:" in out
|
|
|
|
|
|
def test_billing_member_cannot_charge(cli, monkeypatch, capsys):
|
|
state = BillingState(
|
|
logged_in=True, role="MEMBER", balance_usd=Decimal("10"),
|
|
cli_billing_enabled=True, portal_url="https://portal/billing",
|
|
)
|
|
monkeypatch.setattr(bv, "build_billing_state", lambda *a, **kw: state)
|
|
cli._show_billing("/billing")
|
|
out = capsys.readouterr().out
|
|
assert "require an org admin/owner" in out
|
|
|
|
|
|
def test_billing_killswitch_off_blocks(cli, monkeypatch, capsys):
|
|
state = BillingState(
|
|
logged_in=True, role="OWNER", balance_usd=Decimal("10"),
|
|
cli_billing_enabled=False, portal_url="https://portal/billing",
|
|
)
|
|
monkeypatch.setattr(bv, "build_billing_state", lambda *a, **kw: state)
|
|
cli._show_billing("/billing")
|
|
out = capsys.readouterr().out
|
|
assert "turned off for this org" in out
|
|
|
|
|
|
def test_billing_limit_screen_readonly(cli, monkeypatch, capsys):
|
|
state = BillingState(
|
|
logged_in=True, role="OWNER", cli_billing_enabled=True,
|
|
monthly_cap=MonthlyCap(limit_usd=Decimal("1000"), spent_this_month_usd=Decimal("250"),
|
|
is_default_ceiling=True),
|
|
portal_url="https://portal/billing",
|
|
)
|
|
monkeypatch.setattr(bv, "build_billing_state", lambda *a, **kw: state)
|
|
cli._show_billing("/billing limit")
|
|
out = capsys.readouterr().out
|
|
assert "Monthly spend limit" in out
|
|
assert "$250 of $1000 used" in out
|
|
assert "read-only" in out
|
|
|
|
|
|
def test_billing_buy_non_interactive_defers_to_portal(cli, monkeypatch, capsys):
|
|
monkeypatch.setattr(HermesCLI, "_prompt_text_input_modal", _boom_modal, raising=False)
|
|
state = BillingState(
|
|
logged_in=True, role="OWNER", cli_billing_enabled=True,
|
|
charge_presets=(Decimal("100"), Decimal("250")),
|
|
card=CardInfo(brand="visa", last4="4242"),
|
|
portal_url="https://portal/billing",
|
|
)
|
|
monkeypatch.setattr(bv, "build_billing_state", lambda *a, **kw: state)
|
|
cli._show_billing("/billing buy")
|
|
out = capsys.readouterr().out
|
|
assert "Buy usage credits" in out
|
|
assert "$100" in out and "$250" in out
|
|
assert "interactive CLI" in out # defers; no charge attempted non-interactively
|