mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-31 19:16:29 +00:00
f04fd1e7ad
Guard the .lazy-refresh-incomplete marker writer (update_cmd), launch-time recovery (main.py), and _early_recovery repair paths behind a two-condition check: running under pytest AND the target is this live checkout. Sandboxed tmp_path tests still exercise the real code paths. Salvaged from PR #72002 by @fcavalcantirj. Fixes #72000. Co-authored-by: fcavalcantirj <felipe.cavalcanti.rj@gmail.com>
72 lines
2 KiB
Python
72 lines
2 KiB
Python
"""Tests for the TUI gateway relaying MoA reference events to the client.
|
|
|
|
When a MoA preset is the active model, the agent's tool_progress_callback emits
|
|
``moa.reference`` (one per reference model, before the aggregator acts) and a
|
|
single ``moa.aggregating`` marker. ``_on_tool_progress`` must forward these to
|
|
the Ink/desktop client as labelled events so each reference renders like a
|
|
thinking block tagged with its source model.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
import pytest
|
|
|
|
|
|
@pytest.fixture()
|
|
def server():
|
|
# Mocks are scoped to the initial import only (see
|
|
# tests/tui_gateway/test_protocol.py for the rationale).
|
|
with patch.dict(
|
|
"sys.modules",
|
|
{
|
|
"hermes_constants": MagicMock(
|
|
get_hermes_home=MagicMock(return_value="/tmp/hermes_test_moa_emit")
|
|
),
|
|
"hermes_cli.env_loader": MagicMock(),
|
|
"hermes_cli.banner": MagicMock(),
|
|
"hermes_state": MagicMock(),
|
|
},
|
|
):
|
|
import importlib
|
|
|
|
mod = importlib.import_module("tui_gateway.server")
|
|
|
|
yield mod
|
|
mod._sessions.clear()
|
|
|
|
|
|
@pytest.fixture()
|
|
def emits(server, monkeypatch):
|
|
captured: list = []
|
|
monkeypatch.setattr(
|
|
server,
|
|
"_emit",
|
|
lambda event, sid, payload=None: captured.append((event, sid, payload)),
|
|
)
|
|
monkeypatch.setattr(server, "_tool_progress_enabled", lambda sid: True)
|
|
return captured
|
|
|
|
|
|
def test_moa_reference_relayed_with_label_and_index(server, emits):
|
|
server._on_tool_progress(
|
|
"sid-1",
|
|
"moa.reference",
|
|
"openrouter:openai/gpt-5.5",
|
|
"Paris is the capital of France.",
|
|
None,
|
|
moa_index=1,
|
|
moa_count=2,
|
|
)
|
|
|
|
assert len(emits) == 1
|
|
event, sid, payload = emits[0]
|
|
assert event == "moa.reference"
|
|
assert sid == "sid-1"
|
|
assert payload["label"] == "openrouter:openai/gpt-5.5"
|
|
assert payload["text"] == "Paris is the capital of France."
|
|
assert payload["index"] == 1
|
|
assert payload["count"] == 2
|
|
|
|
|