scriptoriumadmin/strix
1
0
Fork 0
mirror of https://github.com/usestrix/strix.git synced 2026-10-14 04:08:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Unwrap chunk envelopes in spend-request poll output

Browse source
This commit is contained in:
Alex Schapiro 2026-09-01 19:17:33 +00:00
parent 995ba47e05
commit 24a5a362fb
2 changed files with 17 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
strix/interface/cloud/billing.py
View file

@ -444,7 +444,14 @@ def _spend_request_records(stdout: str) -> list[dict[str, Any]]:
records: list[dict[str, Any]] = []
for candidate in _embedded_json_documents((stdout or "").strip()):
items = candidate if isinstance(candidate, list) else [candidate]
records.extend(cast("dict[str, Any]", item) for item in items if isinstance(item, dict))
for item in items:
if not isinstance(item, dict):
continue
record = cast("dict[str, Any]", item)
data = record.get("data")
if isinstance(data, dict):
record = cast("dict[str, Any]", data)
records.append(record)
return records

9
tests/test_cloud_wallet.py
View file

@ -95,6 +95,15 @@ def test_final_spend_request_status_reads_the_last_poll_line() -> None:
assert billing._final_spend_request_status("") is None
def test_final_spend_request_status_unwraps_chunk_envelopes() -> None:
stdout = (
'{"type":"chunk","data":{"id":"lsrq_9","status":"pending_approval"}}\n'
'{"type":"chunk","data":{"id":"lsrq_9","status":"approved"}}\n'
'{"type":"done","ok":true,"meta":{"command":"spend-request retrieve"}}\n'
)
assert billing._final_spend_request_status(stdout) == "approved"
def test_prepare_link_wallet_skips_login_when_connected(monkeypatch: pytest.MonkeyPatch) -> None:
monkeypatch.setattr(billing, "_link_wallet_authenticated", lambda _npx: True)
assert billing._prepare_link_wallet(Console(), "npx", as_json=True) is None