149 lines
5.3 KiB
Python
149 lines
5.3 KiB
Python
# tests/core/test_integration.py
|
|
"""
|
|
Smoke test: полный цикл через dispatcher + реальные managers + MockPlatformClient.
|
|
Имитирует что делает адаптер (Telegram или Matrix) при получении события.
|
|
"""
|
|
import pytest
|
|
from sdk.mock import MockPlatformClient
|
|
from sdk.agent_session import build_thread_key
|
|
from sdk.interface import MessageChunk, MessageResponse
|
|
from sdk.prototype_state import PrototypeStateStore
|
|
from sdk.real import RealPlatformClient
|
|
from core.store import InMemoryStore
|
|
from core.chat import ChatManager
|
|
from core.auth import AuthManager
|
|
from core.settings import SettingsManager
|
|
from core.handler import EventDispatcher
|
|
from core.handlers import register_all
|
|
from core.protocol import (
|
|
IncomingCommand, IncomingMessage, IncomingCallback,
|
|
OutgoingMessage, OutgoingUI,
|
|
Attachment, SettingsAction,
|
|
)
|
|
|
|
|
|
class FakeAgentSessionClient:
|
|
def __init__(self) -> None:
|
|
self.send_calls: list[tuple[str, str]] = []
|
|
|
|
async def send_message(self, *, thread_key: str, text: str) -> MessageResponse:
|
|
self.send_calls.append((thread_key, text))
|
|
return MessageResponse(
|
|
message_id=thread_key,
|
|
response=f"[REAL] {text}",
|
|
tokens_used=5,
|
|
finished=True,
|
|
)
|
|
|
|
async def stream_message(self, *, thread_key: str, text: str):
|
|
self.send_calls.append((thread_key, text))
|
|
if False:
|
|
yield MessageChunk(
|
|
message_id=thread_key,
|
|
delta=text,
|
|
tokens_used=0,
|
|
finished=True,
|
|
)
|
|
|
|
|
|
@pytest.fixture
|
|
def dispatcher():
|
|
platform = MockPlatformClient()
|
|
store = InMemoryStore()
|
|
d = EventDispatcher(
|
|
platform=platform,
|
|
chat_mgr=ChatManager(platform, store),
|
|
auth_mgr=AuthManager(platform, store),
|
|
settings_mgr=SettingsManager(platform, store),
|
|
)
|
|
register_all(d)
|
|
return d
|
|
|
|
|
|
@pytest.fixture
|
|
def real_dispatcher():
|
|
agent_sessions = FakeAgentSessionClient()
|
|
platform = RealPlatformClient(
|
|
agent_sessions=agent_sessions,
|
|
prototype_state=PrototypeStateStore(),
|
|
platform="matrix",
|
|
)
|
|
store = InMemoryStore()
|
|
d = EventDispatcher(
|
|
platform=platform,
|
|
chat_mgr=ChatManager(platform, store),
|
|
auth_mgr=AuthManager(platform, store),
|
|
settings_mgr=SettingsManager(platform, store),
|
|
)
|
|
register_all(d)
|
|
return d, agent_sessions
|
|
|
|
|
|
async def test_full_flow_start_then_message(dispatcher):
|
|
start = IncomingCommand(user_id="tg_123", platform="telegram", chat_id="C1", command="start")
|
|
result = await dispatcher.dispatch(start)
|
|
assert any(isinstance(r, OutgoingMessage) for r in result)
|
|
|
|
msg = IncomingMessage(user_id="tg_123", platform="telegram", chat_id="C1", text="Привет!")
|
|
result = await dispatcher.dispatch(msg)
|
|
texts = [r.text for r in result if isinstance(r, OutgoingMessage)]
|
|
assert any("[MOCK]" in t for t in texts)
|
|
|
|
|
|
async def test_new_chat_command(dispatcher):
|
|
start = IncomingCommand(user_id="u1", platform="matrix", chat_id="C1", command="start")
|
|
await dispatcher.dispatch(start)
|
|
|
|
new = IncomingCommand(user_id="u1", platform="matrix", chat_id="C2", command="new", args=["Анализ"])
|
|
result = await dispatcher.dispatch(new)
|
|
assert any("Анализ" in r.text for r in result if isinstance(r, OutgoingMessage))
|
|
|
|
|
|
async def test_settings_menu(dispatcher):
|
|
start = IncomingCommand(user_id="u1", platform="telegram", chat_id="C1", command="start")
|
|
await dispatcher.dispatch(start)
|
|
|
|
s = IncomingCommand(user_id="u1", platform="telegram", chat_id="C1", command="settings")
|
|
result = await dispatcher.dispatch(s)
|
|
assert any(isinstance(r, OutgoingUI) for r in result)
|
|
|
|
|
|
async def test_voice_message_fallback(dispatcher):
|
|
start = IncomingCommand(user_id="u1", platform="telegram", chat_id="C1", command="start")
|
|
await dispatcher.dispatch(start)
|
|
|
|
voice = IncomingMessage(
|
|
user_id="u1", platform="telegram", chat_id="C1", text="",
|
|
attachments=[Attachment(type="audio")],
|
|
)
|
|
result = await dispatcher.dispatch(voice)
|
|
assert any("голосов" in r.text.lower() for r in result if isinstance(r, OutgoingMessage))
|
|
|
|
|
|
async def test_toggle_skill_callback(dispatcher):
|
|
start = IncomingCommand(user_id="u1", platform="telegram", chat_id="C1", command="start")
|
|
await dispatcher.dispatch(start)
|
|
|
|
cb = IncomingCallback(
|
|
user_id="u1", platform="telegram", chat_id="C1",
|
|
action="toggle_skill", payload={"skill": "browser", "enabled": True},
|
|
)
|
|
result = await dispatcher.dispatch(cb)
|
|
assert any("browser" in r.text for r in result if isinstance(r, OutgoingMessage))
|
|
|
|
|
|
async def test_full_flow_with_real_platform_uses_thread_key(real_dispatcher):
|
|
dispatcher, agent_sessions = real_dispatcher
|
|
|
|
start = IncomingCommand(user_id="u1", platform="matrix", chat_id="C1", command="start")
|
|
result = await dispatcher.dispatch(start)
|
|
assert any(isinstance(r, OutgoingMessage) for r in result)
|
|
|
|
msg = IncomingMessage(user_id="u1", platform="matrix", chat_id="C1", text="Привет!")
|
|
result = await dispatcher.dispatch(msg)
|
|
texts = [r.text for r in result if isinstance(r, OutgoingMessage)]
|
|
|
|
assert texts == ["[REAL] Привет!"]
|
|
assert agent_sessions.send_calls == [
|
|
(build_thread_key("matrix", "u1", "C1"), "Привет!")
|
|
]
|