tests: improve streaming test robustness

mysqljs · Jul 10, 2015 · 1e982fb · 1e982fb
1 parent f0dfda8
commit 1e982fb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/FakeServer.js b/test/FakeServer.js
@@ -365,15 +365,15 @@ FakeConnection.prototype._writePacketStream = function _writePacketStream(count)
   this._sendPacket(new Packets.EofPacket());
 
   function cleanup() {
-    var socket = this._socket || this;
-    socket.removeListener('close', cleanup);
-    socket.removeListener('error', cleanup);
     clearInterval(timer);
   }
 
   function writeRow() {
     if (remaining === 0) {
-      cleanup.call(this);
+      cleanup();
+
+      this._socket.removeListener('close', cleanup);
+      this._socket.removeListener('error', cleanup);
 
       this._sendPacket(new Packets.EofPacket());
       this._parser.resetPacketNumber();