fix: invoke end callback when no chunk is passed

src/stream/writable.ts

@@ -96,12 +96,15 @@ export class Writable extends EventEmitter implements NodeStream.Writable {
     const data = arg1 === callback ? undefined : arg1;
     if (data) {
       const encoding = arg2 === callback ? undefined : arg2;
-      this.write(data, encoding, callback);
+      this.write(data, encoding);
     }
     this.writableEnded = true;
     this.writableFinished = true;
     this.emit("close");
     this.emit("finish");
+    if (callback) {
+      callback();
+    }
     return this;
   }
 

test/index.test.ts

@@ -112,3 +112,44 @@ describe("fetchNodeRequestHandler", () => {
     });
   });
 });
+
+describe("ServerResponse#end callback", () => {
+  it("invokes callback with no chunk", async () => {
+    const calls: string[] = [];
+    const res = await fetchNodeRequestHandler((_req, res) => {
+      res.on("finish", () => calls.push("finish"));
+      res.end(() => calls.push("callback"));
+    }, "/test");
+    expect(await res.text()).toBe("");
+    expect(calls).toEqual(["finish", "callback"]);
+  });
+
+  it("invokes callback once with chunk", async () => {
+    const calls: string[] = [];
+    const res = await fetchNodeRequestHandler((_req, res) => {
+      res.on("finish", () => calls.push("finish"));
+      res.end("hello", () => calls.push("callback"));
+    }, "/test");
+    expect(await res.text()).toBe("hello");
+    expect(calls).toEqual(["finish", "callback"]);
+  });
+
+  it("invokes callback once with chunk and encoding", async () => {
+    let calls = 0;
+    const res = await fetchNodeRequestHandler((_req, res) => {
+      res.end("hello", "utf8", () => calls++);
+    }, "/test");
+    expect(await res.text()).toBe("hello");
+    expect(calls).toBe(1);
+  });
+
+  it("invokes callback on an already ended response", async () => {
+    let calls = 0;
+    const res = await fetchNodeRequestHandler((_req, res) => {
+      res.end("hello");
+      res.end(() => calls++);
+    }, "/test");
+    expect(await res.text()).toBe("hello");
+    expect(calls).toBe(1);
+  });
+});