mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-19 15:18:03 +00:00
117 lines
3.4 KiB
Python
117 lines
3.4 KiB
Python
import sys
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
|
|
ROOT = Path(__file__).resolve().parents[2]
|
|
if str(ROOT) not in sys.path:
|
|
sys.path.insert(0, str(ROOT))
|
|
|
|
from gateway.config import Platform
|
|
from plugins.platforms.telegram.adapter import TelegramAdapter
|
|
from gateway.run import GatewayRunner
|
|
from gateway.session import SessionSource
|
|
|
|
|
|
def _source():
|
|
return SessionSource(platform=Platform.TELEGRAM, chat_id="12345", chat_type="dm")
|
|
|
|
|
|
def _runner(adapter=None):
|
|
runner = object.__new__(GatewayRunner)
|
|
runner.config = SimpleNamespace(
|
|
stt_enabled=True,
|
|
group_sessions_per_user=True,
|
|
thread_sessions_per_user=False,
|
|
)
|
|
runner.adapters = {Platform.TELEGRAM: adapter} if adapter else {}
|
|
runner._consume_pending_native_image_paths = lambda _key: []
|
|
runner._session_key_for_source = lambda _source: "telegram:dm:12345"
|
|
runner._thread_metadata_for_source = lambda *_args, **_kwargs: {}
|
|
runner._reply_anchor_for_event = lambda _event: None
|
|
return runner
|
|
|
|
|
|
def test_telegram_audio_size_gate_rejects_oversized_media_before_download():
|
|
adapter = object.__new__(TelegramAdapter)
|
|
adapter._max_doc_bytes = 1024
|
|
|
|
allowed, note = adapter._telegram_media_size_allowed(
|
|
SimpleNamespace(file_size=2048),
|
|
"voice message",
|
|
)
|
|
|
|
assert allowed is False
|
|
assert "exceeds" in note
|
|
assert "voice message" in note
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_telegram_video_size_gate_rejects_oversized_media_before_download():
|
|
adapter = object.__new__(TelegramAdapter)
|
|
adapter._max_doc_bytes = 1024
|
|
adapter._should_process_message = lambda _message: True
|
|
adapter._build_message_event = lambda _message, _type, update_id=None: SimpleNamespace(
|
|
text="caption",
|
|
media_urls=[],
|
|
media_types=[],
|
|
)
|
|
adapter._apply_telegram_group_observe_attribution = lambda event: event
|
|
|
|
handled = []
|
|
|
|
async def handle_message(event):
|
|
handled.append(event)
|
|
|
|
adapter.handle_message = handle_message
|
|
|
|
class OversizedVideo:
|
|
file_size = 2048
|
|
|
|
async def get_file(self): # pragma: no cover - failure path assertion
|
|
pytest.fail("oversized videos must not be downloaded")
|
|
|
|
msg = SimpleNamespace(
|
|
caption=None,
|
|
sticker=None,
|
|
photo=None,
|
|
voice=None,
|
|
audio=None,
|
|
video=OversizedVideo(),
|
|
document=None,
|
|
media_group_id=None,
|
|
)
|
|
update = SimpleNamespace(message=msg, update_id=1)
|
|
|
|
await TelegramAdapter._handle_media_message(adapter, update, SimpleNamespace())
|
|
|
|
assert len(handled) == 1
|
|
assert handled[0].media_urls == []
|
|
assert handled[0].media_types == []
|
|
assert "video file" in handled[0].text
|
|
assert "exceeds" in handled[0].text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_voice_tts_is_explicit_audio_reply_opt_in():
|
|
adapter = SimpleNamespace(
|
|
_auto_tts_disabled_chats=set(),
|
|
_auto_tts_enabled_chats=set(),
|
|
)
|
|
runner = _runner(adapter)
|
|
runner._voice_mode = {}
|
|
runner._voice_provider_mode = {}
|
|
runner._save_voice_modes = lambda: None
|
|
runner._save_voice_provider_modes = lambda: None
|
|
|
|
event = SimpleNamespace(
|
|
source=_source(),
|
|
get_command_args=lambda: "tts",
|
|
)
|
|
result = await GatewayRunner._handle_voice_command(runner, event)
|
|
|
|
assert runner._voice_mode["telegram:12345"] == "all"
|
|
assert "12345" in adapter._auto_tts_enabled_chats
|
|
assert result
|