90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
|
# ============================================================
|
|||
|
|
# Web 本地工程数据文件上传黄金测试（AG-08 扩展）
|
|||
|
|
# 覆盖：multipart 上传 → 项目 workDir 自动落位 → 目录分析可见
|
|||
|
|
# ============================================================
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
import pytest
|
|||
|
|
from fastapi.testclient import TestClient
|
|||
|
|
|
|||
|
|
from server.aps_domain.folder_pack import analyze_work_dir
|
|||
|
|
from server.state.seed import seed_world
|
|||
|
|
from tests.auth_provider import install_test_auth
|
|||
|
|
|
|||
|
|
|
|||
|
|
@pytest.fixture()
|
|||
|
|
def secure_app(tmp_path, monkeypatch):
|
|||
|
|
monkeypatch.setenv("APS_DB_PATH", str(tmp_path / "tenant.db"))
|
|||
|
|
monkeypatch.setenv("APS_WORLD_PATH", str(tmp_path / "world.json"))
|
|||
|
|
monkeypatch.setenv("APS_DATA_DIR", str(tmp_path / "aps-data"))
|
|||
|
|
import server.state.store as world_store
|
|||
|
|
from server.db.database import reset_engine
|
|||
|
|
|
|||
|
|
install_test_auth(monkeypatch, "tenant-a-000000000000000000000001")
|
|||
|
|
world_store._stores.clear()
|
|||
|
|
reset_engine()
|
|||
|
|
from server.gateway.app import create_app
|
|||
|
|
app = create_app()
|
|||
|
|
yield app
|
|||
|
|
reset_engine()
|
|||
|
|
world_store._stores.clear()
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _login(client: TestClient) -> None:
|
|||
|
|
response = client.post("/api/auth/login", json={
|
|||
|
|
"method": "password", "username": "planner", "password": "test",
|
|||
|
|
})
|
|||
|
|
assert response.status_code == 200
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_upload_local_files_then_folder_analysis(secure_app, monkeypatch):
|
|||
|
|
client = TestClient(secure_app)
|
|||
|
|
_login(client)
|
|||
|
|
created = client.post("/api/projects", json={"name": "浏览器项目"}).json()
|
|||
|
|
pid = created["project"]["id"]
|
|||
|
|
sid = created["session"]["id"]
|
|||
|
|
|
|||
|
|
csv = "订单号,客户,产品编码,数量,交期\nSO-DEMO-1,比亚迪,A0050101-00280,1,2026-08-30\n"
|
|||
|
|
response = client.post(
|
|||
|
|
f"/api/projects/{pid}/files/upload",
|
|||
|
|
files=[("files", ("订单-样例.csv", csv.encode("utf-8"), "text/csv"))],
|
|||
|
|
)
|
|||
|
|
assert response.status_code == 200, response.text
|
|||
|
|
body = response.json()
|
|||
|
|
assert body["saved"] == ["订单-样例.csv"]
|
|||
|
|
assert body["workDir"]
|
|||
|
|
assert os.path.isfile(os.path.join(body["workDir"], "订单-样例.csv"))
|
|||
|
|
|
|||
|
|
snap = client.get("/api/workspace").json()
|
|||
|
|
project = next(p for p in snap["projects"] if p["id"] == pid)
|
|||
|
|
assert project["workDir"] == body["workDir"]
|
|||
|
|
assert any(f["projectId"] == pid and f["name"] == "订单-样例.csv" for f in snap["files"])
|
|||
|
|
|
|||
|
|
class FakePS:
|
|||
|
|
def snapshot(self, include_messages=False):
|
|||
|
|
return {
|
|||
|
|
"projects": [{"id": pid, "name": "浏览器项目", "workDir": body["workDir"]}],
|
|||
|
|
"sessions": [{"id": sid, "projectId": pid}],
|
|||
|
|
"files": [],
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
monkeypatch.setattr("server.state.projects.get_project_store", lambda: FakePS())
|
|||
|
|
report = analyze_work_dir(seed_world(), sid)
|
|||
|
|
assert report["ok"] is True
|
|||
|
|
assert any(f["name"] == "订单-样例.csv" for f in report["files"])
|
|||
|
|
|
|||
|
|
|
|||
|
|
def test_upload_rejects_unsupported_extension(secure_app):
|
|||
|
|
client = TestClient(secure_app)
|
|||
|
|
_login(client)
|
|||
|
|
created = client.post("/api/projects", json={"name": "拒绝测试"}).json()
|
|||
|
|
pid = created["project"]["id"]
|
|||
|
|
response = client.post(
|
|||
|
|
f"/api/projects/{pid}/files/upload",
|
|||
|
|
files=[("files", ("说明.docx", b"docx", "application/octet-stream"))],
|
|||
|
|
)
|
|||
|
|
assert response.status_code == 400
|
|||
|
|
assert "仅支持" in response.json()["detail"]
|