Fix memoir image delivery and Android rendering

2026-03-11 10:06:12 +08:00
parent 0970cb7408
commit a76cf8da18
23 changed files with 537 additions and 51 deletions
--- a/api/tests/test_chapters_router_images.py
+++ b/api/tests/test_chapters_router_images.py
@@ -0,0 +1,58 @@
+import os
+import unittest
+from unittest.mock import Mock, patch
+
+from api.routers.chapters import _chapter_to_dict
+
+
+class ChaptersRouterImagesTest(unittest.TestCase):
+    @patch("api.routers.chapters.TencentCosStorageService")
+    @patch.dict(
+        os.environ,
+        {
+            "TENCENT_COS_BUCKET": "life-echo-dev-1319381411",
+            "TENCENT_COS_REGION": "ap-shanghai",
+            "TENCENT_COS_BASE_URL": "https://life-echo-dev-1319381411.cos.ap-shanghai.myqcloud.com",
+        },
+        clear=False,
+    )
+    def test_chapter_to_dict_returns_signed_image_urls_for_response(self, storage_cls):
+        storage = Mock()
+        storage.get_download_url.return_value = "https://signed.example.com/memoirs/u1/c1/0-demo.png?sig=123"
+        storage_cls.from_env.return_value = storage
+
+        chapter = type(
+            "ChapterStub",
+            (),
+            {
+                "id": "chapter-1",
+                "title": "童年的夏天",
+                "content": "{{IMAGE:南方小镇的青石板路}}",
+                "order_index": 0,
+                "status": "completed",
+                "category": "childhood",
+                "images": [
+                    {
+                        "index": 0,
+                        "placeholder": "{{IMAGE:南方小镇的青石板路}}",
+                        "description": "南方小镇的青石板路",
+                        "status": "completed",
+                        "prompt": "A serene southern China town",
+                        "url": "https://life-echo-dev-1319381411.cos.ap-shanghai.myqcloud.com/memoirs/u1/c1/0-demo.png",
+                        "storage_key": "memoirs/u1/c1/0-demo.png",
+                    }
+                ],
+                "updated_at": None,
+                "is_new": False,
+                "source_segments": [],
+            },
+        )()
+
+        payload = _chapter_to_dict(chapter)
+
+        self.assertEqual(
+            payload["images"][0]["url"],
+            "https://signed.example.com/memoirs/u1/c1/0-demo.png?sig=123",
+        )
+        self.assertEqual(payload["images"][0]["prompt"], "A serene southern China town")
+        self.assertNotIn("storage_key", payload["images"][0])