SnapOtter/tests/integration/trim-audio.test.ts

import { mkdtempSync, readFileSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import { join } from "node:path";
import { ffmpegAvailable, probeMedia } from "@snapotter/media-engine";
import { afterAll, beforeAll, describe, expect, it } from "vitest";
import { buildTestApp, createMultipartPayload, loginAsAdmin, type TestApp } from "./test-server.js";

const MP3 = readFileSync(join(__dirname, "..", "fixtures", "media", "tiny.mp3"));

let testApp: TestApp;
let adminToken: string;

beforeAll(async () => {
  testApp = await buildTestApp();
  adminToken = await loginAsAdmin(testApp.app);
}, 30_000);

afterAll(async () => {
  await testApp.cleanup();
}, 10_000);

async function runTool(settings: Record<string, unknown>) {
  const { body, contentType } = createMultipartPayload([
    { name: "file", filename: "tiny.mp3", contentType: "audio/mpeg", content: MP3 },
    { name: "settings", content: JSON.stringify(settings) },
  ]);
  return testApp.app.inject({
    method: "POST",
    url: "/api/v1/tools/trim-audio",
    headers: { authorization: `Bearer ${adminToken}`, "content-type": contentType },
    body,
  });
}

describe.skipIf(!ffmpegAvailable())("trim-audio (requires ffmpeg)", () => {
  it("trims mp3 from 0 to 0.5s and returns 200", async () => {
    const res = await runTool({ startS: 0, endS: 0.5 });
    expect(res.statusCode).toBe(200);
    const envelope = JSON.parse(res.body);
    expect(envelope.downloadUrl).toBeDefined();
    const dl = await testApp.app.inject({
      method: "GET",
      url: envelope.downloadUrl,
    });
    expect(dl.statusCode).toBe(200);
    expect(dl.rawPayload.length).toBeGreaterThan(50);

    // Verify trimmed duration via probeMedia (mirror trim-video's pattern)
    const tmpDir = mkdtempSync(join(tmpdir(), "trim-audio-test-"));
    const probeFile = join(tmpDir, "trimmed.mp3");
    writeFileSync(probeFile, dl.rawPayload);
    const info = await probeMedia(probeFile);
    expect(info.durationS).not.toBeNull();
    // Stream-copy on mp3 may overshoot slightly due to frame boundaries,
    // but should be well under the original 1s duration.
    expect(info.durationS as number).toBeLessThanOrEqual(2);
    expect(info.durationS as number).toBeGreaterThan(0);
  }, 60_000);

  it("rejects when end is before start", async () => {
    const res = await runTool({ startS: 0.5, endS: 0.2 });
    expect(res.statusCode).toBe(400);
  });
});