fix: end stream on Connection.end()

lib/connection.js

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

test/unit/connection/outbound-sending-tests.js

@@ -183,6 +183,7 @@ test('sends end command', function () {
   con.end()
   var expected = Buffer.from([0x58, 0, 0, 0, 4])
   assert.received(stream, expected)
+  assert.equal(stream.closed, true)
 })
 
 test('sends describe command', function () {

test/unit/test-helper.js

@@ -13,12 +13,19 @@ helper.sys.inherits(MemoryStream, EventEmitter)
 
 var p = MemoryStream.prototype
 
-p.write = function (packet) {
+p.write = function (packet, cb) {
   this.packets.push(packet)
+  if(cb){
+    cb();
+  }
 }
 
-p.setKeepAlive = function () {}
+p.end = function() {
+  p.closed = true;
+}
 
+p.setKeepAlive = function () {}
+p.closed = false;
 p.writable = true
 
 const createClient = function () {