4003840378
`GatewayRunner` never had a `get_adapter` method, so every gateway `/save` (Telegram, Discord, ...) rendered the file and then failed with "'GatewayRunner' object has no attribute 'get_adapter'". Resolve the adapter through `_adapter_for_source`, the profile-aware lookup the rest of the runner uses, so multiplex secondaries deliver through their own bot rather than a missing key on the default map. Co-authored-by: pierrenode <298902573+pierrenode@users.noreply.github.com> Co-authored-by: KoNit-K <124019182+KoNit-K@users.noreply.github.com> Co-authored-by: Baophan00 <109447498+Baophan00@users.noreply.github.com>
54 lines
2.1 KiB
Python
54 lines
2.1 KiB
Python
"""Gateway /save: the export document is delivered through the requester's live adapter."""
|
|
|
|
import asyncio
|
|
from datetime import datetime
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
from gateway.config import Platform
|
|
from gateway.platforms.event import MessageEvent
|
|
from gateway.run import GatewayRunner
|
|
from gateway.session import SessionEntry, SessionSource, build_session_key
|
|
from hermes_state import AsyncSessionDB
|
|
|
|
|
|
def _runner(entry, adapters):
|
|
runner = object.__new__(GatewayRunner)
|
|
runner.adapters = adapters
|
|
runner._profile_adapters = {}
|
|
runner.session_store = MagicMock()
|
|
runner.session_store.get_or_create_session.return_value = entry
|
|
runner._session_db = AsyncSessionDB(MagicMock())
|
|
runner._session_db._db.export_session.return_value = {
|
|
"id": "sess-1", "source": "telegram",
|
|
"messages": [{"role": "user", "content": "hi"}, {"role": "assistant", "content": "hello"}],
|
|
}
|
|
return runner
|
|
|
|
|
|
def _save(runner, platform):
|
|
source = SessionSource(platform=platform, user_id="u1", chat_id="c1", user_name="t", chat_type="dm")
|
|
event = MessageEvent(text="/save md save-stuff.md", source=source, message_id="m1")
|
|
return asyncio.run(runner._handle_save_command(event))
|
|
|
|
|
|
def _entry(platform):
|
|
source = SessionSource(platform=platform, user_id="u1", chat_id="c1", user_name="t", chat_type="dm")
|
|
return SessionEntry(session_key=build_session_key(source), session_id="sess-1", created_at=datetime.now(),
|
|
updated_at=datetime.now(), platform=platform, chat_type="dm")
|
|
|
|
|
|
def test_save_sends_document_through_requesting_platform_adapter():
|
|
adapter = MagicMock()
|
|
adapter.send_document = AsyncMock()
|
|
runner = _runner(_entry(Platform.TELEGRAM), {Platform.TELEGRAM: adapter})
|
|
|
|
assert _save(runner, Platform.TELEGRAM) == "Export complete."
|
|
kwargs = adapter.send_document.await_args.kwargs
|
|
assert (kwargs["chat_id"], kwargs["file_name"]) == ("c1", "save-stuff.md")
|
|
|
|
|
|
def test_save_without_live_adapter_reports_missing_adapter_not_a_crash():
|
|
runner = _runner(_entry(Platform.DISCORD), {})
|
|
|
|
assert _save(runner, Platform.DISCORD) == "Platform adapter not found to send the document."
|