diagnostics_channel: return original thenable

lib/diagnostics_channel.js

@@ -10,6 +10,7 @@ const {
   ObjectDefineProperty,
   ObjectGetPrototypeOf,
   ObjectSetPrototypeOf,
+  Promise,
   ReflectApply,
   SafeFinalizationRegistry,
   SafeMap,
@@ -375,6 +376,10 @@ class TracingChannel {
       asyncStart.publish(context);
       // TODO: Is there a way to have asyncEnd _after_ the continuation?
       asyncEnd.publish(context);
+    }
+
+    function rejectAndRethrow(err) {
+      reject(err);
       throw err;
     }
 
@@ -396,7 +401,12 @@ class TracingChannel {
           context.result = result;
           return result;
         }
-        return result.then(resolve, reject);
+        // TODO(qard): Not realm safe? Should be a construct.name check?
+        if (result.constructor === Promise) {
+          return result.then(resolve, rejectAndRethrow);
+        }
+        result.then(resolve, reject);
+        return result;
       } catch (err) {
         context.error = err;
         error.publish(context);

test/parallel/test-diagnostics-channel-tracing-channel-promise-thenable.js

@@ -12,6 +12,9 @@ class ResolvedThenable {
   then(resolve) {
     return new ResolvedThenable(resolve(this.#result));
   }
+  customMethod() {
+    return this.#result;
+  }
 }
 
 const channel = dc.tracingChannel('test');
@@ -49,7 +52,10 @@ const result = channel.tracePromise(common.mustCall(function(value) {
 }), input, thisArg, expectedResult);
 
 assert(result instanceof ResolvedThenable);
-assert.notStrictEqual(result, innerThenable);
+// With branching then, the original thenable is returned directly so that
+// extra methods defined on it remain accessible to the caller.
+assert.strictEqual(result, innerThenable);
+assert.deepStrictEqual(result.customMethod(), expectedResult);
 result.then(common.mustCall((value) => {
   assert.deepStrictEqual(value, expectedResult);
 }));