2eb066dbec
* refactor: 表结构重构，新增段落section和图片image新表 * fix: fix android app import error * refactor: 重构文件名 * fix: 优化提示词 * fix: 消息气泡显示位置异常问题 --------- Co-authored-by: yangshilin <2157598560@qq.com>
100 lines
4.3 KiB
Python
100 lines
4.3 KiB
Python
import os
|
||
import unittest
|
||
import unittest.mock
|
||
|
||
from api.tasks.memoir_tasks import initialize_chapter_images
|
||
|
||
|
||
class MemoirImageBootstrapTest(unittest.TestCase):
|
||
def test_initialize_chapter_images_keeps_only_completed_assets_when_disabled(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op，直接返回 []"""
|
||
chapter = type(
|
||
"ChapterStub",
|
||
(),
|
||
{
|
||
"id": "chapter-1",
|
||
"title": "童年的夏天",
|
||
"category": "childhood",
|
||
},
|
||
)()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "false"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_sets_pending_assets_when_enabled(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type(
|
||
"ChapterStub",
|
||
(),
|
||
{
|
||
"id": "chapter-1",
|
||
"title": "童年的夏天",
|
||
"category": "childhood",
|
||
},
|
||
)()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "true"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_preserves_completed_assets_and_adds_only_new_placeholders(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type("ChapterStub", (), {"id": "chapter-1", "title": "童年的夏天", "category": "childhood"})()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "true"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_accepts_double_brace_placeholders(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type("ChapterStub", (), {"id": "chapter-1", "title": "童年的夏天", "category": "childhood"})()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "true"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_normalizes_invalid_existing_asset_status(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type("ChapterStub", (), {"id": "chapter-1", "title": "童年的夏天", "category": "childhood"})()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "true"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_preserves_existing_completed_assets_beyond_effective_max(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type("ChapterStub", (), {"id": "chapter-1", "title": "童年的夏天", "category": "childhood"})()
|
||
|
||
with unittest.mock.patch.dict(
|
||
os.environ,
|
||
{"MEMOIR_IMAGE_ENABLED": "true", "MEMOIR_IMAGE_MAX_PER_CHAPTER": "2"},
|
||
clear=False,
|
||
):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_increases_limit_for_long_content(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type("ChapterStub", (), {"id": "chapter-1", "title": "童年的夏天", "category": "childhood"})()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "true"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|
||
|
||
def test_initialize_chapter_images_caps_dynamic_limit_at_max_images_cap(self):
|
||
"""图片初始化已迁移到 _save_narrative_to_sections；此处为兼容 no-op"""
|
||
chapter = type("ChapterStub", (), {"id": "chapter-1", "title": "童年的夏天", "category": "childhood"})()
|
||
|
||
with unittest.mock.patch.dict(os.environ, {"MEMOIR_IMAGE_ENABLED": "true"}, clear=False):
|
||
assets = initialize_chapter_images(chapter)
|
||
|
||
self.assertEqual(assets, [])
|