4f6353d555
- 将 api 测试重组为 integration 测试 - 新增 unit 和 e2e 测试目录分类 - 新增 testing-guidelines.md 测试指南文档 - 更新 pyproject.toml 和 run_tests.sh 以适配新结构 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
from yuxi.agents.backends.sandbox import (
|
|
VIRTUAL_PATH_PREFIX,
|
|
ensure_thread_dirs,
|
|
sandbox_outputs_dir,
|
|
sandbox_uploads_dir,
|
|
)
|
|
from yuxi.agents.state import merge_artifacts
|
|
from yuxi.agents.toolkits.buildin.tools import _normalize_presented_artifact_path
|
|
from yuxi.services.chat_service import extract_agent_state
|
|
|
|
|
|
def _runtime_with_thread(thread_id: str):
|
|
context = type("RuntimeContext", (), {"thread_id": thread_id})()
|
|
return type("RuntimeStub", (), {"context": context})()
|
|
|
|
|
|
def test_merge_artifacts_deduplicates_and_preserves_order():
|
|
assert merge_artifacts(
|
|
["/home/gem/user-data/outputs/a.md"],
|
|
["/home/gem/user-data/outputs/a.md", "/home/gem/user-data/outputs/b.md"],
|
|
) == [
|
|
"/home/gem/user-data/outputs/a.md",
|
|
"/home/gem/user-data/outputs/b.md",
|
|
]
|
|
|
|
|
|
def test_normalize_presented_artifact_path_accepts_host_path():
|
|
thread_id = "artifacts-host-path"
|
|
ensure_thread_dirs(thread_id)
|
|
output_file = sandbox_outputs_dir(thread_id) / "report.md"
|
|
output_file.write_text("# demo", encoding="utf-8")
|
|
|
|
normalized = _normalize_presented_artifact_path(str(output_file), _runtime_with_thread(thread_id))
|
|
|
|
assert normalized == f"{VIRTUAL_PATH_PREFIX}/outputs/report.md"
|
|
|
|
|
|
def test_normalize_presented_artifact_path_accepts_virtual_path():
|
|
thread_id = "artifacts-virtual-path"
|
|
ensure_thread_dirs(thread_id)
|
|
output_file = sandbox_outputs_dir(thread_id) / "summary.txt"
|
|
output_file.write_text("demo", encoding="utf-8")
|
|
|
|
normalized = _normalize_presented_artifact_path(
|
|
f"{VIRTUAL_PATH_PREFIX}/outputs/summary.txt",
|
|
_runtime_with_thread(thread_id),
|
|
)
|
|
|
|
assert normalized == f"{VIRTUAL_PATH_PREFIX}/outputs/summary.txt"
|
|
|
|
|
|
def test_normalize_presented_artifact_path_rejects_non_outputs_path():
|
|
thread_id = "artifacts-reject-path"
|
|
ensure_thread_dirs(thread_id)
|
|
upload_file = sandbox_uploads_dir(thread_id) / "note.txt"
|
|
upload_file.write_text("demo", encoding="utf-8")
|
|
|
|
try:
|
|
_normalize_presented_artifact_path(str(upload_file), _runtime_with_thread(thread_id))
|
|
except ValueError as exc:
|
|
assert f"{VIRTUAL_PATH_PREFIX}/outputs/" in str(exc)
|
|
else:
|
|
raise AssertionError("expected ValueError for non-outputs file")
|
|
|
|
|
|
def test_extract_agent_state_includes_artifacts():
|
|
state = extract_agent_state(
|
|
{
|
|
"todos": [{"content": "done", "status": "completed"}],
|
|
"files": {"/tmp/demo.txt": {"content": ["x"]}},
|
|
"artifacts": ["/home/gem/user-data/outputs/demo.txt"],
|
|
}
|
|
)
|
|
|
|
assert state["todos"] == [{"content": "done", "status": "completed"}]
|
|
assert state["files"] == {"/tmp/demo.txt": {"content": ["x"]}}
|
|
assert state["artifacts"] == ["/home/gem/user-data/outputs/demo.txt"]
|