"""快速新对话自动标题的纯函数边界。"""
from __future__ import annotations

import unittest

from core.task_title import clean_generated_title, is_attachment_only_message


class TaskTitleTests(unittest.TestCase):
    def test_strips_model_wrappers_and_forbidden_path_chars(self):
        self.assertEqual(
            clean_generated_title('标题：“低碳水泥/XRD 分析”', "原消息"),
            "低碳水泥·XRD 分析",
        )

    def test_empty_or_generic_output_falls_back_to_user_message(self):
        self.assertEqual(
            clean_generated_title("新对话", "  分析熟料三率值变化\n补充说明"),
            "分析熟料三率值变化",
        )

    def test_title_is_bounded(self):
        title = clean_generated_title("这是一段非常非常长而且不适合作为对话标题的模型输出内容", "")
        self.assertLessEqual(len(title), 24)

    def test_attachment_only_message_is_deferred_for_auto_title(self):
        self.assertTrue(is_attachment_only_message(
            "[用户上传的参考图] 项目/显微照片.png\n[用户上传的文件] 项目/数据表.xlsx"
        ))
        self.assertFalse(is_attachment_only_message(
            "请分析这些数据\n\n[用户上传的文件] 项目/数据表.xlsx"
        ))
        self.assertFalse(is_attachment_only_message(""))


if __name__ == "__main__":
    unittest.main()