From 76a0ba689a0da6f0e431f1c2dc8de4439433dfc1 Mon Sep 17 00:00:00 2001
From: Robert Nagy <ronagy@icloud.com>
Date: Tue, 13 Nov 2018 23:25:51 +0100
Subject: [PATCH] http2: make compat finished match http/1

finished should true directly after end().

PR-URL: https://github.com/nodejs/node/pull/24347
Refs: https://github.com/nodejs/node/issues/24743
Reviewed-By: Matteo Collina <matteo.collina@gmail.com>
Reviewed-By: Ujjwal Sharma <ryzokuken@disroot.org>
Reviewed-By: Anatoli Papirovski <apapirovski@mac.com>
Reviewed-By: James M Snell <jasnell@gmail.com>
---
 lib/internal/http2/compat.js                          | 11 ++++-------
 test/parallel/test-http2-compat-serverresponse-end.js |  4 +++-
 2 files changed, 7 insertions(+), 8 deletions(-)

diff --git a/lib/internal/http2/compat.js b/lib/internal/http2/compat.js
index 8ef5f49a3dbd77..479510152c9522 100644
--- a/lib/internal/http2/compat.js
+++ b/lib/internal/http2/compat.js
@@ -471,10 +471,8 @@ class Http2ServerResponse extends Stream {
   }
 
   get finished() {
-    const stream = this[kStream];
-    return stream.destroyed ||
-           stream._writableState.ended ||
-           this[kState].closed;
+    const state = this[kState];
+    return state.ending;
   }
 
   get socket() {
@@ -700,12 +698,11 @@ class Http2ServerResponse extends Stream {
     if (chunk !== null && chunk !== undefined)
       this.write(chunk, encoding);
 
-    const isFinished = this.finished;
     state.headRequest = stream.headRequest;
     state.ending = true;
 
     if (typeof cb === 'function') {
-      if (isFinished)
+      if (stream.writableEnded)
         this.once('finish', cb);
       else
         stream.once('finish', cb);
@@ -714,7 +711,7 @@ class Http2ServerResponse extends Stream {
     if (!stream.headersSent)
       this.writeHead(this[kState].statusCode);
 
-    if (isFinished)
+    if (this[kState].closed || stream.destroyed)
       onStreamCloseResponse.call(stream);
     else
       stream.end();
diff --git a/test/parallel/test-http2-compat-serverresponse-end.js b/test/parallel/test-http2-compat-serverresponse-end.js
index 5bbb24bb2edb31..8505d6c4969db1 100644
--- a/test/parallel/test-http2-compat-serverresponse-end.js
+++ b/test/parallel/test-http2-compat-serverresponse-end.js
@@ -149,11 +149,13 @@ const {
   // Http2ServerResponse.end is necessary on HEAD requests in compat
   // for http1 compatibility
   const server = createServer(mustCall((request, response) => {
-    strictEqual(response.finished, true);
     strictEqual(response.writableEnded, false);
+    strictEqual(response.finished, false);
     response.writeHead(HTTP_STATUS_OK, { foo: 'bar' });
+    strictEqual(response.finished, false);
     response.end('data', mustCall());
     strictEqual(response.writableEnded, true);
+    strictEqual(response.finished, true);
   }));
   server.listen(0, mustCall(() => {
     const { port } = server.address();