T-579 内置默认AI提示词

2026-07-10 09:00:24 +08:00
parent 9742448750
commit e5b7fc4afa
11 changed files with 306 additions and 4 deletions
@@ -45,6 +45,40 @@ class PromptTests(TempDirMixin, unittest.TestCase):

        self.assert_removed(temp_dir)

+    def test_ensure_default_prompts_seeds_empty_user_prompt_files(self):
+        with self.make_temp_dir() as temp_dir:
+            title_path = os.path.join(temp_dir, "title_prompt.txt")
+            cover_dir = os.path.join(temp_dir, "prompts", "cover")
+
+            prompts.ensure_default_prompts(title_path, cover_dir)
+
+            self.assertIn("蝦皮台灣站", prompts.load_title_prompt(title_path))
+            self.assertEqual(["papa1"], prompts.list_cover_templates(cover_dir))
+            self.assertIn(
+                "商品标题:{新标题}",
+                prompts.load_cover_template("papa1", cover_dir),
+            )
+
+        self.assert_removed(temp_dir)
+
+    def test_ensure_default_prompts_does_not_overwrite_user_prompts(self):
+        with self.make_temp_dir() as temp_dir:
+            title_path = os.path.join(temp_dir, "title_prompt.txt")
+            cover_dir = os.path.join(temp_dir, "prompts", "cover")
+            prompts.save_title_prompt("用户标题提示词", title_path)
+            prompts.save_cover_template("用户模板", "用户封面提示词", cover_dir)
+
+            prompts.ensure_default_prompts(title_path, cover_dir)
+
+            self.assertEqual("用户标题提示词", prompts.load_title_prompt(title_path))
+            self.assertEqual(["用户模板"], prompts.list_cover_templates(cover_dir))
+            self.assertEqual(
+                "用户封面提示词",
+                prompts.load_cover_template("用户模板", cover_dir),
+            )
+
+        self.assert_removed(temp_dir)
+
    def test_render_prompt_replaces_known_variables(self):
        task = {
            "old_title": "舊T恤",