sofa/packages/core/test/thumbhash.test.ts

import { afterEach, beforeEach, describe, expect, test, vi } from "vitest";

import { episodes, seasons, titles } from "@sofa/db/schema";
import { clearAllTables, eq, testDb } from "@sofa/test/db";

const TINY_PNG = Buffer.from(
  "iVBORw0KGgoAAAANSUhEUgAAAAIAAAACCAYAAABytg0kAAAACXBIWXMAAAPoAAAD6AG1e1JrAAAAEUlEQVQImWP4z8DwH4QZYAwAR8oH+Xm0fdIAAAAASUVORK5CYII=",
  "base64",
);

let nextBuffer: Buffer | null = TINY_PNG;

import { generateEpisodeThumbHash, generateThumbHash } from "../src/thumbhash";

beforeEach(() => {
  clearAllTables();
  process.env.IMAGE_CACHE_ENABLED = "false";
  nextBuffer = TINY_PNG;
  vi.spyOn(globalThis, "fetch").mockImplementation((async (
    _input: string | URL | Request,
    _init?: RequestInit,
  ) => {
    if (!nextBuffer) {
      return new Response(null, { status: 404 });
    }

    return new Response(nextBuffer, {
      status: 200,
      headers: { "content-type": "image/png" },
    });
  }) as typeof fetch);
});

afterEach(() => {
  delete process.env.IMAGE_CACHE_ENABLED;
});

describe("thumbhash generation", () => {
  test("generates a thumbhash from a loaded image buffer", async () => {
    const hash = await generateThumbHash("/poster.png", "posters");
    expect(hash).toBeTypeOf("string");
    expect(hash).not.toBe("");
  });

  test("clears a stale episode thumbhash when regeneration fails", async () => {
    testDb.insert(titles).values({ id: "tv-1", tmdbId: 100, type: "tv", title: "Show" }).run();
    testDb.insert(seasons).values({ id: "season-1", titleId: "tv-1", seasonNumber: 1 }).run();
    testDb
      .insert(episodes)
      .values({
        id: "ep-1",
        seasonId: "season-1",
        episodeNumber: 1,
        stillPath: "/old-still.png",
        stillThumbHash: "stale-hash",
      })
      .run();

    nextBuffer = null;
    const hash = await generateEpisodeThumbHash("ep-1", "/new-still.png");

    const episode = testDb.select().from(episodes).where(eq(episodes.id, "ep-1")).get();

    expect(hash).toBeNull();
    expect(episode?.stillThumbHash).toBeNull();
  });
});