50 lines
1.9 KiB
Python
50 lines
1.9 KiB
Python
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
from core.artifacts import ArtifactPathError
|
|
from core.attachments import content_for_model, normalize_attachment_refs
|
|
|
|
|
|
class AttachmentRefTests(unittest.TestCase):
|
|
def test_normalizes_user_relative_path_to_task_relative_metadata(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
wd = root / "实验分析"
|
|
wd.mkdir()
|
|
image = wd / "显微照片.png"
|
|
image.write_bytes(b"png")
|
|
refs = normalize_attachment_refs(
|
|
[{"path": "实验分析/显微照片.png", "kind": "image"}],
|
|
working_dir=wd,
|
|
user_root=root,
|
|
)
|
|
self.assertEqual(refs[0]["path"], "显微照片.png")
|
|
self.assertEqual(refs[0]["kind"], "image")
|
|
self.assertEqual(refs[0]["size_bytes"], 3)
|
|
self.assertEqual(refs[0]["scope"], "working_dir")
|
|
|
|
def test_rejects_attachment_outside_working_dir(self):
|
|
with tempfile.TemporaryDirectory() as tmp:
|
|
root = Path(tmp)
|
|
wd = root / "任务"
|
|
wd.mkdir()
|
|
(root / "outside.txt").write_text("x", encoding="utf-8")
|
|
with self.assertRaises(ArtifactPathError):
|
|
normalize_attachment_refs(
|
|
[{"path": "../outside.txt", "kind": "file"}],
|
|
working_dir=wd,
|
|
user_root=root,
|
|
)
|
|
|
|
def test_builds_provider_content_without_changing_stored_text(self):
|
|
refs = [{"path": "figures/a.png", "kind": "image"}]
|
|
self.assertEqual(
|
|
content_for_model("请分析", refs, "实验分析"),
|
|
"请分析\n\n[用户上传的参考图] 实验分析/figures/a.png",
|
|
)
|
|
self.assertEqual(
|
|
content_for_model("", refs, "实验分析"),
|
|
"[用户上传的参考图] 实验分析/figures/a.png",
|
|
)
|