mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-31 19:16:29 +00:00
6b81590c55
Systematic prune per AGENTS.md test policy, one pass over every major test tree (gateway, hermes_cli, tools, agent, run_agent, plugins, cli, cron, tui_gateway, honcho/openviking, root-level): - DELETE: source-reading tests (read_text/getsource on prod files), change-detector tests (exact catalog counts, model-name snapshots, config version literals), mock-echo tests (assert a mock returns what it was told), assertion-free/trivial tests, near-duplicate parametrizations (boundaries + one representative kept), async/sync twin duplicates, cosmetic within-file variations. - KEEP (mandatory): security/redaction/approval guards, message-role alternation invariants, prompt-caching/deterministic-call-id invariants, issue-number regression tests (deduped), E2E tests. - 6 test files deleted outright (script-style/no-assert or fully redundant); conftest.py, fakes/, fixtures/ untouched. - tests/acp/conftest.py added: autouse fixture stubs the live models.dev/GitHub/Copilot/Anthropic inventory fetches that ACP server tests performed on every session create — test_server.py 147s → 3.4s, and the tests are now genuinely hermetic. - Sleep-based slowness shrunk where safe (codex_ttfb_watchdog, compression_concurrent_fork, etc.); no wall-clock assertion tightened. Verification: full hermetic suite via scripts/run_tests.sh — 2439 files, 31,130 tests passed, 0 failed, 0 flaky retries, 315s wall (baseline: 583s wall, 13,564s subprocess CPU).
96 lines
3.4 KiB
Python
96 lines
3.4 KiB
Python
"""Tests for the Phase 2b billing JSON-RPC methods (tui_gateway/server.py).
|
|
|
|
Verifies the structured envelope contract the Ink side branches on:
|
|
- billing.state serializes BillingState (Decimals → strings) + fails open.
|
|
- billing.charge / charge_status / auto_reload return typed error envelopes
|
|
(result.ok=false, result.error=<code>) instead of JSON-RPC errors.
|
|
- billing.charge mints + echoes an idempotency_key for retry reuse.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from decimal import Decimal
|
|
|
|
import pytest
|
|
|
|
import tui_gateway.server as srv
|
|
import hermes_cli.nous_billing as nb
|
|
import agent.billing_view as bv
|
|
from agent.billing_view import BillingState, CardInfo, MonthlyCap, PaymentMethodInfo
|
|
|
|
|
|
def _call(method: str, params: dict) -> dict:
|
|
"""Invoke a registered RPC method and return its result dict."""
|
|
envelope = srv._methods[method](1, params)
|
|
return envelope["result"]
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# billing.state
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_billing_state_serializes_decimals_as_strings(monkeypatch):
|
|
state = BillingState(
|
|
logged_in=True,
|
|
org_name="Acme",
|
|
role="OWNER",
|
|
balance_usd=Decimal("142.5"),
|
|
cli_billing_enabled=True,
|
|
charge_presets=(Decimal("100"), Decimal("250")),
|
|
min_usd=Decimal("10"),
|
|
max_usd=Decimal("10000"),
|
|
card=CardInfo(brand="visa", last4="4242"),
|
|
payment_method=PaymentMethodInfo(
|
|
kind="link",
|
|
email="billing@example.com",
|
|
resolved_via="customerDefault",
|
|
),
|
|
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)
|
|
res = _call("billing.state", {})
|
|
assert res["ok"] is True and res["logged_in"] is True
|
|
# Money on the wire is STRING, not float/number.
|
|
assert res["balance_usd"] == "142.5"
|
|
assert res["balance_display"] == "$142.50"
|
|
assert res["charge_presets"] == ["100", "250"]
|
|
assert res["card"] == {
|
|
"brand": "visa",
|
|
"last4": "4242",
|
|
"masked": "visa ····4242",
|
|
"display": "visa ····4242",
|
|
"resolved_via": None,
|
|
}
|
|
assert res["payment_method"] == {
|
|
"kind": "link",
|
|
"email": "billing@example.com",
|
|
"resolved_via": "customerDefault",
|
|
}
|
|
assert res["monthly_cap"]["is_default_ceiling"] is True
|
|
assert res["is_admin"] is True and res["can_charge"] is True
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# billing.charge — typed error envelopes
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# billing.charge_status — the poll
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# billing.auto_reload
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# billing.step_up
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|