mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-12 13:52:15 +00:00
e1067dbbe5
Main's rewritten test_tui_npm_install.py tests call _make_tui_argv expecting the Ink/npm flow unconditionally; with the dual-engine dispatch merged in, _resolve_tui_engine() auto-selects opentui whenever ui-opentui/dist is built in the repo, routing the call away from the path under test (first subprocess became 'node --version' instead of 'npm run build'). Pin the engine to ink via an autouse fixture, mirroring the existing pinning precedent in test_tui_resume_flow.py.
138 lines
5.2 KiB
Python
138 lines
5.2 KiB
Python
"""Group-chat mention-gating tests for PhotonAdapter.
|
|
|
|
Parity with the BlueBubbles iMessage channel: when ``require_mention`` is
|
|
enabled, group messages are dropped unless they hit a wake-word pattern,
|
|
and the leading wake word is stripped from the ones that pass. DMs are
|
|
never gated.
|
|
|
|
These call ``_dispatch_inbound`` directly (no aiohttp / ports) and assert
|
|
on what reaches ``handle_message``.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
from typing import List
|
|
|
|
import pytest
|
|
|
|
from gateway.config import PlatformConfig
|
|
from gateway.platforms.base import MessageEvent
|
|
from plugins.platforms.photon.adapter import PhotonAdapter
|
|
|
|
|
|
def _make_adapter(monkeypatch: pytest.MonkeyPatch, extra: dict | None = None) -> PhotonAdapter:
|
|
monkeypatch.setenv("PHOTON_PROJECT_ID", "test-project-id")
|
|
monkeypatch.setenv("PHOTON_PROJECT_SECRET", "test-project-secret")
|
|
monkeypatch.delenv("PHOTON_REQUIRE_MENTION", raising=False)
|
|
monkeypatch.delenv("PHOTON_MENTION_PATTERNS", raising=False)
|
|
cfg = PlatformConfig(enabled=True, token="", extra=extra or {})
|
|
return PhotonAdapter(cfg)
|
|
|
|
|
|
def _group_payload(text: str) -> dict:
|
|
return {
|
|
"messageId": f"grp-{abs(hash(text))}",
|
|
"space": {"id": "group-guid-xyz", "type": "group", "phone": None},
|
|
"sender": {"id": "+15551234567"},
|
|
"content": {"type": "text", "text": text},
|
|
"timestamp": "2026-05-14T19:06:32.000Z",
|
|
}
|
|
|
|
|
|
def _dm_payload(text: str) -> dict:
|
|
return {
|
|
"messageId": f"dm-{abs(hash(text))}",
|
|
"space": {"id": "+15551234567", "type": "dm", "phone": "+15551234567"},
|
|
"sender": {"id": "+15551234567"},
|
|
"content": {"type": "text", "text": text},
|
|
"timestamp": "2026-05-14T19:06:32.000Z",
|
|
}
|
|
|
|
|
|
def _capture(adapter: PhotonAdapter, monkeypatch: pytest.MonkeyPatch) -> List[MessageEvent]:
|
|
captured: List[MessageEvent] = []
|
|
|
|
async def fake_handle(event: MessageEvent) -> None:
|
|
captured.append(event)
|
|
|
|
monkeypatch.setattr(adapter, "handle_message", fake_handle)
|
|
return captured
|
|
|
|
|
|
def test_require_mention_defaults_off(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(monkeypatch)
|
|
assert adapter.require_mention is False
|
|
# Defaults compile to the two Hermes wake-word patterns.
|
|
assert len(adapter._mention_patterns) == 2
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_group_message_dropped_without_mention(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(monkeypatch, extra={"require_mention": True})
|
|
captured = _capture(adapter, monkeypatch)
|
|
|
|
await adapter._dispatch_inbound(_group_payload("just chatting, no wake word"))
|
|
assert captured == []
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_group_message_passes_and_strips_wake_word(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(monkeypatch, extra={"require_mention": True})
|
|
captured = _capture(adapter, monkeypatch)
|
|
|
|
await adapter._dispatch_inbound(_group_payload("Hermes what's the weather"))
|
|
assert len(captured) == 1
|
|
# Leading wake word stripped before dispatch.
|
|
assert captured[0].text == "what's the weather"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_dm_never_gated(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(monkeypatch, extra={"require_mention": True})
|
|
captured = _capture(adapter, monkeypatch)
|
|
|
|
await adapter._dispatch_inbound(_dm_payload("no wake word here"))
|
|
assert len(captured) == 1
|
|
assert captured[0].text == "no wake word here"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_require_mention_off_passes_group_messages(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(monkeypatch) # require_mention defaults off
|
|
captured = _capture(adapter, monkeypatch)
|
|
|
|
await adapter._dispatch_inbound(_group_payload("plain group chatter"))
|
|
assert len(captured) == 1
|
|
assert captured[0].text == "plain group chatter"
|
|
|
|
|
|
def test_custom_mention_patterns_from_config(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(
|
|
monkeypatch,
|
|
extra={"require_mention": True, "mention_patterns": [r"(?<![\w@])@?amos\b[,:\-]?"]},
|
|
)
|
|
assert adapter.require_mention is True
|
|
assert len(adapter._mention_patterns) == 1
|
|
assert adapter._message_matches_mention_patterns("amos help me") is True
|
|
assert adapter._message_matches_mention_patterns("hermes help me") is False
|
|
|
|
|
|
def test_mention_patterns_env_comma_separated(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
monkeypatch.setenv("PHOTON_PROJECT_ID", "test-project-id")
|
|
monkeypatch.setenv("PHOTON_PROJECT_SECRET", "test-project-secret")
|
|
monkeypatch.setenv("PHOTON_REQUIRE_MENTION", "true")
|
|
monkeypatch.setenv("PHOTON_MENTION_PATTERNS", r"bot\b, assistant\b")
|
|
cfg = PlatformConfig(enabled=True, token="", extra={})
|
|
adapter = PhotonAdapter(cfg)
|
|
assert adapter.require_mention is True
|
|
assert len(adapter._mention_patterns) == 2
|
|
assert adapter._message_matches_mention_patterns("hey bot") is True
|
|
|
|
|
|
def test_invalid_pattern_skipped(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
adapter = _make_adapter(
|
|
monkeypatch,
|
|
extra={"require_mention": True, "mention_patterns": ["(unclosed", r"good\b"]},
|
|
)
|
|
# Bad regex dropped, good one kept.
|
|
assert len(adapter._mention_patterns) == 1
|
|
assert adapter._message_matches_mention_patterns("a good thing") is True
|