张胜宇
e239eb778b
1) 客服 Agent 四份交付文档 + 构建脚手架：品牌由包装占位 XX科技 / 旧名 南方财富 统一为南方基金（热线 400-889-8899 / 官网 nffund.com），系统名改为「智能服务系统」； 同步追加 §0.4 修订记录行，工程记录行保留原占位字面以支撑硬编码扫描验收。 2) 开发文档：清理 28 份已作废/残留文档（14 份移出归档 + 14 份仓库副本）， 新增《文档规整方案与开发前待决事项-2026-09-17》。 3) 客服agent 四份交付文档首次纳入本分支。
136 lines
4.3 KiB
Python
136 lines
4.3 KiB
Python
import shutil
|
||
import uuid
|
||
from collections.abc import Iterator
|
||
from pathlib import Path
|
||
|
||
import pytest
|
||
|
||
from app.infrastructure.document_storage import DocumentStorage, LocalDocumentStorage
|
||
|
||
WORKROOT = Path(__file__).resolve().parents[2] / ".workdir"
|
||
|
||
|
||
@pytest.fixture
|
||
def tmp_path() -> Iterator[Path]:
|
||
"""工作区内的临时目录。
|
||
|
||
本机沙箱禁止枚举 `%TEMP%\\pytest-of-*`，pytest 内建 `tmp_path` 会 PermissionError，
|
||
故自管一个工作区目录；语义与 `tmp_path` 一致（每个用例唯一、用完即删）。
|
||
"""
|
||
path = WORKROOT / uuid.uuid4().hex[:12]
|
||
path.mkdir(parents=True)
|
||
try:
|
||
yield path
|
||
finally:
|
||
shutil.rmtree(path, ignore_errors=True)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_save_read_delete_round_trip(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
key = "kb/2026/faq.txt"
|
||
|
||
path = await storage.save(key=key, content=b"hello", content_type="text/plain")
|
||
assert path == key
|
||
assert await storage.read(key=key) == b"hello"
|
||
|
||
await storage.delete(key=key)
|
||
with pytest.raises(FileNotFoundError):
|
||
await storage.read(key=key)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_archive_moves_without_deleting(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
await storage.save(key="a.txt", content=b"x", content_type="text/plain")
|
||
|
||
await storage.archive(key="a.txt")
|
||
|
||
assert (tmp_path / "archive" / "a.txt").read_bytes() == b"x"
|
||
assert not (tmp_path / "a.txt").exists()
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_save_creates_nested_directories(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path / "not-yet-created")
|
||
|
||
await storage.save(
|
||
key="kb/2026/09/faq.md", content="标题".encode(), content_type="text/markdown"
|
||
)
|
||
|
||
assert (tmp_path / "not-yet-created" / "kb" / "2026" / "09" / "faq.md").read_bytes() == (
|
||
"标题".encode()
|
||
)
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_archive_preserves_sub_directory_layout(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
await storage.save(key="kb/faq.txt", content=b"body", content_type="text/plain")
|
||
|
||
await storage.archive(key="kb/faq.txt")
|
||
|
||
assert (tmp_path / "archive" / "kb" / "faq.txt").read_bytes() == b"body"
|
||
assert not (tmp_path / "kb" / "faq.txt").exists()
|
||
assert (tmp_path / "kb").is_dir()
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_archive_is_idempotent_for_missing_key(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
|
||
await storage.archive(key="missing.txt")
|
||
|
||
assert not (tmp_path / "archive").exists()
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_archive_rejects_already_archived_key(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
(tmp_path / "archive").mkdir()
|
||
(tmp_path / "archive" / "a.txt").write_bytes(b"x")
|
||
|
||
with pytest.raises(ValueError):
|
||
await storage.save(key="archive/a.txt", content=b"x", content_type="text/plain")
|
||
with pytest.raises(ValueError):
|
||
await storage.archive(key="archive/a.txt")
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_delete_is_idempotent(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
|
||
await storage.delete(key="never-existed.txt")
|
||
|
||
|
||
@pytest.mark.asyncio
|
||
async def test_save_overwrites_existing_key(tmp_path: Path) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
await storage.save(key="a.txt", content=b"old", content_type="text/plain")
|
||
|
||
await storage.save(key="a.txt", content=b"new", content_type="text/plain")
|
||
|
||
assert await storage.read(key="a.txt") == b"new"
|
||
|
||
|
||
@pytest.mark.parametrize("key", ["../escape.txt", "kb/../../escape.txt", "..\\escape.txt"])
|
||
@pytest.mark.asyncio
|
||
async def test_parent_directory_reference_is_rejected(tmp_path: Path, key: str) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
|
||
with pytest.raises(ValueError):
|
||
await storage.save(key=key, content=b"x", content_type="text/plain")
|
||
|
||
|
||
@pytest.mark.parametrize("key", ["", "/etc/passwd", "C:/Windows/win.ini"])
|
||
@pytest.mark.asyncio
|
||
async def test_absolute_and_empty_keys_are_rejected(tmp_path: Path, key: str) -> None:
|
||
storage = LocalDocumentStorage(tmp_path)
|
||
|
||
with pytest.raises(ValueError):
|
||
await storage.read(key=key)
|
||
|
||
|
||
def test_local_storage_implements_protocol(tmp_path: Path) -> None:
|
||
assert isinstance(LocalDocumentStorage(tmp_path), DocumentStorage)
|