mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-06-27 11:22:03 +00:00
130 lines
4 KiB
Python
130 lines
4 KiB
Python
"""Focused tests for dashboard PTY reconnect breadcrumbs."""
|
|
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
from urllib.parse import urlencode
|
|
|
|
import pytest
|
|
|
|
|
|
pytestmark = pytest.mark.skipif(
|
|
sys.platform.startswith("win"), reason="PTY bridge is POSIX-only"
|
|
)
|
|
|
|
|
|
class _OneFrameBridge:
|
|
def __init__(self):
|
|
self._sent = False
|
|
|
|
@classmethod
|
|
def spawn(cls, *args, **kwargs):
|
|
return cls()
|
|
|
|
def read(self, timeout):
|
|
if not self._sent:
|
|
self._sent = True
|
|
return b"ready"
|
|
return None
|
|
|
|
def resize(self, *, cols, rows):
|
|
pass
|
|
|
|
def write(self, raw):
|
|
pass
|
|
|
|
def close(self):
|
|
pass
|
|
|
|
|
|
@pytest.fixture
|
|
def pty_client(monkeypatch, _isolate_hermes_home):
|
|
from starlette.testclient import TestClient
|
|
|
|
import hermes_cli.web_server as ws
|
|
|
|
monkeypatch.setattr(ws, "_DASHBOARD_EMBEDDED_CHAT_ENABLED", True)
|
|
monkeypatch.setattr(ws.PtyBridge, "spawn", _OneFrameBridge.spawn)
|
|
ws.app.state.pty_active_session_files = {}
|
|
|
|
client = TestClient(ws.app)
|
|
return ws, client, ws._SESSION_TOKEN
|
|
|
|
|
|
def _url(token: str, **params: str) -> str:
|
|
return f"/api/pty?{urlencode({'token': token, **params})}"
|
|
|
|
|
|
def test_resolve_chat_argv_sets_active_session_file_env(monkeypatch):
|
|
"""Dashboard chat gives the TUI a breadcrumb file for reconnect resume."""
|
|
import hermes_cli.main as main_mod
|
|
import hermes_cli.web_server as ws
|
|
|
|
monkeypatch.setattr(
|
|
main_mod,
|
|
"_make_tui_argv",
|
|
lambda project_root, tui_dev=False: (["node", "dist/entry.js"], "/tmp/ui-tui"),
|
|
)
|
|
|
|
_argv, _cwd, env = ws._resolve_chat_argv(
|
|
active_session_file="/tmp/hermes-active-session.json"
|
|
)
|
|
|
|
assert env["HERMES_TUI_ACTIVE_SESSION_FILE"] == "/tmp/hermes-active-session.json"
|
|
|
|
|
|
def test_channel_reconnect_resumes_active_session_file(pty_client, monkeypatch):
|
|
"""A new /api/pty socket on the same channel resumes the last TUI sid."""
|
|
ws, client, token = pty_client
|
|
captured = []
|
|
|
|
def fake_resolve(resume=None, sidecar_url=None, profile=None, active_session_file=None):
|
|
captured.append(
|
|
{
|
|
"active_session_file": active_session_file,
|
|
"resume": resume,
|
|
"sidecar_url": sidecar_url,
|
|
}
|
|
)
|
|
if active_session_file and not resume:
|
|
Path(active_session_file).write_text(
|
|
json.dumps({"session_id": "sess-live"}),
|
|
encoding="utf-8",
|
|
)
|
|
return (["fake-hermes-tui"], None, None)
|
|
|
|
monkeypatch.setattr(ws, "_resolve_chat_argv", fake_resolve)
|
|
|
|
with client.websocket_connect(_url(token, channel="reconnect-chan")) as conn:
|
|
assert conn.receive_bytes() == b"ready"
|
|
|
|
with client.websocket_connect(_url(token, channel="reconnect-chan")) as conn:
|
|
assert conn.receive_bytes() == b"ready"
|
|
|
|
assert captured[0]["resume"] is None
|
|
assert captured[0]["active_session_file"]
|
|
assert captured[1]["resume"] == "sess-live"
|
|
assert captured[1]["active_session_file"] == captured[0]["active_session_file"]
|
|
|
|
|
|
def test_fresh_param_ignores_channel_active_session_file(pty_client, monkeypatch):
|
|
"""Explicit fresh starts must not resurrect the prior channel session."""
|
|
ws, client, token = pty_client
|
|
channel = "fresh-chan"
|
|
active_file = ws._active_session_file_for_channel(ws.app, channel)
|
|
active_file.write_text(json.dumps({"session_id": "sess-old"}), encoding="utf-8")
|
|
captured = {}
|
|
|
|
def fake_resolve(resume=None, sidecar_url=None, profile=None, active_session_file=None):
|
|
captured["active_session_file"] = active_session_file
|
|
captured["resume"] = resume
|
|
return (["fake-hermes-tui"], None, None)
|
|
|
|
monkeypatch.setattr(ws, "_resolve_chat_argv", fake_resolve)
|
|
|
|
with client.websocket_connect(_url(token, channel=channel, fresh="1")) as conn:
|
|
assert conn.receive_bytes() == b"ready"
|
|
|
|
assert captured["resume"] is None
|
|
assert captured["active_session_file"] == str(active_file)
|
|
assert not active_file.exists()
|