92 lines
3.2 KiB
Python
92 lines
3.2 KiB
Python
"""asset_resolver：旧占位符清理与 asset:// 解析。"""
|
||
|
||
import unittest
|
||
from types import SimpleNamespace
|
||
|
||
from app.features.memoir.asset_resolver import (
|
||
collect_asset_ids_for_chapter,
|
||
collect_asset_ids_from_markdown,
|
||
resolve_asset_refs_in_markdown,
|
||
split_markdown_by_asset_refs,
|
||
strip_asset_image_refs_from_markdown,
|
||
strip_legacy_image_placeholders,
|
||
)
|
||
from app.features.memoir.models import Chapter
|
||
|
||
|
||
class AssetResolverTest(unittest.TestCase):
|
||
def test_strip_legacy_image_placeholders_double_brace(self):
|
||
md = "正文\n\n{{IMAGE:院子里的树}}\n\n结尾"
|
||
out = strip_legacy_image_placeholders(md)
|
||
self.assertNotIn("IMAGE", out)
|
||
self.assertIn("正文", out)
|
||
self.assertIn("结尾", out)
|
||
|
||
def test_strip_legacy_image_placeholders_quad_brace(self):
|
||
md = "a\n\n{{{{IMAGE:描述}}}}\n\nb"
|
||
out = strip_legacy_image_placeholders(md)
|
||
self.assertNotIn("IMAGE", out)
|
||
|
||
def test_collect_and_split_asset_refs(self):
|
||
md = "前\n\n\n\n后"
|
||
self.assertEqual(collect_asset_ids_from_markdown(md), ["abc-123"])
|
||
blocks = split_markdown_by_asset_refs(md, lambda aid: f"https://x/{aid}")
|
||
self.assertEqual(len(blocks), 3)
|
||
self.assertEqual(blocks[0]["type"], "text")
|
||
self.assertEqual(blocks[1]["type"], "image")
|
||
self.assertIn("https://x/abc-123", blocks[1]["url"])
|
||
|
||
def test_resolve_asset_refs_in_markdown(self):
|
||
md = ""
|
||
out = resolve_asset_refs_in_markdown(md, lambda aid: "https://cdn/u")
|
||
self.assertIn("https://cdn/u", out)
|
||
self.assertNotIn("asset://", out)
|
||
|
||
def test_collect_asset_ids_for_chapter(self):
|
||
ch = Chapter(
|
||
id="c1",
|
||
user_id="u1",
|
||
title="t",
|
||
order_index=0,
|
||
canonical_markdown="",
|
||
cover_asset_id="cov1",
|
||
)
|
||
ids = collect_asset_ids_for_chapter(ch)
|
||
self.assertEqual(ids, {"a1", "cov1"})
|
||
|
||
def test_strip_asset_image_refs_removes_all_and_collapses_blank_lines(self):
|
||
md = (
|
||
"第一段\n\n\n\n第二段\n\n\n"
|
||
"\n\n第三段"
|
||
)
|
||
out = strip_asset_image_refs_from_markdown(md)
|
||
self.assertNotIn("asset://", out)
|
||
self.assertIn("第一段", out)
|
||
self.assertIn("第二段", out)
|
||
self.assertIn("第三段", out)
|
||
self.assertNotIn("\n\n\n", out)
|
||
|
||
def test_strip_asset_image_refs_empty(self):
|
||
self.assertEqual(strip_asset_image_refs_from_markdown(""), "")
|
||
self.assertEqual(strip_asset_image_refs_from_markdown(" "), "")
|
||
|
||
def test_collect_asset_ids_includes_linked_story_markdown(self):
|
||
ch = SimpleNamespace(
|
||
canonical_markdown="",
|
||
sections=[],
|
||
cover_asset_id=None,
|
||
story_links=[
|
||
SimpleNamespace(
|
||
story=SimpleNamespace(
|
||
canonical_markdown=""
|
||
)
|
||
)
|
||
],
|
||
)
|
||
ids = collect_asset_ids_for_chapter(ch)
|
||
self.assertEqual(ids, {"from-story-1"})
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|