"""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=) 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 # ---------------------------------------------------------------------------