aps-agent/tests/golden/test_project_file_upload.py

# ============================================================
# 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"))