fix(stream): Start capturing incoming messages before stream is read (#…

…76)

Co-authored-by: kodiakhq[bot] <49736102+kodiakhq[bot]@users.noreply.github.com>

src/lib/MockWebSocket.ts

@@ -73,16 +73,16 @@ export class MockWebSocket extends EventEmitter {
         callback(error);
       },
       read(_size: number): void {
-        connection.once('message', (message) => {
-          duplex.push(message);
-        });
+        // Do nothing. We're already recording incoming messages.
       },
       write(chunk: any, _encoding: string, callback: (error?: Error | null) => void): void {
         connection.send(chunk);
         callback();
       },
     });
 
+    this.on('message', (message) => duplex.push(message));
+
     this.on('close', () => duplex.push(null));
 
     this.on('error', (error) => duplex.destroy(error));