feat: Close writable stream too when it's closed

relaycorp · Jul 28, 2021 · 95bf3f3 · 95bf3f3
1 parent 979bd77
commit 95bf3f3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/lib/MockWebSocket.ts b/src/lib/MockWebSocket.ts
@@ -107,9 +107,9 @@ export class MockWebSocket extends EventEmitter {
 
     this.on('message', (message) => duplex.push(message));
 
-    this.on('close', () => duplex.push(null));
+    this.once('close', () => duplex.destroy());
 
-    this.on('error', (error) => duplex.destroy(error));
+    this.once('error', (error) => duplex.destroy(error));
 
     return duplex;
   }