1 files changed, 218 insertions, 0 deletions
diff --git a/test/bun.js/response.file.test.js b/test/bun.js/response.file.test.js
new file mode 100644
index 000000000..2d0b6506e
--- /dev/null
+++ b/test/bun.js/response.file.test.js
@@ -0,0 +1,218 @@
+import fs from "fs";
+import { it, expect } from "bun:test";
+import path from "path";
+import { gcTick } from "./gc";
+
+it("Bun.file not found returns ENOENT", async () => {
+  try {
+    await gcTick();
+    await Bun.file("/does/not/exist.txt").text();
+    await gcTick();
+  } catch (exception) {
+    expect(exception.code).toBe("ENOENT");
+  }
+  await gcTick();
+});
+
+it("Bun.write('out.txt', 'string')", async () => {
+  for (let erase of [true, false]) {
+    if (erase) {
+      try {
+        fs.unlinkSync(path.join("/tmp", "out.txt"));
+      } catch (e) {}
+    }
+    await gcTick();
+    expect(await Bun.write("/tmp/out.txt", "string")).toBe("string".length);
+    await gcTick();
+    const out = Bun.file("/tmp/out.txt");
+    await gcTick();
+    expect(await out.text()).toBe("string");
+    await gcTick();
+    expect(await out.text()).toBe(fs.readFileSync("/tmp/out.txt", "utf8"));
+    await gcTick();
+  }
+});
+
+it("Bun.write blob", async () => {
+  await Bun.write(
+    Bun.file("/tmp/response-file.test.txt"),
+    Bun.file(path.join(import.meta.dir, "fetch.js.txt"))
+  );
+  await gcTick();
+  await Bun.write(Bun.file("/tmp/response-file.test.txt"), "blah blah blha");
+  await gcTick();
+  await Bun.write(
+    Bun.file("/tmp/response-file.test.txt"),
+    new Uint32Array(1024)
+  );
+  await gcTick();
+  await Bun.write("/tmp/response-file.test.txt", new Uint32Array(1024));
+  await gcTick();
+  expect(
+    await Bun.write(
+      new TextEncoder().encode("/tmp/response-file.test.txt"),
+      new Uint32Array(1024)
+    )
+  ).toBe(new Uint32Array(1024).byteLength);
+  await gcTick();
+});
+
+it("Bun.file -> Bun.file", async () => {
+  try {
+    fs.unlinkSync(path.join("/tmp", "fetch.js.in"));
+  } catch (e) {}
+  await gcTick();
+  try {
+    fs.unlinkSync(path.join("/tmp", "fetch.js.out"));
+  } catch (e) {}
+  await gcTick();
+  const file = path.join(import.meta.dir, "fetch.js.txt");
+  await gcTick();
+  const text = fs.readFileSync(file, "utf8");
+  fs.writeFileSync("/tmp/fetch.js.in", text);
+  await gcTick();
+  {
+    const result = await Bun.write(
+      Bun.file("/tmp/fetch.js.out"),
+      Bun.file("/tmp/fetch.js.in")
+    );
+    await gcTick();
+    expect(await Bun.file("/tmp/fetch.js.out").text()).toBe(text);
+    await gcTick();
+  }
+
+  {
+    await Bun.write(
+      Bun.file("/tmp/fetch.js.in").slice(0, (text.length / 2) | 0),
+      Bun.file("/tmp/fetch.js.out")
+    );
+    expect(await Bun.file("/tmp/fetch.js.in").text()).toBe(
+      text.substring(0, (text.length / 2) | 0)
+    );
+  }
+
+  {
+    await gcTick();
+    await Bun.write("/tmp/fetch.js.in", Bun.file("/tmp/fetch.js.out"));
+    await gcTick();
+    expect(await Bun.file("/tmp/fetch.js.in").text()).toBe(text);
+  }
+});
+
+it("Bun.file", async () => {
+  const file = path.join(import.meta.dir, "fetch.js.txt");
+  await gcTick();
+  expect(await Bun.file(file).text()).toBe(fs.readFileSync(file, "utf8"));
+  await gcTick();
+});
+
+it("Bun.file as a Blob", async () => {
+  const filePath = path.join(import.meta.path, "../fetch.js.txt");
+  const fixture = fs.readFileSync(filePath, "utf8");
+  // this is a Blob object with the same interface as the one returned by fetch
+  // internally, instead of a byte array, it stores the file path!
+  // this enables several performance optimizations
+  var blob = Bun.file(filePath);
+  await gcTick();
+
+  // no size because we haven't read it from disk yet
+  expect(blob.size).toBe(0);
+  await gcTick();
+  // now it reads "./fetch.js.txt" from the filesystem
+  // it's lazy, only loads once we ask for it
+  // if it fails, the promise will reject at this point
+  expect(await blob.text()).toBe(fixture);
+  await gcTick();
+  // now that it's loaded, the size updates
+  expect(blob.size).toBe(fixture.length);
+  await gcTick();
+  // and it only loads once for _all_ blobs pointing to that file path
+  // until all references are released
+  expect((await blob.arrayBuffer()).byteLength).toBe(fixture.length);
+  await gcTick();
+
+  const array = new Uint8Array(await blob.arrayBuffer());
+  await gcTick();
+  const text = fixture;
+  for (let i = 0; i < text.length; i++) {
+    expect(array[i]).toBe(text.charCodeAt(i));
+  }
+  await gcTick();
+  expect(blob.size).toBe(fixture.length);
+  blob = null;
+  await gcTick();
+  await new Promise((resolve) => setTimeout(resolve, 1));
+  // now we're back
+  var blob = Bun.file(filePath);
+  expect(blob.size).toBe(0);
+});
+
+it("Response -> Bun.file", async () => {
+  const file = path.join(import.meta.dir, "fetch.js.txt");
+  await gcTick();
+  const text = fs.readFileSync(file, "utf8");
+  await gcTick();
+  const response = new Response(Bun.file(file));
+  await gcTick();
+  expect(await response.text()).toBe(text);
+  await gcTick();
+});
+
+it("Bun.file -> Response", async () => {
+  // ensure the file doesn't already exist
+  try {
+    fs.unlinkSync("/tmp/fetch.js.out");
+  } catch {}
+  await gcTick();
+  const file = path.join(import.meta.dir, "fetch.js.txt");
+  await gcTick();
+  const text = fs.readFileSync(file, "utf8");
+  await gcTick();
+  const resp = await fetch("https://example.com");
+  await gcTick();
+
+  expect(await Bun.write("/tmp/fetch.js.out", resp)).toBe(text.length);
+  await gcTick();
+  expect(await Bun.file("/tmp/fetch.js.out").text()).toBe(text);
+  await gcTick();
+});
+
+it("Response -> Bun.file -> Response -> text", async () => {
+  await gcTick();
+  const file = path.join(import.meta.dir, "fetch.js.txt");
+  await gcTick();
+  const text = fs.readFileSync(file, "utf8");
+  await gcTick();
+  const response = new Response(Bun.file(file));
+  await gcTick();
+  const response2 = response.clone();
+  await gcTick();
+  expect(await response2.text()).toBe(text);
+  await gcTick();
+});
+
+// If you write nothing to a file, it shouldn't modify it
+// If you want to truncate a file, it should be more explicit
+it("Bun.write('output.html', '')", async () => {
+  await Bun.write("/tmp/output.html", "lalalala");
+  expect(await Bun.write("/tmp/output.html", "")).toBe(0);
+  expect(await Bun.file("/tmp/output.html").text()).toBe("lalalala");
+});
+
+// Since Bun.file is resolved lazily, this needs to specifically be checked
+it("Bun.write('output.html', HTMLRewriter.transform(Bun.file)))", async () => {
+  var rewriter = new HTMLRewriter();
+  rewriter.on("div", {
+    element(element) {
+      element.setInnerContent("<blink>it worked!</blink>", { html: true });
+    },
+  });
+  await Bun.write("/tmp/html-rewriter.txt.js", "<div>hello</div>");
+  var input = new Response(Bun.file("/tmp/html-rewriter.txt.js"));
+  var output = rewriter.transform(input);
+  const outpath = `/tmp/html-rewriter.${Date.now()}.html`;
+  await Bun.write(outpath, output);
+  expect(await Bun.file(outpath).text()).toBe(
+    "<div><blink>it worked!</blink></div>"
+  );
+});