add .end()

Author: Elexy

Diff for: lib/connection.js

@@ -309,7 +309,8 @@ Connection.prototype.end = function () {
   // 0x58 = 'X'
   this.writer.add(emptyBuffer)
   this._ending = true
-  return this.stream.write(END_BUFFER)
+  this.stream.write(END_BUFFER)
+  return this.stream.end()
 }
 
 Connection.prototype.close = function (msg, more) {

Diff for: test/unit/test-helper.js

@@ -17,6 +17,8 @@ p.write = function (packet) {
   this.packets.push(packet)
 }
 
+p.end = function () {}
+
 p.setKeepAlive = function () {}
 
 p.writable = true