补齐桌面端附件文件流回归
This commit is contained in:
@@ -0,0 +1,56 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
const apiDelete = vi.fn();
|
||||
const apiGet = vi.fn();
|
||||
const apiPost = vi.fn();
|
||||
|
||||
vi.mock("./axios", () => ({
|
||||
apiDelete,
|
||||
apiGet,
|
||||
apiPost,
|
||||
}));
|
||||
|
||||
describe("attachments api", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
});
|
||||
|
||||
it("downloads attachment blobs without putting credentials in the URL", async () => {
|
||||
const { downloadAttachment } = await import("./attachments");
|
||||
|
||||
downloadAttachment("attachment-1");
|
||||
|
||||
expect(apiGet).toHaveBeenCalledWith("/api/v1/attachments/attachment-1/download", {
|
||||
responseType: "blob",
|
||||
});
|
||||
expect(apiGet.mock.calls[0][0]).not.toContain("token");
|
||||
expect(apiGet.mock.calls[0][0]).not.toContain("access_token");
|
||||
});
|
||||
|
||||
it("uploads attachments as multipart form data", async () => {
|
||||
const { uploadAttachment } = await import("./attachments");
|
||||
const file = new File(["content"], "report.pdf", { type: "application/pdf" });
|
||||
const onUploadProgress = vi.fn();
|
||||
|
||||
uploadAttachment("study-1", "startup_initiation", "entity-1", file, { onUploadProgress });
|
||||
|
||||
expect(apiPost).toHaveBeenCalledWith(
|
||||
"/api/v1/studies/study-1/startup_initiation/entity-1/attachments/",
|
||||
expect.any(FormData),
|
||||
{
|
||||
headers: { "Content-Type": "multipart/form-data" },
|
||||
onUploadProgress,
|
||||
},
|
||||
);
|
||||
const formData = apiPost.mock.calls[0][1] as FormData;
|
||||
expect(formData.get("file")).toBe(file);
|
||||
});
|
||||
|
||||
it("deletes attachments through the scoped attachment endpoint", async () => {
|
||||
const { deleteAttachment } = await import("./attachments");
|
||||
|
||||
deleteAttachment("attachment-1");
|
||||
|
||||
expect(apiDelete).toHaveBeenCalledWith("/api/v1/attachments/attachment-1");
|
||||
});
|
||||
});
|
||||
@@ -0,0 +1,100 @@
|
||||
import { beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
const capabilitiesMock = vi.hoisted(() => vi.fn());
|
||||
const openFileMock = vi.hoisted(() => vi.fn());
|
||||
const pickFilesMock = vi.hoisted(() => vi.fn());
|
||||
const saveFileMock = vi.hoisted(() => vi.fn());
|
||||
const messageInfoMock = vi.hoisted(() => vi.fn());
|
||||
const messageSuccessMock = vi.hoisted(() => vi.fn());
|
||||
|
||||
vi.mock("../runtime", () => ({
|
||||
clientRuntime: {
|
||||
capabilities: capabilitiesMock,
|
||||
},
|
||||
openFile: openFileMock,
|
||||
pickFiles: pickFilesMock,
|
||||
saveFile: saveFileMock,
|
||||
}));
|
||||
|
||||
vi.mock("element-plus", () => ({
|
||||
ElMessage: {
|
||||
info: messageInfoMock,
|
||||
success: messageSuccessMock,
|
||||
},
|
||||
}));
|
||||
|
||||
const output = {
|
||||
suggestedName: "report.pdf",
|
||||
mimeType: "application/pdf",
|
||||
data: new Blob(["content"], { type: "application/pdf" }),
|
||||
};
|
||||
|
||||
describe("file task feedback", () => {
|
||||
beforeEach(() => {
|
||||
vi.clearAllMocks();
|
||||
capabilitiesMock.mockReturnValue({ nativeFiles: false });
|
||||
pickFilesMock.mockResolvedValue([]);
|
||||
saveFileMock.mockResolvedValue("saved");
|
||||
openFileMock.mockResolvedValue(undefined);
|
||||
});
|
||||
|
||||
it("reports native file picker selections", async () => {
|
||||
const files = [
|
||||
new File(["a"], "a.txt", { type: "text/plain" }),
|
||||
new File(["b"], "b.txt", { type: "text/plain" }),
|
||||
];
|
||||
pickFilesMock.mockResolvedValue(files);
|
||||
const { pickFilesWithFeedback } = await import("./fileTaskFeedback");
|
||||
|
||||
const selected = await pickFilesWithFeedback({ multiple: true, title: "附件" });
|
||||
|
||||
expect(selected).toBe(files);
|
||||
expect(pickFilesMock).toHaveBeenCalledWith({ multiple: true, title: "附件" });
|
||||
expect(messageSuccessMock).toHaveBeenCalledWith("已选择 2 个文件");
|
||||
});
|
||||
|
||||
it("does not show a selection toast when the picker is cancelled", async () => {
|
||||
const { pickFilesWithFeedback } = await import("./fileTaskFeedback");
|
||||
|
||||
await expect(pickFilesWithFeedback({ multiple: true })).resolves.toEqual([]);
|
||||
|
||||
expect(messageSuccessMock).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("uses download feedback for web save flows", async () => {
|
||||
const { saveFileWithFeedback } = await import("./fileTaskFeedback");
|
||||
|
||||
await expect(saveFileWithFeedback(output)).resolves.toBe("saved");
|
||||
|
||||
expect(saveFileMock).toHaveBeenCalledWith(output);
|
||||
expect(messageSuccessMock).toHaveBeenCalledWith("文件下载已开始");
|
||||
});
|
||||
|
||||
it("uses saved feedback for native save flows", async () => {
|
||||
capabilitiesMock.mockReturnValue({ nativeFiles: true });
|
||||
const { saveFileWithFeedback } = await import("./fileTaskFeedback");
|
||||
|
||||
await expect(saveFileWithFeedback(output)).resolves.toBe("saved");
|
||||
|
||||
expect(messageSuccessMock).toHaveBeenCalledWith("文件已保存");
|
||||
});
|
||||
|
||||
it("reports cancelled save flows without treating them as downloads", async () => {
|
||||
saveFileMock.mockResolvedValue("cancelled");
|
||||
const { saveFileWithFeedback } = await import("./fileTaskFeedback");
|
||||
|
||||
await expect(saveFileWithFeedback(output)).resolves.toBe("cancelled");
|
||||
|
||||
expect(messageInfoMock).toHaveBeenCalledWith("已取消保存文件");
|
||||
expect(messageSuccessMock).not.toHaveBeenCalled();
|
||||
});
|
||||
|
||||
it("reports external open completion", async () => {
|
||||
const { openFileWithFeedback } = await import("./fileTaskFeedback");
|
||||
|
||||
await openFileWithFeedback(output);
|
||||
|
||||
expect(openFileMock).toHaveBeenCalledWith(output);
|
||||
expect(messageSuccessMock).toHaveBeenCalledWith("文件已打开");
|
||||
});
|
||||
});
|
||||
Reference in New Issue
Block a user