"""测试 memory_prompts.inject_image_placeholder_template：占位符花括号统一为四层，避免多余花括号残留"""

import unittest

from app.agents.memoir.prompts import (
    IMAGE_PLACEHOLDER_TEMPLATE,
    inject_image_placeholder_template,
)


class InjectImagePlaceholderTemplateTest(unittest.TestCase):
    def test_normalizes_double_brace_to_four(self):
        content = "段落。\n\n{{IMAGE:南方小镇的青石板路}}\n\n结尾。"
        out = inject_image_placeholder_template(content)
        self.assertIn("{{{{IMAGE:", out)
        self.assertIn("}}}}", out)
        # 应为四层占位符，且「结尾。」前不应有多余的 }}
        self.assertIn("\n\n结尾。", out)
        self.assertEqual(out.count("}}}}"), 1)

    def test_normalizes_quad_brace_unchanged(self):
        content = "段落。\n\n{{{{IMAGE:南方小镇的青石板路}}}}\n\n结尾。"
        out = inject_image_placeholder_template(content)
        self.assertEqual(out.count("{{{{"), out.count("}}}}"))
        self.assertIn("{{{{IMAGE:", out)
        self.assertNotRegex(out, r"\}\}\}\}\}\}")  # 不应出现五层及以上闭合括号

    def test_normalizes_six_braces_to_four_so_no_residue(self):
        # LLM 有时会多打花括号，导致客户端按四层占位符 split 后残留 "{{" "}}" 被显示
        content = "段落。\n\n{{{{{{IMAGE:南方小镇的青石板路}}}}}}\n\n结尾。"
        out = inject_image_placeholder_template(content)
        # 整段应被替换为四层，不应留下多余的 "{{" 或 "}}"
        self.assertIn("{{{{IMAGE:" + IMAGE_PLACEHOLDER_TEMPLATE, out)
        self.assertIn("}}}}", out)
        self.assertNotIn("{{{{{{", out)
        self.assertNotIn("}}}}}}", out)
        # 正文前后不应出现裸花括号
        parts = out.split("}}}}")
        for i, p in enumerate(parts):
            if "南方小镇" in p or i == 0:
                continue
            self.assertNotRegex(p, r"^\s*\{\{", msg=f"残留开括号 in part: {p!r}")
        before, sep, after = out.partition("{{{{IMAGE:")
        self.assertNotRegex(before, r"\{\{\s*$", msg="开头段不应以 {{ 结尾")
        self.assertNotRegex(after, r"^\s*\}\}", msg="占位符后不应以 }} 开头")

    def test_normalizes_eight_braces_to_four(self):
        content = "前\n\n{{{{{{{{IMAGE:奶奶的藤椅}}}}}}}}\n\n后"
        out = inject_image_placeholder_template(content)
        self.assertIn("{{{{IMAGE:", out)
        self.assertNotIn("{{{{{{{{", out)
        self.assertNotIn("}}}}}}}}", out)