modify mock to try to fix html test

Author: pranavjoshi001

__tests__/assets/esm/speechToSpeech/mockMediaDevices.js

@@ -1,32 +1,76 @@
-/* global AudioContext, navigator */
+/* global MessageChannel, navigator, setTimeout, URL, window */
 
 /**
- * Mocks navigator.mediaDevices.getUserMedia for testing speechToSpeech functionality.
+ * Mocks browser audio APIs for speechToSpeech testing.
+ *
+ * - Intercepts AudioContext.audioWorklet.addModule() to prevent blob execution
+ * - Mocks AudioWorkletNode for the 'audio-recorder' processor
+ * - Mocks navigator.mediaDevices.getUserMedia() to return a test audio stream
  */
 export function setupMockMediaDevices() {
-  if (!navigator.mediaDevices) {
-    navigator.mediaDevices = {};
-  }
+  const OriginalAudioContext = window.AudioContext;
 
+  // Intercept AudioContext to mock audioWorklet.addModule
+  window.AudioContext = function (options) {
+    const ctx = new OriginalAudioContext(options);
+
+    ctx.audioWorklet.addModule = url => {
+      if (url.startsWith('blob:')) {
+        URL.revokeObjectURL(url);
+      }
+      return Promise.resolve();
+    };
+
+    return ctx;
+  };
+
+  Object.setPrototypeOf(window.AudioContext, OriginalAudioContext);
+  window.AudioContext.prototype = OriginalAudioContext.prototype;
+
+  // Mock AudioWorkletNode - uses GainNode as base so source.connect() works
+  window.AudioWorkletNode = function (context, name, options) {
+    const node = context.createGain();
+    const channel = new MessageChannel();
+    let recording = false;
+
+    node.port = channel.port1;
+
+    channel.port2.onmessage = ({ data }) => {
+      if (data.command === 'START') {
+        recording = true;
+        const bufferSize = options?.processorOptions?.bufferSize || 2400;
+        setTimeout(() => {
+          if (recording) {
+            channel.port1.postMessage({ eventType: 'audio', audioData: new Float32Array(bufferSize) });
+          }
+        }, 100);
+      } else if (data.command === 'STOP') {
+        recording = false;
+      }
+    };
+
+    return node;
+  };
+
+  // Mock getUserMedia with oscillator-based test stream
   navigator.mediaDevices.getUserMedia = constraints => {
-    const audioContext = new AudioContext({ sampleRate: constraints?.audio?.sampleRate || 24000 });
-    const oscillator = audioContext.createOscillator();
-    const destination = audioContext.createMediaStreamDestination();
+    const sampleRate = constraints?.audio?.sampleRate || 24000;
+    const ctx = new OriginalAudioContext({ sampleRate });
+    const oscillator = ctx.createOscillator();
+    const destination = ctx.createMediaStreamDestination();
 
     oscillator.connect(destination);
     oscillator.start();
 
-    const { stream } = destination;
-
-    stream.getTracks().forEach(track => {
+    destination.stream.getTracks().forEach(track => {
       const originalStop = track.stop.bind(track);
       track.stop = () => {
         oscillator.stop();
-        audioContext.close();
+        ctx.close();
         originalStop();
       };
     });
 
-    return stream;
+    return Promise.resolve(destination.stream);
   };
 }