- platform/interface.py: PlatformClient Protocol + Pydantic models (User, MessageResponse, UserSettings) — no explicit session management, Master handles container lifecycle - platform/mock.py: MockPlatformClient with simulated latency, [MOCK] responses, is_new correctly True only on first creation - core/protocol.py: unified dataclasses for all events and responses (IncomingMessage/Command/Callback, OutgoingMessage/UI/Notification, AuthFlow, ChatContext, SettingsAction, etc.) - core/store.py: StateStore Protocol + InMemoryStore (tests) + SQLiteStore (prod) with JSON serialization - core/chat.py: ChatManager — chat metadata (C1/C2/C3), not container lifecycle (that's the platform's job) - core/auth.py: AuthManager — start_flow / confirm / is_authenticated - core/settings.py: SettingsManager — get/apply with store cache - core/handler.py: EventDispatcher — registry-based routing with keys (command name, action name, attachment type, "*" catch-all) - core/handlers/: register_all() + start/new/message/callback/settings handlers; voice slot falls back to stub text until voice_handler added - conftest.py: sys.path fix so local platform/ shadows stdlib platform - docs/api-contract.md: rewritten for Lambda Lab 3.0 container model 46 tests passing, 0 warnings.
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
# tests/core/test_chat.py
|
|
import pytest
|
|
from core.chat import ChatManager
|
|
from core.store import InMemoryStore
|
|
from platform.mock import MockPlatformClient
|
|
|
|
|
|
@pytest.fixture
|
|
def mgr():
|
|
return ChatManager(MockPlatformClient(), InMemoryStore())
|
|
|
|
|
|
async def test_get_or_create_new_chat(mgr):
|
|
ctx = await mgr.get_or_create("u1", "C1", "telegram", "topic-123")
|
|
assert ctx.chat_id == "C1"
|
|
assert ctx.platform == "telegram"
|
|
assert ctx.is_archived is False
|
|
|
|
|
|
async def test_get_or_create_idempotent(mgr):
|
|
c1 = await mgr.get_or_create("u1", "C1", "telegram", "t1")
|
|
c2 = await mgr.get_or_create("u1", "C1", "telegram", "t1")
|
|
assert c1.chat_id == c2.chat_id
|
|
assert c1.display_name == c2.display_name
|
|
|
|
|
|
async def test_get_or_create_with_custom_name(mgr):
|
|
ctx = await mgr.get_or_create("u1", "C1", "telegram", "t1", name="Анализ рынка")
|
|
assert ctx.display_name == "Анализ рынка"
|
|
|
|
|
|
async def test_rename_chat(mgr):
|
|
await mgr.get_or_create("u1", "C1", "telegram", "t1")
|
|
ctx = await mgr.rename("C1", "Новое название")
|
|
assert ctx.display_name == "Новое название"
|
|
|
|
|
|
async def test_archive_chat(mgr):
|
|
await mgr.get_or_create("u1", "C1", "telegram", "t1")
|
|
await mgr.archive("C1")
|
|
ctx = await mgr.get("C1")
|
|
assert ctx is not None
|
|
assert ctx.is_archived is True
|
|
|
|
|
|
async def test_list_active_excludes_archived(mgr):
|
|
await mgr.get_or_create("u1", "C1", "telegram", "t1")
|
|
await mgr.get_or_create("u1", "C2", "telegram", "t2")
|
|
await mgr.archive("C2")
|
|
chats = await mgr.list_active("u1")
|
|
ids = [c.chat_id for c in chats]
|
|
assert "C1" in ids
|
|
assert "C2" not in ids
|