128 lines
4.8 KiB
TypeScript
128 lines
4.8 KiB
TypeScript
import { DownloadService, type DownloadItem } from "$services/downloadService";
|
|
import { saveAs } from "file-saver";
|
|
import JSZip from "jszip";
|
|
import { beforeEach, describe, expect, it, vi } from "vitest";
|
|
|
|
vi.mock("file-saver", () => {
|
|
return {
|
|
saveAs: vi.fn(),
|
|
};
|
|
});
|
|
|
|
vi.mock("jszip", () => {
|
|
// Создаем объект с методами заранее, чтобы иметь к ним доступ
|
|
const mockZipInstance = {
|
|
file: vi.fn().mockReturnThis(),
|
|
generateAsync: vi.fn().mockResolvedValue(new Blob([])),
|
|
};
|
|
|
|
// Возвращаем функцию-конструктор
|
|
return {
|
|
default: vi.fn(function () {
|
|
return mockZipInstance;
|
|
}),
|
|
};
|
|
});
|
|
|
|
describe("DownloadService", () => {
|
|
let service: DownloadService;
|
|
let mockKonvaStage: any;
|
|
|
|
beforeEach(() => {
|
|
service = new DownloadService();
|
|
vi.clearAllMocks();
|
|
|
|
mockKonvaStage = {
|
|
toBlob: vi.fn(({ callback }) => {
|
|
callback(new Blob(["test image data"], { type: "image/png" }));
|
|
}),
|
|
};
|
|
});
|
|
|
|
describe("downloadPanel", () => {
|
|
it("should export panel successfully", async () => {
|
|
const result = await service.downloadPanel(mockKonvaStage, "test-panel-1");
|
|
|
|
const [blobArg, fileNameArg] = vi.mocked(saveAs).mock.calls[0];
|
|
|
|
expect(result.success).toBe(true);
|
|
expect(mockKonvaStage.toBlob).toHaveBeenCalled();
|
|
expect(blobArg instanceof Blob).toBe(true);
|
|
expect(fileNameArg).toBe("test-panel-1.png");
|
|
});
|
|
|
|
it("should handle Konva stage without toBlob method", async () => {
|
|
const invalidStage = { toBlob: undefined };
|
|
|
|
const result = await service.downloadPanel(invalidStage as any, "test-panel-1.png");
|
|
|
|
expect(result.success).toBe(false);
|
|
if (!result.success) {
|
|
expect(result.error).toContain("Konva Stage не найден или не поддерживает toBlob");
|
|
}
|
|
});
|
|
|
|
it("should handle blob creation failure", async () => {
|
|
mockKonvaStage.toBlob = vi.fn(({ callback }) => {
|
|
callback(null);
|
|
});
|
|
|
|
const result = await service.downloadPanel(mockKonvaStage, "test-panel-1.png");
|
|
|
|
expect(result.success).toBe(false);
|
|
if (!result.success) {
|
|
expect(result.error).toContain("Не удалось создать изображение");
|
|
}
|
|
});
|
|
});
|
|
|
|
describe("downloadAllPanels", () => {
|
|
it("should handle successful download of multiple panels", async () => {
|
|
const panels: Array<DownloadItem> = [{ filename: "test-panel-1", stage: mockKonvaStage }];
|
|
const result = await service.downloadAll(panels);
|
|
const [blobArg, fileNameArg] = vi.mocked(saveAs).mock.calls[0];
|
|
const zipInstance = vi.mocked(JSZip).mock.instances[0];
|
|
|
|
expect(result.success).toBe(true);
|
|
expect(mockKonvaStage.toBlob).toHaveBeenCalled();
|
|
expect(blobArg instanceof Blob).toBe(true);
|
|
expect(fileNameArg).toBe("panels.zip");
|
|
expect(zipInstance.file).toHaveBeenCalledTimes(1);
|
|
expect(zipInstance.file).toHaveBeenCalledWith("test-panel-1.png", expect.anything());
|
|
});
|
|
|
|
it("should add to zip all files", async () => {
|
|
const panels: Array<DownloadItem> = [
|
|
{ filename: "test-panel-1", stage: mockKonvaStage },
|
|
{ filename: "test-panel-2", stage: mockKonvaStage },
|
|
{ filename: "test-panel-3", stage: mockKonvaStage },
|
|
{ filename: "test-panel-4", stage: mockKonvaStage },
|
|
{ filename: "test-panel-5", stage: mockKonvaStage },
|
|
];
|
|
const result = await service.downloadAll(panels);
|
|
const [blobArg, fileNameArg] = vi.mocked(saveAs).mock.calls[0];
|
|
const zipInstance = vi.mocked(JSZip).mock.instances[0];
|
|
|
|
expect(result.success).toBe(true);
|
|
expect(mockKonvaStage.toBlob).toHaveBeenCalledTimes(5);
|
|
expect(zipInstance.file).toHaveBeenCalledTimes(5);
|
|
expect(zipInstance.file).toHaveBeenCalledWith("test-panel-1.png", expect.anything());
|
|
expect(zipInstance.file).toHaveBeenCalledWith("test-panel-2.png", expect.anything());
|
|
expect(zipInstance.file).toHaveBeenCalledWith("test-panel-3.png", expect.anything());
|
|
expect(zipInstance.file).toHaveBeenCalledWith("test-panel-4.png", expect.anything());
|
|
expect(zipInstance.file).toHaveBeenCalledWith("test-panel-5.png", expect.anything());
|
|
});
|
|
|
|
it("should handle failure", async () => {
|
|
const invalidStage = { toBlob: undefined };
|
|
const panels: Array<DownloadItem> = [{ filename: "test-panel-1", stage: invalidStage as any }];
|
|
const result = await service.downloadAll(panels);
|
|
|
|
expect(result.success).toBe(false);
|
|
if (!result.success) {
|
|
expect(result.error).toContain("Ошибка сохранения архива");
|
|
}
|
|
});
|
|
});
|
|
});
|