Fix memory leak in gzip pool + add test for gzip'd data

1 files changed, 116 insertions, 0 deletions

diff --git a/test/bun.js/fetch-gzip.test.ts b/test/bun.js/fetch-gzip.test.ts
new file mode 100644
index 000000000..a75e01701
--- /dev/null
+++ b/test/bun.js/fetch-gzip.test.ts
@@ -0,0 +1,116 @@
+import { concatArrayBuffers } from "bun";
+import { it, describe, expect } from "bun:test";
+import fs from "fs";
+import { gc } from "./gc";
+
+it("fetch() with a buffered gzip response works (one chunk)", async () => {
+  var server = Bun.serve({
+    port: 6025,
+
+    async fetch(req) {
+      return new Response(
+        await Bun.file(import.meta.dir + "/fixture.html.gz").arrayBuffer(),
+        {
+          headers: {
+            "Content-Encoding": "gzip",
+            "Content-Type": "text/html; charset=utf-8",
+          },
+        },
+      );
+    },
+  });
+
+  const res = await fetch(
+    `http://${server.hostname}:${server.port}`,
+    {},
+    { verbose: true },
+  );
+  const arrayBuffer = await res.arrayBuffer();
+  expect(
+    new Buffer(arrayBuffer).equals(
+      new Buffer(
+        await Bun.file(import.meta.dir + "/fixture.html").arrayBuffer(),
+      ),
+    ),
+  ).toBe(true);
+  server.stop();
+});
+
+it("fetch() with a gzip response works (one chunk)", async () => {
+  var server = Bun.serve({
+    port: 6023,
+
+    fetch(req) {
+      return new Response(Bun.file(import.meta.dir + "/fixture.html.gz"), {
+        headers: {
+          "Content-Encoding": "gzip",
+          "Content-Type": "text/html; charset=utf-8",
+        },
+      });
+    },
+  });
+
+  const res = await fetch(`http://${server.hostname}:${server.port}`);
+  const arrayBuffer = await res.arrayBuffer();
+  expect(
+    new Buffer(arrayBuffer).equals(
+      new Buffer(
+        await Bun.file(import.meta.dir + "/fixture.html").arrayBuffer(),
+      ),
+    ),
+  ).toBe(true);
+  server.stop();
+});
+
+it("fetch() with a gzip response works (multiple chunks)", async () => {
+  var server = Bun.serve({
+    port: 6024,
+
+    fetch(req) {
+      return new Response(
+        new ReadableStream({
+          type: "direct",
+          async pull(controller) {
+            var chunks: ArrayBuffer[] = [];
+            const buffer = await Bun.file(
+              import.meta.dir + "/fixture.html.gz",
+            ).arrayBuffer();
+            var remaining = buffer;
+            for (var i = 100; i < buffer.byteLength; i += 100) {
+              var chunk = remaining.slice(0, i);
+              remaining = remaining.slice(i);
+              controller.write(chunk);
+              chunks.push(chunk);
+              await controller.flush();
+            }
+
+            await controller.flush();
+            // sanity check
+            expect(
+              new Buffer(concatArrayBuffers(chunks)).equals(new Buffer(buffer)),
+            ).toBe(true);
+
+            controller.end();
+          },
+        }),
+        {
+          headers: {
+            "Content-Encoding": "gzip",
+            "Content-Type": "text/html; charset=utf-8",
+            "Content-Length": "1",
+          },
+        },
+      );
+    },
+  });
+
+  const res = await fetch(`http://${server.hostname}:${server.port}`, {});
+  const arrayBuffer = await res.arrayBuffer();
+  expect(
+    new Buffer(arrayBuffer).equals(
+      new Buffer(
+        await Bun.file(import.meta.dir + "/fixture.html").arrayBuffer(),
+      ),
+    ),
+  ).toBe(true);
+});