From cfb52f1b7c41fd72e2848ea6e832b1d498f3b587 Mon Sep 17 00:00:00 2001
From: Jake Wang <jake.wang@argonlabs.tech>
Date: Sat, 19 Sep 2026 01:25:33 -0400
Subject: [PATCH] fix(stream-transform): handle promise rejections without
 leaking

---
 packages/stream-transform/lib/index.js        | 14 +++---
 .../test/handler.promise.error.js             | 47 +++++++++++++++++++
 2 files changed, 55 insertions(+), 6 deletions(-)
 create mode 100644 packages/stream-transform/test/handler.promise.error.js

diff --git a/packages/stream-transform/lib/index.js b/packages/stream-transform/lib/index.js
index e2fe8257..07e31c12 100644
--- a/packages/stream-transform/lib/index.js
+++ b/packages/stream-transform/lib/index.js
@@ -50,12 +50,14 @@ Transformer.prototype._transform = function (chunk, _, cb) {
       // sync
       const result = this.handler.call(this, chunk, this.options.params);
       if (result && result.then) {
-        result.then((result) => {
-          this.__done(null, [result], cb);
-        });
-        result.catch((err) => {
-          this.__done(err);
-        });
+        result.then(
+          (result) => {
+            this.__done(null, [result], cb);
+          },
+          (err) => {
+            this.__done(err);
+          },
+        );
       } else {
         this.__done(null, [result], cb);
       }
diff --git a/packages/stream-transform/test/handler.promise.error.js b/packages/stream-transform/test/handler.promise.error.js
new file mode 100644
index 00000000..6a96d252
--- /dev/null
+++ b/packages/stream-transform/test/handler.promise.error.js
@@ -0,0 +1,47 @@
+import assert from "node:assert/strict";
+import { spawnSync } from "node:child_process";
+
+describe("handler.promise.error", function () {
+  for (const api of ["callback", "pipeline"]) {
+    it(`handles rejected promises with the ${api} API`, function () {
+      const result = spawnSync(
+        process.execPath,
+        [
+          "--unhandled-rejections=strict",
+          "--input-type=module",
+          "--eval",
+          `
+            import assert from "node:assert/strict";
+            import { Readable, Writable } from "node:stream";
+            import { pipeline } from "node:stream/promises";
+            import { transform } from ${JSON.stringify(new URL("../lib/index.js", import.meta.url).href)};
+
+            const error = new Error("Catchme");
+            const handler = async (record) => {
+              throw error;
+            };
+            if (${JSON.stringify(api)} === "callback") {
+              await new Promise((resolve) => {
+                transform([["value"]], handler, (err) => {
+                  assert.equal(err, error);
+                  resolve();
+                });
+              });
+            } else {
+              await assert.rejects(
+                pipeline(
+                  Readable.from([["value"]]),
+                  transform(handler),
+                  new Writable({ objectMode: true, write: (_, __, next) => next() }),
+                ),
+                (err) => err === error,
+              );
+            }
+          `,
+        ],
+        { encoding: "utf8" },
+      );
+      assert.equal(result.status, 0, result.stderr);
+    });
+  }
+});