fix: normalize transfer event targets

2026-06-09 07:26:48 +00:00 · 2026-05-20 20:48:53 +08:00
parent ad7cce72f4
commit 5d34bc5c56
2 changed files with 193 additions and 0 deletions
--- a/tests/test_event_transfer_normalization.py
+++ b/tests/test_event_transfer_normalization.py
@@ -0,0 +1,89 @@
+import unittest
+from unittest.mock import patch
+
+from app.core.event import EventManager
+from app.schemas import FileItem, TransferInfo
+from app.schemas.types import EventType
+
+
+class EventTransferNormalizationTest(unittest.TestCase):
+    def test_transfer_event_fills_missing_target_items_before_dispatch(self):
+        """
+        整理事件投递给插件前，应补齐可读取 path 的目标文件和目标目录项。
+        """
+        event_manager = EventManager()
+        transferinfo = TransferInfo(
+            success=True,
+            fileitem=FileItem(
+                storage="alist",
+                path="/downloads/Test.Show.S01E01.mkv",
+                type="file",
+                name="Test.Show.S01E01.mkv",
+                size=1024,
+            ),
+            file_list_new=[
+                "/library/Test Show (2026)/Season 1/Test.Show.S01E01.mkv"
+            ],
+            transfer_type="move",
+        )
+        event_data = {"transferinfo": transferinfo}
+
+        with patch.object(
+                event_manager, "_EventManager__trigger_broadcast_event"
+        ):
+            event_manager.send_event(EventType.TransferComplete, event_data)
+
+        self.assertIsNotNone(transferinfo.target_item)
+        self.assertIsNotNone(transferinfo.target_diritem)
+        self.assertEqual(
+            "/library/Test Show (2026)/Season 1/Test.Show.S01E01.mkv",
+            transferinfo.target_item.path,
+        )
+        self.assertEqual(
+            "/library/Test Show (2026)/Season 1",
+            transferinfo.target_diritem.path,
+        )
+        self.assertEqual("alist", transferinfo.target_item.storage)
+        self.assertEqual("alist", transferinfo.target_diritem.storage)
+
+    def test_transfer_event_fills_missing_target_diritem_from_target_item(self):
+        """
+        目标文件项已存在但目录项缺失时，事件数据应补齐 target_diritem。
+        """
+        event_manager = EventManager()
+        transferinfo = TransferInfo(
+            success=True,
+            fileitem=FileItem(
+                storage="alist",
+                path="/downloads/Test.Show.S01E02.mkv",
+                type="file",
+                name="Test.Show.S01E02.mkv",
+            ),
+            target_item=FileItem(
+                storage="alist",
+                path="/library/Test Show (2026)/Season 1/Test.Show.S01E02.mkv",
+                type="file",
+                name="Test.Show.S01E02.mkv",
+            ),
+            file_list_new=[
+                "/library/Test Show (2026)/Season 1/Test.Show.S01E02.mkv"
+            ],
+            transfer_type="move",
+        )
+        event_data = {"transferinfo": transferinfo}
+
+        with patch.object(
+                event_manager, "_EventManager__trigger_broadcast_event"
+        ):
+            event_manager.send_event(EventType.TransferComplete, event_data)
+
+        self.assertIsNotNone(transferinfo.target_diritem)
+        self.assertEqual(
+            "/library/Test Show (2026)/Season 1",
+            transferinfo.target_diritem.path,
+        )
+        self.assertEqual("alist", transferinfo.target_diritem.storage)
+
+
+if __name__ == "__main__":
+    unittest.main()