test(18-01): add save-generation route tests with fs and fetch mocking

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
6 months ago · 17ee68c7d7
1 changed files with 414 additions and 0 deletions
--- a/src/app/api/save-generation/tests/route.test.ts
+++ b/src/app/api/save-generation/tests/route.test.ts
@ -0,0 +1,414 @@
+import { describe, it, expect, vi, beforeEach, afterEach } from "vitest";
+import { NextRequest } from "next/server";
+import * as crypto from "crypto";
+
+// Mock fs/promises before importing the route
+const mockStat = vi.fn();
+const mockMkdir = vi.fn();
+const mockWriteFile = vi.fn();
+const mockReaddir = vi.fn();
+
+vi.mock("fs/promises", () => ({
+  stat: (...args: unknown[]) => mockStat(...args),
+  mkdir: (...args: unknown[]) => mockMkdir(...args),
+  writeFile: (...args: unknown[]) => mockWriteFile(...args),
+  readdir: (...args: unknown[]) => mockReaddir(...args),
+}));
+
+// Mock logger to avoid console noise during tests
+vi.mock("@/utils/logger", () => ({
+  logger: {
+    info: vi.fn(),
+    warn: vi.fn(),
+    error: vi.fn(),
+  },
+}));
+
+// Store original fetch
+const originalFetch = global.fetch;
+
+import { POST } from "../route";
+
+// Helper to create mock NextRequest for POST
+function createMockPostRequest(body: unknown): NextRequest {
+  return {
+    json: vi.fn().mockResolvedValue(body),
+  } as unknown as NextRequest;
+}
+
+// Helper to compute expected hash for testing
+function computeExpectedHash(buffer: Buffer): string {
+  return crypto.createHash("md5").update(buffer).digest("hex");
+}
+
+// Helper to create base64 data URL from string content
+function createBase64DataUrl(content: string, mimeType = "image/png"): string {
+  const buffer = Buffer.from(content);
+  return `data:${mimeType};base64,${buffer.toString("base64")}`;
+}
+
+describe("/api/save-generation route", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+    // Reset fetch mock
+    global.fetch = originalFetch;
+  });
+
+  afterEach(() => {
+    vi.resetAllMocks();
+    global.fetch = originalFetch;
+  });
+
+  describe("POST - Save generation", () => {
+    it("should save base64 image with hash-based filename", async () => {
+      const imageContent = "test-image-content";
+      const base64Image = createBase64DataUrl(imageContent, "image/png");
+      const expectedHash = computeExpectedHash(Buffer.from(imageContent));
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: base64Image,
+        prompt: "A test image",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.isDuplicate).toBe(false);
+      expect(data.filename).toContain(expectedHash);
+      expect(data.filename.endsWith(".png")).toBe(true);
+      expect(data.filePath).toContain("/test/generations/");
+      expect(mockWriteFile).toHaveBeenCalled();
+    });
+
+    it("should save base64 video with hash-based filename", async () => {
+      const videoContent = "test-video-content";
+      const base64Video = createBase64DataUrl(videoContent, "video/mp4");
+      const expectedHash = computeExpectedHash(Buffer.from(videoContent));
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        video: base64Video,
+        prompt: "A test video",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.isDuplicate).toBe(false);
+      expect(data.filename).toContain(expectedHash);
+      expect(data.filename.endsWith(".mp4")).toBe(true);
+    });
+
+    it("should deduplicate existing files by hash suffix", async () => {
+      const imageContent = "duplicate-image-content";
+      const base64Image = createBase64DataUrl(imageContent, "image/png");
+      const expectedHash = computeExpectedHash(Buffer.from(imageContent));
+      const existingFilename = `existing_prompt_${expectedHash}.png`;
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([existingFilename]);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: base64Image,
+        prompt: "Another prompt",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.isDuplicate).toBe(true);
+      expect(data.filename).toBe(existingFilename);
+      expect(mockWriteFile).not.toHaveBeenCalled();
+    });
+
+    it("should reject missing directoryPath", async () => {
+      const request = createMockPostRequest({
+        image: createBase64DataUrl("content"),
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(response.status).toBe(400);
+      expect(data.success).toBe(false);
+      expect(data.error).toBe("Missing required fields");
+    });
+
+    it("should reject missing content (no image or video)", async () => {
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        prompt: "A prompt without content",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(response.status).toBe(400);
+      expect(data.success).toBe(false);
+      expect(data.error).toBe("Missing required fields");
+    });
+
+    it("should reject non-directory path", async () => {
+      mockStat.mockResolvedValue({
+        isDirectory: () => false,
+      });
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/file.txt",
+        image: createBase64DataUrl("content"),
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(response.status).toBe(400);
+      expect(data.success).toBe(false);
+      expect(data.error).toBe("Path is not a directory");
+    });
+
+    it("should reject non-existent directory", async () => {
+      mockStat.mockRejectedValue(new Error("ENOENT"));
+
+      const request = createMockPostRequest({
+        directoryPath: "/nonexistent/dir",
+        image: createBase64DataUrl("content"),
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(response.status).toBe(400);
+      expect(data.success).toBe(false);
+      expect(data.error).toBe("Directory does not exist");
+    });
+
+    it("should handle various MIME types correctly", async () => {
+      const testCases = [
+        { mimeType: "image/jpeg", expectedExt: ".jpg" },
+        { mimeType: "image/gif", expectedExt: ".gif" },
+        { mimeType: "image/webp", expectedExt: ".webp" },
+        { mimeType: "video/webm", expectedExt: ".webm" },
+        { mimeType: "video/quicktime", expectedExt: ".mov" },
+      ];
+
+      for (const { mimeType, expectedExt } of testCases) {
+        vi.clearAllMocks();
+
+        const content = `test-content-${mimeType}`;
+        const dataUrl = createBase64DataUrl(content, mimeType);
+
+        mockStat.mockResolvedValue({
+          isDirectory: () => true,
+        });
+        mockReaddir.mockResolvedValue([]);
+        mockWriteFile.mockResolvedValue(undefined);
+
+        const request = createMockPostRequest({
+          directoryPath: "/test/generations",
+          image: dataUrl,
+          prompt: "Test",
+        });
+
+        const response = await POST(request);
+        const data = await response.json();
+
+        expect(data.success).toBe(true);
+        expect(data.filename.endsWith(expectedExt)).toBe(true);
+      }
+    });
+
+    it("should handle HTTP URLs by fetching content", async () => {
+      const mockContent = "fetched-image-content";
+      const expectedHash = computeExpectedHash(Buffer.from(mockContent));
+
+      // Mock fetch
+      global.fetch = vi.fn().mockResolvedValue({
+        ok: true,
+        headers: new Map([["content-type", "image/png"]]),
+        arrayBuffer: () => Promise.resolve(new TextEncoder().encode(mockContent).buffer),
+      }) as unknown as typeof fetch;
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: "https://example.com/image.png",
+        prompt: "Fetched image",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.filename).toContain(expectedHash);
+      expect(global.fetch).toHaveBeenCalledWith("https://example.com/image.png");
+    });
+
+    it("should handle failed HTTP fetch", async () => {
+      // Mock fetch to return error
+      global.fetch = vi.fn().mockResolvedValue({
+        ok: false,
+        status: 404,
+        statusText: "Not Found",
+      }) as unknown as typeof fetch;
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: "https://example.com/nonexistent.png",
+        prompt: "Missing image",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(response.status).toBe(500);
+      expect(data.success).toBe(false);
+      expect(data.error).toContain("Failed to fetch content");
+    });
+
+    it("should handle raw base64 without data URL prefix", async () => {
+      const content = "raw-base64-content";
+      const rawBase64 = Buffer.from(content).toString("base64");
+      const expectedHash = computeExpectedHash(Buffer.from(content));
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: rawBase64,
+        prompt: "Raw base64",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.filename).toContain(expectedHash);
+      // Falls back to png when no data URL prefix
+      expect(data.filename.endsWith(".png")).toBe(true);
+    });
+
+    it("should sanitize prompt for filename", async () => {
+      const imageContent = "content-for-sanitize-test";
+      const base64Image = createBase64DataUrl(imageContent, "image/png");
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: base64Image,
+        prompt: "Hello! @World# with $pecial chars%",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      // Prompt should be sanitized - no special chars
+      expect(data.filename).toMatch(/^[a-z0-9_]+_[a-f0-9]+\.png$/);
+    });
+
+    it("should use 'generation' as default prompt snippet when prompt is empty", async () => {
+      const imageContent = "content-no-prompt";
+      const base64Image = createBase64DataUrl(imageContent, "image/png");
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: base64Image,
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.filename).toMatch(/^generation_[a-f0-9]+\.png$/);
+    });
+
+    it("should return 500 on write failure", async () => {
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockRejectedValue(new Error("Disk full"));
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: createBase64DataUrl("content"),
+        prompt: "Test",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(response.status).toBe(500);
+      expect(data.success).toBe(false);
+      expect(data.error).toBe("Disk full");
+    });
+
+    it("should return imageId without extension", async () => {
+      const imageContent = "content-for-id-test";
+      const base64Image = createBase64DataUrl(imageContent, "image/png");
+      const expectedHash = computeExpectedHash(Buffer.from(imageContent));
+
+      mockStat.mockResolvedValue({
+        isDirectory: () => true,
+      });
+      mockReaddir.mockResolvedValue([]);
+      mockWriteFile.mockResolvedValue(undefined);
+
+      const request = createMockPostRequest({
+        directoryPath: "/test/generations",
+        image: base64Image,
+        prompt: "Test",
+      });
+
+      const response = await POST(request);
+      const data = await response.json();
+
+      expect(data.success).toBe(true);
+      expect(data.imageId).not.toContain(".png");
+      expect(data.imageId).toContain(expectedHash);
+    });
+  });
+});