refactor: remove persist_output_message functionality and related database save logic

tests/test_message_notifications.py

@@ -0,0 +1,169 @@
+import json
+from unittest.mock import Mock
+
+from app.db import SessionFactory
+from app.db.message_oper import MessageOper
+from app.db.models.message import Message
+from app.chain import ChainBase
+from app.helper.message import MessageHelper
+from app.schemas import Notification
+from app.schemas.types import NotificationType
+
+
+def _clear_messages() -> None:
+    """
+    清空消息表，隔离通知测试数据。
+    """
+    with SessionFactory() as db:
+        db.query(Message).delete()
+        db.commit()
+
+
+def _reset_message_helper(helper: MessageHelper) -> None:
+    """
+    清空单例消息队列和去重缓存，避免用例间互相影响。
+    """
+    while helper.get() is not None:
+        pass
+    helper._recent_notification_keys.clear()
+
+
+def test_notification_history_only_lists_sent_messages() -> None:
+    """
+    通知历史应返回已发送消息，包含通过消息链登记的智能体消息。
+    """
+    _clear_messages()
+    oper = MessageOper()
+    oper.add(title="系统通知", text="下载完成", action=1, mtype=NotificationType.Download)
+    oper.add(title="用户消息", text="帮我搜索", action=0)
+    oper.add(title="智能体回复", text="已处理", action=1, mtype=NotificationType.Agent)
+
+    messages = MessageOper().list_by_page(page=1, count=10)
+    assert [message.title for message in messages if message.action == 1] == ["智能体回复", "系统通知"]
+
+
+def test_web_message_history_returns_all_messages() -> None:
+    """
+    Web 消息历史返回消息表中的全部记录。
+    """
+    _clear_messages()
+    oper = MessageOper()
+    oper.add(title="智能体回复", text="已处理", action=1, mtype=NotificationType.Agent)
+    oper.add(title="用户消息", text="/ai 帮我处理", action=0)
+    oper.add(title="普通通知", text="下载完成", action=1, mtype=NotificationType.Download)
+
+    messages = MessageOper().list_by_page(page=1, count=10)
+    assert [message.title for message in messages] == ["普通通知", "用户消息", "智能体回复"]
+
+
+def test_system_helper_message_only_enters_sse_queue() -> None:
+    """
+    系统实时消息只进入前端 SSE 队列，不写入通知历史。
+    """
+    _clear_messages()
+    helper = MessageHelper()
+    _reset_message_helper(helper)
+
+    helper.put("调度任务执行失败", role="system", title="系统错误")
+
+    assert MessageOper().list_by_page(page=1, count=10) == []
+    realtime_message = json.loads(helper.get())
+    assert realtime_message["type"] == "system"
+    assert realtime_message["title"] == "系统错误"
+    assert realtime_message["text"] == "调度任务执行失败"
+
+
+def test_plugin_helper_message_deduplicates_recent_sse_messages() -> None:
+    """
+    短时间内相同插件实时消息只应推送一次，不写入通知历史。
+    """
+    _clear_messages()
+    helper = MessageHelper()
+    _reset_message_helper(helper)
+
+    helper.put("站点刷流任务出错，获取下载器实例失败，请检查配置", role="plugin", title="站点刷流")
+    helper.put("站点刷流任务出错，获取下载器实例失败，请检查配置", role="plugin", title="站点刷流")
+
+    assert MessageOper().list_by_page(page=1, count=10) == []
+    assert json.loads(helper.get())["title"] == "站点刷流"
+    assert helper.get() is None
+
+
+def test_agent_helper_message_does_not_enter_sse_queue() -> None:
+    """
+    智能体消息不进入前端 SSE 队列。
+    """
+    helper = MessageHelper()
+    _reset_message_helper(helper)
+
+    helper.put("智能体回复", role="agent", title="MoviePilot助手")
+
+    assert helper.get() is None
+
+
+def test_user_helper_message_does_not_enter_sse_queue() -> None:
+    """
+    用户消息不进入前端 SSE 队列。
+    """
+    helper = MessageHelper()
+    _reset_message_helper(helper)
+
+    helper.put("用户输入", role="user", title="admin")
+
+    assert helper.get() is None
+
+
+def test_notification_post_message_is_persisted_without_sse_queue() -> None:
+    """
+    业务通知通过消息链发送时只登记数据库，不进入前端 SSE 队列。
+    """
+    _clear_messages()
+    helper = MessageHelper()
+    _reset_message_helper(helper)
+    chain = ChainBase()
+
+    chain.messagequeue.send_message = Mock()
+    chain.eventmanager.send_event = Mock()
+
+    chain.post_message(
+        Notification(
+            mtype=NotificationType.Download,
+            title="下载完成",
+            text="影片已加入下载器",
+        )
+    )
+
+    messages = MessageOper().list_by_page(page=1, count=10)
+    assert len(messages) == 1
+    assert messages[0].title == "下载完成"
+    assert messages[0].mtype == NotificationType.Download.value
+    assert helper.get() is None
+    chain.messagequeue.send_message.assert_called_once()
+
+
+def test_agent_notification_post_message_is_persisted_without_sse_queue() -> None:
+    """
+    智能体消息通过消息链发送时登记数据库，但不进入前端 SSE 队列。
+    """
+    _clear_messages()
+    helper = MessageHelper()
+    _reset_message_helper(helper)
+    chain = ChainBase()
+
+    chain.messagequeue.send_message = Mock()
+    chain.eventmanager.send_event = Mock()
+
+    chain.post_message(
+        Notification(
+            mtype=NotificationType.Agent,
+            title="MoviePilot助手",
+            text="已完成处理",
+        )
+    )
+
+    messages = MessageOper().list_by_page(page=1, count=10)
+    assert len(messages) == 1
+    assert messages[0].title == "MoviePilot助手"
+    assert messages[0].mtype == NotificationType.Agent.value
+    assert helper.get() is None
+    chain.messagequeue.send_message.assert_called_once()