msg.finished is only finished if socket has been detached

jshttp · Jun 25, 2019 · 200fb77 · 200fb77
1 parent 08345db
commit 200fb77
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/index.js b/index.js
@@ -64,10 +64,17 @@ function onFinished (msg, listener) {
 
 function isFinished (msg) {
   var socket = msg.socket
+  var stream = msg.stream
+
+  if (stream && typeof stream.closed === 'boolean') {
+    // Http2ServerRequest
+    // Http2ServerResponse
+    return stream.closed
+  }
 
   if (typeof msg.finished === 'boolean') {
     // OutgoingMessage
-    return Boolean(msg.finished || (socket && !socket.writable))
+    return Boolean((!socket && msg.finished && msg.outputSize === 0) || (socket && !socket.writable))
   }
 
   if (typeof msg.complete === 'boolean') {

diff --git a/test/test.js b/test/test.js
@@ -219,8 +219,8 @@ describe('isFinished(res)', function () {
 
   it('should be false before response finishes', function (done) {
     var server = http.createServer(function (req, res) {
-      assert.ok(!onFinished.isFinished(res))
       res.end()
+      assert.ok(!onFinished.isFinished(res))
       done()
     })