feat: 增强日历功能并集成 AgentScope 代理服务
This commit is contained in:
@@ -56,14 +56,14 @@ async def test_list_by_recipient_returns_messages() -> None:
|
||||
execute_result.scalars.return_value.all.return_value = [message_one, message_two]
|
||||
session.execute.return_value = execute_result
|
||||
|
||||
result = await repository.list_by_recipient(uuid4(), "pending")
|
||||
result = await repository.list_by_recipient(uuid4(), False)
|
||||
|
||||
assert result == [message_one, message_two]
|
||||
session.execute.assert_awaited_once()
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_update_status_returns_updated_message_and_flushes() -> None:
|
||||
async def test_mark_as_read_returns_updated_message_and_flushes() -> None:
|
||||
session = AsyncMock()
|
||||
repository = SQLAlchemyInboxMessageRepository(session)
|
||||
updated = MagicMock()
|
||||
@@ -71,7 +71,7 @@ async def test_update_status_returns_updated_message_and_flushes() -> None:
|
||||
execute_result.scalar_one_or_none.return_value = updated
|
||||
session.execute.return_value = execute_result
|
||||
|
||||
result = await repository.update_status(uuid4(), uuid4(), "dismissed")
|
||||
result = await repository.mark_as_read(uuid4(), uuid4())
|
||||
|
||||
assert result is updated
|
||||
session.execute.assert_awaited_once()
|
||||
|
||||
Reference in New Issue
Block a user