fix(jest): add missing methods to mock

mybigday · Oct 2, 2023 · 2854265 · 2854265
1 parent a66e51a
commit 2854265
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 2 deletions.
diff --git a/jest/mock.js b/jest/mock.js
@@ -48,6 +48,16 @@ if (!NativeModules.RNWhisper) {
     releaseContext: jest.fn(() => Promise.resolve()),
     releaseAllContexts: jest.fn(() => Promise.resolve()),
 
+    // iOS AudioSession utils
+    getAudioSessionCurrentCategory: jest.fn(() => Promise.resolve({ 
+      category: 'AVAudioSessionCategoryPlayAndRecord',
+      options: [],
+    })),
+    getAudioSessionCurrentMode: jest.fn(() => Promise.resolve('')),
+    setAudioSessionCategory: jest.fn(() => Promise.resolve()),
+    setAudioSessionMode: jest.fn(() => Promise.resolve()),
+    setAudioSessionActive: jest.fn(() => Promise.resolve()),
+
     // For NativeEventEmitter
     addListener: jest.fn(),
     removeListeners: jest.fn(),

diff --git a/src/__tests__/index.test.ts b/src/__tests__/index.test.ts
@@ -1,4 +1,4 @@
-import { initWhisper, releaseAllWhisper } from '..'
+import { initWhisper, releaseAllWhisper, AudioSessionIos } from '..'
 
 jest.mock('..', () => require('../../jest/mock'))
 
@@ -16,7 +16,17 @@ test('Mock', async () => {
     segments: [{ text: ' Test', t0: 0, t1: 33 }],
   })
 
-  const { subscribe } = await context.transcribeRealtime()
+  const { subscribe } = await context.transcribeRealtime({
+    audioSessionOnStartIos: {
+      category: AudioSessionIos.Category.PlayAndRecord,
+      options: [
+        AudioSessionIos.CategoryOption.MixWithOthers,
+        AudioSessionIos.CategoryOption.AllowBluetooth,
+      ],
+      mode: AudioSessionIos.Mode.Default,
+    },
+    audioSessionOnStopIos: 'restore',
+  })
   const events: any[] = []
   subscribe((event) => events.push(event))
   await new Promise((resolve) => setTimeout(resolve, 0))