@vellumai/assistant 0.5.3 → 0.5.5

package/src/__tests__/openai-whisper.test.ts

@@ -0,0 +1,93 @@
+import { afterEach, beforeEach, describe, expect, test } from "bun:test";
+
+import { OpenAIWhisperProvider } from "../providers/speech-to-text/openai-whisper.js";
+
+// ---------------------------------------------------------------------------
+// Mock fetch — capture outgoing FormData so we can assert filenames
+// ---------------------------------------------------------------------------
+
+const originalFetch = globalThis.fetch;
+
+let capturedFormData: FormData | null = null;
+
+function mockFetch(
+  _url: string | URL | Request,
+  init?: RequestInit,
+): Promise<Response> {
+  if (init?.body instanceof FormData) {
+    capturedFormData = init.body;
+  }
+  return Promise.resolve(
+    new Response(JSON.stringify({ text: "hello world" }), {
+      status: 200,
+      headers: { "Content-Type": "application/json" },
+    }),
+  );
+}
+
+beforeEach(() => {
+  capturedFormData = null;
+  globalThis.fetch = mockFetch as typeof fetch;
+});
+
+afterEach(() => {
+  globalThis.fetch = originalFetch;
+});
+
+// ---------------------------------------------------------------------------
+// Tests
+// ---------------------------------------------------------------------------
+
+describe("OpenAIWhisperProvider", () => {
+  const provider = new OpenAIWhisperProvider("test-api-key");
+  const dummyAudio = Buffer.from("fake-audio-data");
+
+  describe("extensionFromMime (via transcribe filename)", () => {
+    test("plain MIME type resolves to correct extension", async () => {
+      await provider.transcribe(dummyAudio, "audio/ogg");
+      const file = capturedFormData?.get("file") as File;
+      expect(file.name).toBe("audio.ogg");
+    });
+
+    test("MIME type with parameters resolves to correct extension", async () => {
+      await provider.transcribe(dummyAudio, "audio/ogg; codecs=opus");
+      const file = capturedFormData?.get("file") as File;
+      expect(file.name).toBe("audio.ogg");
+    });
+
+    test("MIME type with extra whitespace around parameters", async () => {
+      await provider.transcribe(dummyAudio, "audio/mpeg ; bitrate=128");
+      const file = capturedFormData?.get("file") as File;
+      expect(file.name).toBe("audio.mp3");
+    });
+
+    test("unknown MIME type falls back to .audio", async () => {
+      await provider.transcribe(dummyAudio, "audio/unknown-format");
+      const file = capturedFormData?.get("file") as File;
+      expect(file.name).toBe("audio.audio");
+    });
+
+    test("unknown MIME type with parameters still falls back", async () => {
+      await provider.transcribe(dummyAudio, "audio/unknown; foo=bar");
+      const file = capturedFormData?.get("file") as File;
+      expect(file.name).toBe("audio.audio");
+    });
+
+    test.each([
+      ["audio/wav", "audio.wav"],
+      ["audio/x-wav", "audio.wav"],
+      ["audio/mpeg", "audio.mp3"],
+      ["audio/mp3", "audio.mp3"],
+      ["audio/ogg", "audio.ogg"],
+      ["audio/opus", "audio.opus"],
+      ["audio/webm", "audio.webm"],
+      ["audio/mp4", "audio.m4a"],
+      ["audio/x-m4a", "audio.m4a"],
+      ["audio/flac", "audio.flac"],
+    ])("%s → %s", async (mime, expectedFilename) => {
+      await provider.transcribe(dummyAudio, mime);
+      const file = capturedFormData?.get("file") as File;
+      expect(file.name).toBe(expectedFilename);
+    });
+  });
+});