# ============================================================ # Web 本地工程数据文件上传黄金测试（AG-08 扩展） # 覆盖：multipart 上传 → 项目 workDir 自动落位 → 目录分析可见 # ============================================================ from __future__ import annotations import hashlib import os from pathlib import Path import pytest from fastapi.testclient import TestClient from server.aps_domain.folder_pack import analyze_work_dir, prepare_folder_schedule from server.state.seed import seed_world from tests.auth_provider import install_test_auth from tests.external_data import external_dir @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"] is None assert body["workDirConfigured"] is True assert all(project["workDir"] is None for project in body["workspace"]["projects"]) snap = client.get("/api/workspace").json() project = next(p for p in snap["projects"] if p["id"] == pid) assert project["workDir"] assert os.path.isfile(os.path.join(project["workDir"], "订单-样例.csv")) 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": project["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"]) KANGNI_DATA_DIR = external_dir("KANGNI_DATA_DIR", "kangni") KANGNI_WORKBOOKS = ( "订单.xlsx", "工艺路线.xlsx", "工时.xlsx", "BOM.xlsx", "设备.xlsx", "模具.xlsx", "物料.xlsx", "设备能力映射模板.xlsx", "模具适配映射模板.xlsx", ) @pytest.mark.skipif( not all((KANGNI_DATA_DIR / name).is_file() for name in KANGNI_WORKBOOKS), reason="康尼现场只读数据不完整", ) def test_upload_kangni_workbooks_builds_frozen_trial_payload(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"] source_hashes = { name: hashlib.sha256((KANGNI_DATA_DIR / name).read_bytes()).hexdigest() for name in KANGNI_WORKBOOKS } files = [ ( "files", ( name, (KANGNI_DATA_DIR / name).read_bytes(), "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", ), ) for name in KANGNI_WORKBOOKS ] response = client.post(f"/api/projects/{pid}/files/upload", files=files) assert response.status_code == 200, response.text assert set(response.json()["saved"]) == set(KANGNI_WORKBOOKS) snap = client.get("/api/workspace").json() project = next(row for row in snap["projects"] if row["id"] == pid) class FakePS: def snapshot(self, include_messages=False): return { "projects": [{"id": pid, "name": "康尼上传试排", "workDir": project["workDir"]}], "sessions": [{"id": sid, "projectId": pid}], "files": [], } monkeypatch.setattr("server.state.projects.get_project_store", lambda: FakePS()) report = prepare_folder_schedule(seed_world(), sid) assert report["kangniDetected"] is True assert report["trialReady"] is True assert report["productionReady"] is False assert report["kangniMeta"]["orderCount"] == 10 assert report["kangniMeta"]["routingRecordCount"] == 72 assert len(report["sourceManifest"]) == len(KANGNI_WORKBOOKS) assert { name: hashlib.sha256((KANGNI_DATA_DIR / name).read_bytes()).hexdigest() for name in KANGNI_WORKBOOKS } == source_hashes 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"] def test_same_name_upload_preserves_file_and_metadata_when_registration_fails( secure_app, monkeypatch, tmp_path, ): from server.state.projects import ProjectStore client = TestClient(secure_app) _login(client) created = client.post( "/api/projects", json={"name": "原子上传", "workDir": str(tmp_path)} ).json() pid = created["project"]["id"] first = client.post( f"/api/projects/{pid}/files/upload", files=[("files", ("same.dxf", b"original", "application/octet-stream"))], ) assert first.status_code == 200, first.text before = client.get("/api/workspace").json() old_rows = [ row for row in before["files"] if row["projectId"] == pid and row["name"] == "same.dxf" ] assert len(old_rows) == 1 def fail_create_file(self, project_id, name, kind="other", note="", file_id=None): raise RuntimeError("injected metadata failure") monkeypatch.setattr(ProjectStore, "create_file", fail_create_file) failed = client.post( f"/api/projects/{pid}/files/upload", files=[("files", ("same.dxf", b"replacement", "application/octet-stream"))], ) assert failed.status_code == 200, failed.text assert failed.json()["saved"] == [] assert "injected metadata failure" in failed.json()["errors"][0]["error"] assert (tmp_path / "same.dxf").read_bytes() == b"original" after = client.get("/api/workspace").json() rows = [ row for row in after["files"] if row["projectId"] == pid and row["name"] == "same.dxf" ] assert rows == old_rows assert not list(tmp_path.glob(".*.upload")) assert not list(tmp_path.glob(".*.backup"))