tests: improve streaming test robustness

mysqljs · Jul 10, 2015 · e1ebca0 · e1ebca0
1 parent f0dfda8
commit e1ebca0
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/FakeServer.js b/test/FakeServer.js
@@ -386,7 +386,12 @@ FakeConnection.prototype._writePacketStream = function _writePacketStream(count)
     var writer = new PacketWriter();
     writer.writeLengthCodedString(num);
     writer.writeLengthCodedString('Row #' + num);
-    this._socket.write(writer.toBuffer(this._parser));
+
+    try {
+      this._socket.write(writer.toBuffer(this._parser));
+    } catch (e) {
+      cleanup.call(this);
+    }
   }
 };