2026-06-02 18:22:36 +08:00

94 lines
3.2 KiB
TypeScript

import { afterEach, expect, test, vi } from "vitest";
import { setAccessToken } from "@/shared/api/request";
import { uploadFile, uploadFilesBatch, uploadImage, uploadImagesBatch } from "@/shared/api/upload";
afterEach(() => {
setAccessToken("");
vi.unstubAllGlobals();
});
test("uploadImage sends multipart data to generated image endpoint", async () => {
vi.stubGlobal(
"fetch",
vi.fn(
async () =>
new Response(
JSON.stringify({ code: 0, data: { url: "https://media.haiyihy.com/admin/images/a.png" } }),
),
),
);
const result = await uploadImage(new File(["image"], "avatar.png", { type: "image/png" }));
const [url, init] = vi.mocked(fetch).mock.calls[0];
expect(result.url).toContain("/admin/images/");
expect(String(url)).toContain("/api/v1/admin/files/image/upload");
expect(init?.method).toBe("POST");
expect(init?.body).toBeInstanceOf(FormData);
expect(init?.headers).not.toHaveProperty("Content-Type");
});
test("uploadFile sends multipart data to generated file endpoint", async () => {
vi.stubGlobal(
"fetch",
vi.fn(
async () =>
new Response(JSON.stringify({ code: 0, data: { url: "https://media.haiyihy.com/admin/files/a.pdf" } })),
),
);
await uploadFile(new File(["file"], "doc.pdf", { type: "application/pdf" }));
const [url, init] = vi.mocked(fetch).mock.calls[0];
expect(String(url)).toContain("/api/v1/admin/files/upload");
expect(init?.method).toBe("POST");
expect(init?.body).toBeInstanceOf(FormData);
expect(init?.headers).not.toHaveProperty("Content-Type");
});
test("uploadFilesBatch sends multipart data to admin generic batch endpoint", async () => {
vi.stubGlobal(
"fetch",
vi.fn(
async () =>
new Response(
JSON.stringify({
code: 0,
data: [{ url: "https://media.haiyihy.com/admin/files/gift.mp4" }],
}),
),
),
);
await uploadFilesBatch([new File(["video"], "gift.mp4", { type: "video/mp4" })]);
const [url, init] = vi.mocked(fetch).mock.calls[0];
expect(String(url)).toContain("/api/v1/admin/files/batch-upload");
expect(init?.method).toBe("POST");
expect(init?.body).toBeInstanceOf(FormData);
expect(init?.headers).not.toHaveProperty("Content-Type");
});
test("uploadImagesBatch sends multipart data to admin image batch endpoint", async () => {
vi.stubGlobal(
"fetch",
vi.fn(
async () =>
new Response(
JSON.stringify({
code: 0,
data: [{ url: "https://media.haiyihy.com/admin/images/cover.png" }],
}),
),
),
);
await uploadImagesBatch([new File(["image"], "cover.png", { type: "image/png" })]);
const [url, init] = vi.mocked(fetch).mock.calls[0];
expect(String(url)).toContain("/api/v1/admin/files/image/batch-upload");
expect(init?.method).toBe("POST");
expect(init?.body).toBeInstanceOf(FormData);
expect(init?.headers).not.toHaveProperty("Content-Type");
});