Mikhail Putilovskij
97a3dc35ea
- add MAT-01..MAT-07 and MAT-09..MAT-12 regression tests for matrix adapter - extend store and dispatcher coverage for pending confirmations and settings dashboard - verify matrix adapter suite and full pytest suite stay green
125 lines
4.3 KiB
Python
125 lines
4.3 KiB
Python
from __future__ import annotations
|
||
|
||
from types import SimpleNamespace
|
||
from unittest.mock import AsyncMock
|
||
|
||
from nio.responses import RoomCreateError
|
||
|
||
from adapter.matrix.handlers.chat import make_handle_archive, make_handle_new_chat
|
||
from adapter.matrix.store import set_user_meta
|
||
from core.auth import AuthManager
|
||
from core.chat import ChatManager
|
||
from core.protocol import IncomingCommand, OutgoingMessage
|
||
from core.settings import SettingsManager
|
||
from core.store import InMemoryStore
|
||
from sdk.mock import MockPlatformClient
|
||
|
||
|
||
async def _setup():
|
||
platform = MockPlatformClient()
|
||
store = InMemoryStore()
|
||
chat_mgr = ChatManager(platform, store)
|
||
auth_mgr = AuthManager(platform, store)
|
||
settings_mgr = SettingsManager(platform, store)
|
||
await auth_mgr.confirm("@alice:example.org")
|
||
return platform, store, chat_mgr, auth_mgr, settings_mgr
|
||
|
||
|
||
async def test_mat04_new_chat_calls_room_put_state_with_space_id():
|
||
platform, store, chat_mgr, auth_mgr, settings_mgr = await _setup()
|
||
await set_user_meta(store, "@alice:example.org", {"space_id": "!space:ex", "next_chat_index": 2})
|
||
|
||
client = SimpleNamespace(
|
||
room_create=AsyncMock(return_value=SimpleNamespace(room_id="!newroom:ex")),
|
||
room_put_state=AsyncMock(),
|
||
room_invite=AsyncMock(),
|
||
)
|
||
handler = make_handle_new_chat(client, store)
|
||
event = IncomingCommand(
|
||
user_id="@alice:example.org",
|
||
platform="matrix",
|
||
chat_id="C1",
|
||
command="new",
|
||
args=["Test"],
|
||
)
|
||
result = await handler(event, auth_mgr, platform, chat_mgr, settings_mgr)
|
||
|
||
client.room_put_state.assert_awaited_once()
|
||
kwargs = client.room_put_state.call_args.kwargs
|
||
assert kwargs.get("room_id") == "!space:ex"
|
||
assert kwargs.get("event_type") == "m.space.child"
|
||
assert kwargs.get("state_key") == "!newroom:ex"
|
||
assert any(isinstance(item, OutgoingMessage) and "Test" in item.text for item in result)
|
||
|
||
|
||
async def test_mat05_new_chat_without_space_id_returns_error():
|
||
platform, store, chat_mgr, auth_mgr, settings_mgr = await _setup()
|
||
await set_user_meta(store, "@alice:example.org", {"next_chat_index": 1})
|
||
|
||
client = SimpleNamespace(
|
||
room_create=AsyncMock(),
|
||
room_put_state=AsyncMock(),
|
||
room_invite=AsyncMock(),
|
||
)
|
||
handler = make_handle_new_chat(client, store)
|
||
event = IncomingCommand(
|
||
user_id="@alice:example.org",
|
||
platform="matrix",
|
||
chat_id="C1",
|
||
command="new",
|
||
)
|
||
result = await handler(event, auth_mgr, platform, chat_mgr, settings_mgr)
|
||
|
||
assert len(result) == 1
|
||
assert isinstance(result[0], OutgoingMessage)
|
||
assert "Space" in result[0].text or "ошибка" in result[0].text.lower()
|
||
client.room_create.assert_not_awaited()
|
||
|
||
|
||
async def test_mat10_archive_calls_chat_mgr_archive():
|
||
platform, store, chat_mgr, auth_mgr, settings_mgr = await _setup()
|
||
|
||
handler = make_handle_archive(None, store)
|
||
event = IncomingCommand(
|
||
user_id="@alice:example.org",
|
||
platform="matrix",
|
||
chat_id="C1",
|
||
command="archive",
|
||
)
|
||
await chat_mgr.get_or_create(
|
||
user_id="@alice:example.org",
|
||
chat_id="C1",
|
||
platform="matrix",
|
||
surface_ref="!room:ex",
|
||
name="Test",
|
||
)
|
||
|
||
result = await handler(event, auth_mgr, platform, chat_mgr, settings_mgr)
|
||
|
||
assert len(result) == 1
|
||
assert "архивирован" in result[0].text
|
||
|
||
|
||
async def test_mat12_room_create_error_returns_user_message():
|
||
platform, store, chat_mgr, auth_mgr, settings_mgr = await _setup()
|
||
await set_user_meta(store, "@alice:example.org", {"space_id": "!space:ex", "next_chat_index": 2})
|
||
|
||
client = SimpleNamespace(
|
||
room_create=AsyncMock(return_value=RoomCreateError(message="rate limited", status_code="429")),
|
||
room_put_state=AsyncMock(),
|
||
room_invite=AsyncMock(),
|
||
)
|
||
handler = make_handle_new_chat(client, store)
|
||
event = IncomingCommand(
|
||
user_id="@alice:example.org",
|
||
platform="matrix",
|
||
chat_id="C1",
|
||
command="new",
|
||
args=["Fail"],
|
||
)
|
||
result = await handler(event, auth_mgr, platform, chat_mgr, settings_mgr)
|
||
|
||
assert len(result) == 1
|
||
assert isinstance(result[0], OutgoingMessage)
|
||
assert "Не удалось" in result[0].text or "не удалось" in result[0].text
|
||
client.room_put_state.assert_not_awaited()
|