Files
hermes-agent/tests/tui_gateway/test_ping_probe.py
T
Owenz-creator cdd37035d9 fix(desktop): probe half-open gateway socket on wake and reconnect
macOS sleep/wake (or a silent network drop) can leave the renderer's
WebSocket half-open: no close event fires, so connectionState stays
'open' while every RPC hangs until its per-call timeout. prompt.submit's
timeout is 30 minutes, so the user's next message reads as "enter does
nothing until I restart the app".

- Add a minimal ping RPC (tui_gateway/server.py) answered synchronously
  on the WS reader thread.
- On wake signals, reconnectNow now probes the open-looking socket with a
  5s-bounded ping and force-closes it on failure, letting the existing
  reconnect machinery (backoff, tile rebinding, session refresh) take
  over. A pre-ping backend answering -32601 is treated as healthy.
- Tests: half-open socket force-reconnects; healthy socket untouched;
  method-not-found backend untouched; backend ping envelope contract.
2026-08-24 03:14:19 -07:00

32 lines
1.1 KiB
Python

"""Tests for the ``ping`` liveness probe (tui_gateway/server.py).
The desktop client probes ``ping`` after sleep/wake to distinguish a
half-open TCP socket (connectionState still ``open`` while every RPC hangs)
from a genuinely healthy connection. The contract is minimal on purpose:
answered synchronously on the WS reader thread, no session, no IO, no agent.
"""
from __future__ import annotations
import tui_gateway.server as srv
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"]
def test_ping_registered_and_answers_pong():
res = _call("ping", {})
assert res == {"pong": True}
def test_ping_ignores_params_and_returns_ok_envelope():
# The probe must be parameter-agnostic: the client sends {} but a stray
# future caller must not be able to make it fail.
envelope = srv._methods["ping"](7, {"anything": "value"})
assert envelope["jsonrpc"] == "2.0"
assert envelope["id"] == 7
assert envelope["result"] == {"pong": True}