openclaw/extensions/microsoft/tts.test.ts

import { mkdtempSync, rmSync, writeFileSync } from "node:fs";
import { tmpdir } from "node:os";
import path from "node:path";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";

let edgeTTS: typeof import("./tts.js").edgeTTS;

let mockTtsPromise = vi.fn<(text: string, filePath: string) => Promise<void>>();

vi.mock("node-edge-tts", () => ({
  EdgeTTS: class {
    ttsPromise(text: string, filePath: string) {
      return mockTtsPromise(text, filePath);
    }
  },
}));

const baseEdgeConfig = {
  voice: "en-US-MichelleNeural",
  lang: "en-US",
  outputFormat: "audio-24khz-48kbitrate-mono-mp3",
  saveSubtitles: false,
};

describe("edgeTTS empty audio validation", () => {
  let tempDir: string | undefined;

  beforeEach(async () => {
    vi.resetModules();
    vi.doMock("node-edge-tts", () => ({
      EdgeTTS: class {
        ttsPromise(text: string, filePath: string) {
          return mockTtsPromise(text, filePath);
        }
      },
    }));
    ({ edgeTTS } = await import("./tts.js"));
  });

  afterEach(() => {
    if (tempDir) {
      rmSync(tempDir, { recursive: true, force: true });
      tempDir = undefined;
    }
  });

  it("throws when the output file is 0 bytes", async () => {
    tempDir = mkdtempSync(path.join(tmpdir(), "tts-test-"));
    const outputPath = path.join(tempDir, "voice.mp3");

    mockTtsPromise = vi.fn(async (_text: string, filePath: string) => {
      writeFileSync(filePath, "");
    });

    await expect(
      edgeTTS({
        text: "Hello",
        outputPath,
        config: baseEdgeConfig,
        timeoutMs: 10000,
      }),
    ).rejects.toThrow("Edge TTS produced empty audio file");
  });

  it("succeeds when the output file has content", async () => {
    tempDir = mkdtempSync(path.join(tmpdir(), "tts-test-"));
    const outputPath = path.join(tempDir, "voice.mp3");

    mockTtsPromise = vi.fn(async (_text: string, filePath: string) => {
      writeFileSync(filePath, Buffer.from([0xff, 0xfb, 0x90, 0x00]));
    });

    await expect(
      edgeTTS({
        text: "Hello",
        outputPath,
        config: baseEdgeConfig,
        timeoutMs: 10000,
      }),
    ).resolves.toBeUndefined();
  });
});