openreader/tests/unit/tts-generate.vitest.spec.ts

import { describe, expect, test } from 'vitest';

import {
  buildReplicateInput,
  buildTTSCacheKey,
  extractReplicateAudioUrl,
  resolveReplicateLanguageValue,
} from '../../src/lib/server/tts/generate';
import { REPLICATE_KOKORO_82M_VERSIONED_MODEL } from '../../src/lib/shared/tts-provider-catalog';

describe('replicate output URL extraction', () => {
  test('returns direct URL string output', () => {
    expect(extractReplicateAudioUrl('https://replicate.delivery/audio.mp3')).toBe(
      'https://replicate.delivery/audio.mp3'
    );
  });

  test('extracts URL from FileOutput-like objects', () => {
    const output = {
      url: () => new URL('https://replicate.delivery/file.wav'),
    };
    expect(extractReplicateAudioUrl(output)).toBe('https://replicate.delivery/file.wav');
  });

  test('extracts first URL from array outputs', () => {
    const output: unknown[] = [
      { value: 'not-a-url' },
      { toString: () => 'https://replicate.delivery/chunk-0.mp3' },
      'https://replicate.delivery/chunk-1.mp3',
    ];
    expect(extractReplicateAudioUrl(output)).toBe('https://replicate.delivery/chunk-0.mp3');
  });

  test('extracts nested URL from object outputs', () => {
    const output = {
      output: {
        audio: {
          url: 'https://replicate.delivery/nested.mp3',
        },
      },
    };
    expect(extractReplicateAudioUrl(output)).toBe('https://replicate.delivery/nested.mp3');
  });

  test('returns null for non-url outputs', () => {
    const output = { status: 'ok', value: 123 };
    expect(extractReplicateAudioUrl(output)).toBeNull();
  });

  test('allows replicate.delivery subdomains', () => {
    expect(extractReplicateAudioUrl('https://pbxt.replicate.delivery/out-0.wav')).toBe(
      'https://pbxt.replicate.delivery/out-0.wav'
    );
  });

  test('rejects URLs from non-replicate hosts (SSRF guard)', () => {
    expect(extractReplicateAudioUrl('http://169.254.169.254/latest/meta-data/')).toBeNull();
    expect(extractReplicateAudioUrl('https://evil.example.com/audio.mp3')).toBeNull();
    // a host that merely embeds the allowed host as a substring must not pass
    expect(extractReplicateAudioUrl('https://replicate.delivery.evil.com/x.mp3')).toBeNull();
    expect(extractReplicateAudioUrl({ url: () => new URL('http://localhost:6379/') })).toBeNull();
  });
});

describe('TTS upstream cache identity', () => {
  test('includes language in the upstream audio cache key', () => {
    const request = {
      text: '同じ文章です。',
      voice: 'jf_alpha',
      speed: 1,
      format: 'mp3',
      model: 'kokoro',
      provider: 'custom-openai',
      apiKey: 'test',
    };

    expect(buildTTSCacheKey({ ...request, language: 'ja' }))
      .not.toBe(buildTTSCacheKey({ ...request, language: 'en' }));
  });
});

describe('Replicate language schema values', () => {
  test('uses language codes or advertised display names without a model table', () => {
    expect(resolveReplicateLanguageValue('ja-JP', [])).toBe('ja-JP');
    expect(resolveReplicateLanguageValue('ja-JP', ['en', 'ja', 'zh'])).toBe('ja');
    expect(resolveReplicateLanguageValue('ja-JP', ['English', 'Japanese'])).toBe('Japanese');
    expect(resolveReplicateLanguageValue('ja-JP', ['English', 'French'])).toBeNull();
  });

  test('includes language_code for the built-in Replicate Kokoro model', async () => {
    await expect(buildReplicateInput({
      text: 'Hello world',
      voice: 'af_sarah',
      speed: 1,
      format: 'mp3',
      model: REPLICATE_KOKORO_82M_VERSIONED_MODEL,
      language: 'en',
      provider: 'replicate',
      apiKey: 'r8_token',
      testNamespace: null,
    })).resolves.toEqual({
      text: 'Hello world',
      voice: 'af_sarah',
      language_code: 'a',
    });

    await expect(buildReplicateInput({
      text: 'Hello world',
      voice: 'bf_emma',
      speed: 1,
      format: 'mp3',
      model: REPLICATE_KOKORO_82M_VERSIONED_MODEL,
      language: 'en',
      provider: 'replicate',
      apiKey: 'r8_token',
      testNamespace: null,
    })).resolves.toEqual({
      text: 'Hello world',
      voice: 'bf_emma',
      language_code: 'b',
    });

    await expect(buildReplicateInput({
      text: 'Hello world',
      voice: 'af_sarah',
      speed: 1,
      format: 'mp3',
      model: REPLICATE_KOKORO_82M_VERSIONED_MODEL,
      provider: 'replicate',
      apiKey: 'r8_token',
      testNamespace: null,
    })).resolves.toEqual({
      text: 'Hello world',
      voice: 'af_sarah',
      language_code: 'a',
    });
  });
});