From da17a2538e3c5283c17211f1298cb1a915048c7a Mon Sep 17 00:00:00 2001
From: Pulkit Gupta <pulkit30.bsr@gmail.com>
Date: Mon, 18 Dec 2023 02:45:02 +0530
Subject: [PATCH] test_runner: fixed to run after hook if before throws an
 error

PR-URL: https://github.com/nodejs/node/pull/51062
Refs: https://github.com/nodejs/node/issues/50842
Reviewed-By: Moshe Atlow <moshe@atlow.co.il>
Reviewed-By: Raz Luvaton <rluvaton@gmail.com>
---
 lib/internal/test_runner/test.js              |  4 +-
 test/fixtures/test-runner/output/hooks.js     |  8 ++++
 .../test-runner/output/hooks.snapshot         | 39 ++++++++++++++++--
 .../output/hooks_spec_reporter.snapshot       | 41 +++++++++++++++++--
 4 files changed, 84 insertions(+), 8 deletions(-)

diff --git a/lib/internal/test_runner/test.js b/lib/internal/test_runner/test.js
index 4eb02c5d4e3118..19ae283ed9ad78 100644
--- a/lib/internal/test_runner/test.js
+++ b/lib/internal/test_runner/test.js
@@ -924,6 +924,7 @@ class Suite extends Test {
     const hookArgs = this.getRunArgs();
 
     let stopPromise;
+    const after = runOnce(() => this.runHook('after', hookArgs));
     try {
       this.parent.activeSubtests++;
       await this.buildSuite;
@@ -946,10 +947,11 @@ class Suite extends Test {
       const promise = SafePromiseAll(subtests, (subtests) => subtests.start());
 
       await SafePromiseRace([promise, stopPromise]);
-      await this.runHook('after', hookArgs);
+      await after();
 
       this.pass();
     } catch (err) {
+      try { await after(); } catch { /* suite is already failing */ }
       if (isTestFailureError(err)) {
         this.fail(err);
       } else {
diff --git a/test/fixtures/test-runner/output/hooks.js b/test/fixtures/test-runner/output/hooks.js
index 00a6c23499f52e..531615fd97d69b 100644
--- a/test/fixtures/test-runner/output/hooks.js
+++ b/test/fixtures/test-runner/output/hooks.js
@@ -166,5 +166,13 @@ test('t.after() is called if test body throws', (t) => {
   throw new Error('bye');
 });
 
+describe('run after when before throws', () => {
+  after(common.mustCall(() => {
+    console.log("- after() called")
+  }));
+  before(() => { throw new Error('before')});
+  it('1', () => {});
+});
+
 before((t) => t.diagnostic('before 2 called'));
 after((t) => t.diagnostic('after 2 called'));
diff --git a/test/fixtures/test-runner/output/hooks.snapshot b/test/fixtures/test-runner/output/hooks.snapshot
index 6cf29612c535cb..edb455b025ff58 100644
--- a/test/fixtures/test-runner/output/hooks.snapshot
+++ b/test/fixtures/test-runner/output/hooks.snapshot
@@ -1,3 +1,4 @@
+- after() called
 TAP version 13
 # Subtest: describe hooks
     # Subtest: 1
@@ -568,16 +569,46 @@ not ok 14 - t.after() is called if test body throws
     *
   ...
 # - after() called
-1..14
+# Subtest: run after when before throws
+    # Subtest: 1
+    not ok 1 - 1
+      ---
+      duration_ms: *
+      location: '/test/fixtures/test-runner/output/hooks.js:(LINE):3'
+      failureType: 'cancelledByParent'
+      error: 'test did not finish before its parent and was cancelled'
+      code: 'ERR_TEST_FAILURE'
+      ...
+    1..1
+not ok 15 - run after when before throws
+  ---
+  duration_ms: *
+  type: 'suite'
+  location: '/test/fixtures/test-runner/output/hooks.js:(LINE):1'
+  failureType: 'hookFailed'
+  error: 'before'
+  code: 'ERR_TEST_FAILURE'
+  stack: |-
+    *
+    *
+    *
+    *
+    *
+    *
+    *
+    *
+    *
+  ...
+1..15
 # before 1 called
 # before 2 called
 # after 1 called
 # after 2 called
-# tests 38
-# suites 8
+# tests 39
+# suites 9
 # pass 14
 # fail 22
-# cancelled 2
+# cancelled 3
 # skipped 0
 # todo 0
 # duration_ms *
diff --git a/test/fixtures/test-runner/output/hooks_spec_reporter.snapshot b/test/fixtures/test-runner/output/hooks_spec_reporter.snapshot
index 82c4704df4e63f..4cd47bf042b95d 100644
--- a/test/fixtures/test-runner/output/hooks_spec_reporter.snapshot
+++ b/test/fixtures/test-runner/output/hooks_spec_reporter.snapshot
@@ -1,3 +1,4 @@
+- after() called
  describe hooks
    1 (*ms)
    2 (*ms)
@@ -293,15 +294,32 @@
       *
 
  - after() called
+ run after when before throws
+   1
+    'test did not finish before its parent and was cancelled'
+
+ run after when before throws (*ms)
+
+  Error: before
+      *
+      *
+      *
+      *
+      *
+      *
+      *
+      *
+      *
+
  before 1 called
  before 2 called
  after 1 called
  after 2 called
- tests 38
- suites 8
+ tests 39
+ suites 9
  pass 14
  fail 22
- cancelled 2
+ cancelled 3
  skipped 0
  todo 0
  duration_ms *
@@ -567,3 +585,20 @@
       *
       *
       *
+
+*
+ 1
+  'test did not finish before its parent and was cancelled'
+
+*
+ run after when before throws (*ms)
+  Error: before
+      *
+      *
+      *
+      *
+      *
+      *
+      *
+      *
+      *