Reorganize tests (#2332)

1 files changed, 181 insertions, 0 deletions

diff --git a/test/js/web/fetch/fetch-gzip.test.ts b/test/js/web/fetch/fetch-gzip.test.ts
new file mode 100644
index 000000000..01eedc54a
--- /dev/null
+++ b/test/js/web/fetch/fetch-gzip.test.ts
@@ -0,0 +1,181 @@
+import { concatArrayBuffers } from "bun";
+import { it, describe, expect } from "bun:test";
+import fs from "fs";
+import { gc, gcTick } from "harness";
+
+it("fetch() with a buffered gzip response works (one chunk)", async () => {
+  var server = Bun.serve({
+    port: 6025,
+
+    async fetch(req) {
+      gcTick(true);
+      return new Response(require("fs").readFileSync(import.meta.dir + "/fixture.html.gz"), {
+        headers: {
+          "Content-Encoding": "gzip",
+          "Content-Type": "text/html; charset=utf-8",
+        },
+      });
+    },
+  });
+  gcTick(true);
+
+  const res = await fetch(`http://${server.hostname}:${server.port}`, { verbose: true });
+  gcTick(true);
+  const arrayBuffer = await res.arrayBuffer();
+  const clone = new Buffer(arrayBuffer);
+  gcTick(true);
+  await (async function () {
+    const second = new Buffer(await Bun.file(import.meta.dir + "/fixture.html").arrayBuffer());
+    gcTick(true);
+    expect(second.equals(clone)).toBe(true);
+  })();
+  gcTick(true);
+  server.stop();
+});
+
+it("fetch() with a redirect that returns a buffered gzip response works (one chunk)", async () => {
+  var server = Bun.serve({
+    port: 6020,
+
+    async fetch(req) {
+      if (req.url.endsWith("/redirect"))
+        return new Response(await Bun.file(import.meta.dir + "/fixture.html.gz").arrayBuffer(), {
+          headers: {
+            "Content-Encoding": "gzip",
+            "Content-Type": "text/html; charset=utf-8",
+          },
+        });
+
+      return Response.redirect("/redirect");
+    },
+  });
+
+  const res = await fetch(`http://${server.hostname}:${server.port}/hey`, { 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 protocol-relative redirect that returns a buffered gzip response works (one chunk)", async () => {
+  const server = Bun.serve({
+    port: 5018,
+
+    async fetch(req, server) {
+      if (req.url.endsWith("/redirect"))
+        return new Response(await Bun.file(import.meta.dir + "/fixture.html.gz").arrayBuffer(), {
+          headers: {
+            "Content-Encoding": "gzip",
+            "Content-Type": "text/html; charset=utf-8",
+          },
+        });
+
+      return Response.redirect(`://${server.hostname}:${server.port}/redirect`);
+    },
+  });
+
+  const res = await fetch(`http://${server.hostname}:${server.port}/hey`, { verbose: true });
+  expect(res.url).toBe(`http://${server.hostname}:${server.port}/redirect`);
+  expect(res.redirected).toBe(true);
+  expect(res.status).toBe(200);
+  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, streamed, with a delay", async () => {
+  var server = Bun.serve({
+    port: 6081,
+
+    fetch(req) {
+      return new Response(
+        new ReadableStream({
+          type: "direct",
+          async pull(controller) {
+            await 2;
+
+            const buffer = await Bun.file(import.meta.dir + "/fixture.html.gz").arrayBuffer();
+            controller.write(buffer);
+            controller.close();
+          },
+        }),
+        {
+          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);
+  server.stop();
+});
+
+it("fetch() with a gzip response works (multiple chunks, TCP server", async done => {
+  const compressed = await Bun.file(import.meta.dir + "/fixture.html.gz").arrayBuffer();
+  var socketToClose;
+  const server = Bun.listen({
+    port: 4024,
+    hostname: "0.0.0.0",
+    socket: {
+      async open(socket) {
+        socketToClose = socket;
+
+        var corked: any[] = [];
+        var cork = true;
+        async function write(chunk) {
+          await new Promise<void>((resolve, reject) => {
+            if (cork) {
+              corked.push(chunk);
+            }
+
+            if (!cork && corked.length) {
+              socket.write(corked.join(""));
+              corked.length = 0;
+            }
+
+            if (!cork) {
+              socket.write(chunk);
+            }
+
+            resolve();
+          });
+        }
+        await write("HTTP/1.1 200 OK\r\n");
+        await write("Content-Encoding: gzip\r\n");
+        await write("Content-Type: text/html; charset=utf-8\r\n");
+        await write("Content-Length: " + compressed.byteLength + "\r\n");
+        await write("X-WTF: " + "lol".repeat(1000) + "\r\n");
+        await write("\r\n");
+        for (var i = 100; i < compressed.byteLength; i += 100) {
+          cork = false;
+          await write(compressed.slice(i - 100, i));
+        }
+        await write(compressed.slice(i - 100));
+        socket.flush();
+      },
+      drain(socket) {},
+    },
+  });
+  await 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);
+  socketToClose.end();
+  server.stop();
+  done();
+});