tts tests

nbonamy · Jan 17, 2025 · 7ad7353 · 7ad7353
1 parent 60d461f
commit 7ad7353
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 5 deletions.
diff --git a/src/voice/tts-replicate.ts b/src/voice/tts-replicate.ts
@@ -1,5 +1,5 @@
 
-import { Configuration } from '../types/config'
+/*import { Configuration } from '../types/config'
 import { SynthesisResponse, TTSEngine } from './tts'
 import Replicate, { FileOutput, Prediction } from 'replicate'
 
@@ -54,4 +54,4 @@ export default class TTSReplicate implements TTSEngine {
         reader.readAsDataURL(blob)
     })
   }
-}
+}*/
diff --git a/tests/unit/tts.test.ts b/tests/unit/tts.test.ts
@@ -14,12 +14,35 @@ vi.mock('openai', async () => {
   return { default : OpenAI }
 })
 
+vi.mock('elevenlabs', async () => {
+  const ElevenLabsClient = vi.fn()
+  ElevenLabsClient.prototype.textToSpeech = {
+    convertAsStream: vi.fn((_, opts) => opts.text)
+  }
+  return { ElevenLabsClient }
+})
+
 beforeEach(() => {
   store.config = defaults
 })
 
-test('Synthetizes text', async () => {
+test('OpenAI', async () => {
+  const tts = getTTSEngine(store.config)
+  const response = await tts.synthetize('hello openai')
+  expect(response).toStrictEqual({ type: 'audio', content: 'hello openai' })
+})
+
+test('ElevenLabs', async () => {
+  store.config.tts.engine = 'elevenlabs'
   const tts = getTTSEngine(store.config)
-  const response = await tts.synthetize('hello')
-  expect(response).toStrictEqual({ type: 'audio', content: 'hello' })
+  const response = await tts.synthetize('hello elevenlabs')
+  expect(response).toStrictEqual({ type: 'audio', content: 'hello elevenlabs' })
 })
+
+test('Kokoro', async () => {
+  store.config.tts.engine = 'kokoro'
+  const tts = getTTSEngine(store.config)
+  const response = await tts.synthetize('hello kokoro')
+  expect(response).toStrictEqual({ type: 'audio', content: 'hello kokoro' })
+})
+