lib: return this from net.Socket.end()

PR-URL: #13481 Reviewed-By: Colin Ihrig <cjihrig@gmail.com> Reviewed-By: Luigi Pinca <luigipinca@gmail.com> Reviewed-By: Refael Ackermann <refack@gmail.com> Reviewed-By: Evan Lucas <evanlucas@me.com> Reviewed-By: Brian White <mscdex@mscdex.net>
nodejs · danbev · Dec 9, 2016 · Apr 20, 2017 · Jun 5, 2017 · Jun 7, 2017
commit 97c290dd380cab939962b07a2a3651173c8cd610
diff --git a/doc/api/net.md b/doc/api/net.md
@@ -575,6 +575,8 @@ server will still send some data.
 If `data` is specified, it is equivalent to calling
 `socket.write(data, encoding)` followed by `socket.end()`.
 
+Returns `socket`.
+
 ### socket.localAddress
 <!-- YAML
 added: v0.9.6

diff --git a/lib/net.js b/lib/net.js
@@ -444,6 +444,8 @@ Socket.prototype.end = function(data, encoding) {
     this.read(0);
   else
     maybeDestroy(this);
+
+  return this;
 };
 
 

diff --git a/test/parallel/test-socket-write-after-fin.js b/test/parallel/test-socket-write-after-fin.js
@@ -2,7 +2,7 @@
 const common = require('../common');
 const assert = require('assert');
 const net = require('net');
-const expected = 'hello1hello2hello3\nTHUNDERMUSCLE!';
+const expected = 'hello1hello2hello3\nbye';
 
 const server = net.createServer({
   allowHalfOpen: true
@@ -35,5 +35,6 @@ server.listen(0, common.mustCall(function() {
   sock.write('hello1');
   sock.write('hello2');
   sock.write('hello3\n');
-  sock.end('THUNDERMUSCLE!');
+  assert.strictEqual(sock.end('bye'), sock);
+
 }));
-Original file line number
+Diff line change
@@ Expand Up / @@ -444,6 +444,8 @@ Socket.prototype.end = function(data, encoding) { @@
         this.read(0);
       else
         maybeDestroy(this);
+      return this;
     };
@@ Expand Down @@