one/dist/esm/vite/plugins/imageDataPlugin.test.mjs

One is a new React Framework that makes Vite serve both native and web.

import { resolve } from "node:path";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
const PROJECT_ROOT = resolve("/project");
const PUBLIC_DIR = resolve("/project/public");
const COMPONENTS_DIR = resolve("/project/src/components");
const HERO_FILE = resolve("/project/src/components/Hero.tsx");
vi.mock("node:fs", async () => {
  const actual = await vi.importActual("node:fs");
  return {
    ...actual,
    existsSync: vi.fn(path => {
      return path.includes("test-image.jpg") || path.includes("hero.png");
    })
  };
});
function mockConfig() {
  return {
    publicDir: PUBLIC_DIR,
    root: PROJECT_ROOT
  };
}
describe("imageDataPlugin", () => {
  beforeEach(() => {
    vi.resetModules();
  });
  afterEach(() => {
    vi.clearAllMocks();
  });
  describe("resolveId", () => {
    it("filters out ids without a trailing ?imagedata", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      const filter = plugin.resolveId.filter.id;
      expect(filter.test("./image.jpg")).toBe(false);
      expect(filter.test("./image?imagedata.jpg")).toBe(false);
      expect(filter.test("./image.jpg?imagedata")).toBe(true);
    });
    it("should resolve public dir paths starting with /", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("/test-image.jpg?imagedata", void 0);
      expect(result).toBe("\0imagedata:" + resolve(PUBLIC_DIR, "test-image.jpg"));
    });
    it("should resolve relative imports", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("./test-image.jpg?imagedata", HERO_FILE);
      expect(result).toBe("\0imagedata:" + resolve(COMPONENTS_DIR, "test-image.jpg"));
    });
    it("should return null for non-existent files", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("/nonexistent.jpg?imagedata", void 0);
      expect(result).toBeNull();
    });
  });
  describe("path traversal prevention", () => {
    it("should block path traversal from public dir", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("/../../../etc/passwd?imagedata", void 0);
      expect(result).toBeNull();
    });
    it("should block path traversal from relative imports", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("../../../../etc/passwd?imagedata", HERO_FILE);
      expect(result).toBeNull();
    });
    it("should block path traversal from root", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("../../../etc/passwd?imagedata", void 0);
      expect(result).toBeNull();
    });
    it("should allow paths within project boundaries", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const result = await plugin.resolveId.handler("../../src/test-image.jpg?imagedata", HERO_FILE);
      expect(result).toBe("\0imagedata:" + resolve(PROJECT_ROOT, "src/test-image.jpg"));
    });
  });
  describe("load", () => {
    it("filters out non-virtual modules before the handler runs", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      const filter = plugin.load.filter.id;
      expect(filter.test("./image.jpg")).toBe(false);
      expect(filter.test("\0imagedata:/abs/path/image.jpg")).toBe(true);
    });
    it("should return fallback data when sharp fails", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      if (plugin.configResolved) {
        ;
        plugin.configResolved(mockConfig());
      }
      const mockContext = {
        addWatchFile: vi.fn()
      };
      const testFilePath = resolve(PUBLIC_DIR, "test-image.jpg");
      const result = await plugin.load.handler.call(mockContext, "\0imagedata:" + testFilePath);
      expect(result).toContain("export default");
      expect(result).toContain('"src":"/test-image.jpg"');
      expect(result).toContain('"width":');
      expect(result).toContain('"height":');
      expect(mockContext.addWatchFile).toHaveBeenCalledWith(testFilePath);
    });
  });
  describe("plugin metadata", () => {
    it("should have correct name", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      expect(plugin.name).toBe("one:imagedata");
    });
    it("should enforce pre", async () => {
      const {
        imageDataPlugin
      } = await import("./imageDataPlugin.mjs");
      const plugin = imageDataPlugin();
      expect(plugin.enforce).toBe("pre");
    });
  });
});
describe("imageDataPlugin output format", () => {
  it("should export a valid JSON object", async () => {
    vi.resetModules();
    const {
      imageDataPlugin
    } = await import("./imageDataPlugin.mjs");
    const plugin = imageDataPlugin();
    if (plugin.configResolved) {
      ;
      plugin.configResolved(mockConfig());
    }
    const mockContext = {
      addWatchFile: vi.fn()
    };
    const testFilePath = resolve(PUBLIC_DIR, "test-image.jpg");
    const result = await plugin.load.handler.call(mockContext, "\0imagedata:" + testFilePath);
    const jsonMatch = result?.match(/export default (.+)$/);
    expect(jsonMatch).toBeTruthy();
    if (jsonMatch) {
      const parsed = JSON.parse(jsonMatch[1]);
      expect(parsed).toHaveProperty("src");
      expect(parsed).toHaveProperty("width");
      expect(parsed).toHaveProperty("height");
      expect(parsed).toHaveProperty("blurDataURL");
      expect(typeof parsed.src).toBe("string");
      expect(typeof parsed.width).toBe("number");
      expect(typeof parsed.height).toBe("number");
      expect(typeof parsed.blurDataURL).toBe("string");
    }
  });
});
//# sourceMappingURL=imageDataPlugin.test.mjs.map