From e1a055b95955c307c44381a44ae4853282621190 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=99=B4=E7=8C=AB?= <bbleae@outlook.com>
Date: Sun, 15 Mar 2026 04:02:28 +0900
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BC=98=E5=8C=96=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E7=94=A8=E4=BE=8B=E4=B8=AD=E7=9A=84=E6=96=87=E4=BB=B6=E5=86=99?=
 =?UTF-8?q?=E5=85=A5=E5=92=8C=E8=AF=BB=E5=8F=96=E9=94=99=E8=AF=AF=E5=A4=84?=
 =?UTF-8?q?=E7=90=86=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pytests/prompt_test/test_prompt_manager.py | 53 ++++++++--------------
 1 file changed, 18 insertions(+), 35 deletions(-)

diff --git a/pytests/prompt_test/test_prompt_manager.py b/pytests/prompt_test/test_prompt_manager.py
index c06cc754..58505da3 100644
--- a/pytests/prompt_test/test_prompt_manager.py
+++ b/pytests/prompt_test/test_prompt_manager.py
@@ -686,17 +686,14 @@ def test_prompt_manager_save_prompts_io_error_on_write(tmp_path, monkeypatch):
     p1 = Prompt(prompt_name="save_error", template="T")
     manager.add_prompt(p1, need_save=True)
 
-    class FakeFile:
-        def __enter__(self):
+    original_write_text = Path.write_text
+
+    def fake_write_text(self, *args, **kwargs):
+        if self == custom_dir / f"save_error{SUFFIX_PROMPT}":
             raise OSError("disk write error")
+        return original_write_text(self, *args, **kwargs)
 
-        def __exit__(self, exc_type, exc, tb):
-            return False
-
-    def fake_open(*_args, **_kwargs):
-        return FakeFile()
-
-    monkeypatch.setattr("builtins.open", fake_open)
+    monkeypatch.setattr(Path, "write_text", fake_write_text)
 
     # Act / Assert
     with pytest.raises(OSError) as exc_info:
@@ -721,21 +718,14 @@ def test_prompt_manager_load_prompts_io_error_from_default_dir(tmp_path, monkeyp
 
     prompt_file = write_source_prompt(prompts_dir, "bad", "content")
 
-    class FakeFile:
-        def __enter__(self):
+    original_read_text = Path.read_text
+
+    def fake_read_text(self, *args, **kwargs):
+        if self == prompt_file:
             raise OSError("read error")
+        return original_read_text(self, *args, **kwargs)
 
-        def __exit__(self, exc_type, exc, tb):
-            return False
-
-    def fake_open(*args, **kwargs):
-        # 只对 default 目录下的文件触发错误，其余正常（如果有）
-        file_path = Path(args[0])
-        if file_path == prompt_file:
-            return FakeFile()
-        return open(*args, **kwargs)
-
-    monkeypatch.setattr("builtins.open", fake_open)
+    monkeypatch.setattr(Path, "read_text", fake_read_text)
     manager = PromptManager()
 
     # Act / Assert
@@ -772,21 +762,14 @@ def test_prompt_manager_load_prompts_io_error_from_custom_dir(tmp_path, monkeypa
     only_custom_file = custom_dir / f"only_custom{SUFFIX_PROMPT}"
     only_custom_file.write_text("only", encoding="utf-8")
 
-    class FakeFile:
-        def __enter__(self):
+    original_read_text = Path.read_text
+
+    def fake_read_text(self, *args, **kwargs):
+        if self.parent == custom_dir:
             raise OSError("custom read error")
+        return original_read_text(self, *args, **kwargs)
 
-        def __exit__(self, exc_type, exc, tb):
-            return False
-
-    def fake_open(*args, **kwargs):
-        file_path = Path(args[0])
-        # 对 custom 目录下的 prompt 文件统一触发错误
-        if file_path.parent == custom_dir:
-            return FakeFile()
-        return open(*args, **kwargs)
-
-    monkeypatch.setattr("builtins.open", fake_open)
+    monkeypatch.setattr(Path, "read_text", fake_read_text)
     manager = PromptManager()
 
     # Act / Assert