add unit tests

Author: bdon

js/src/index.ts

@@ -345,7 +345,7 @@ export class FetchSource implements Source {
    * This should be used instead of maplibre's [transformRequest](https://maplibre.org/maplibre-gl-js/docs/API/classes/Map/#example) for PMTiles archives.
    */
   customHeaders: Headers;
-  credentials: string;
+  credentials: "same-origin" | "include" | undefined;
   /** @hidden */
   mustReload: boolean;
   /** @hidden */
@@ -354,7 +354,7 @@ export class FetchSource implements Source {
   constructor(
     url: string,
     customHeaders: Headers = new Headers(),
-    credentials = "same-origin"
+    credentials: "same-origin" | "include" | undefined = undefined
   ) {
     this.url = url;
     this.customHeaders = customHeaders;
@@ -408,7 +408,7 @@ export class FetchSource implements Source {
     // * it requires CORS configuration becasue If-Match is not a CORs-safelisted header
     // CORs configuration should expose ETag.
     // if any etag mismatch is detected, we need to ignore the browser cache
-    let cache: string | undefined;
+    let cache: "no-store" | "reload" | undefined;
     if (this.mustReload) {
       cache = "reload";
     } else if (this.chromeWindowsNoCache) {
@@ -420,8 +420,7 @@ export class FetchSource implements Source {
       cache: cache,
       headers: requestHeaders,
       credentials: this.credentials,
-      //biome-ignore lint: "cache" is incompatible between cloudflare workers and browser
-    } as any);
+    });
 
     // handle edge case where the archive is < 16384 kb total.
     if (offset === 0 && resp.status === 416) {
@@ -436,8 +435,7 @@ export class FetchSource implements Source {
         cache: "reload",
         headers: requestHeaders,
         credentials: this.credentials,
-        //biome-ignore lint: "cache" is incompatible between cloudflare workers and browser
-      } as any);
+      });
     }
 
     // if it's a weak etag, it's not useful for us, so ignore it.

js/test/adapter.test.ts

@@ -1,5 +1,5 @@
 import assert from "node:assert";
-import { describe, mock, test } from "node:test";
+import { describe, test } from "node:test";
 import { PMTiles, Protocol } from "../src";
 import { mockServer } from "./utils";
 

js/test/utils.ts

@@ -6,21 +6,27 @@ class MockServer {
   etag?: string;
   numRequests: number;
   lastCache?: string;
+  lastRequestHeaders: Headers | null;
+  lastCredentials?: string;
 
   reset() {
     this.numRequests = 0;
     this.etag = undefined;
+    this.lastRequestHeaders = null;
   }
 
   constructor() {
     this.numRequests = 0;
     this.etag = undefined;
+    this.lastRequestHeaders = null;
     const serverBuffer = fs.readFileSync("test/data/test_fixture_1.pmtiles");
     const server = setupServer(
       http.get(
         "http://localhost:1337/example.pmtiles",
         ({ request, params }) => {
           this.lastCache = request.cache;
+          this.lastRequestHeaders = request.headers;
+          this.lastCredentials = request.credentials;
           this.numRequests++;
           const range = request.headers.get("range")?.substr(6).split("-");
           if (!range) {
@@ -35,7 +41,31 @@ class MockServer {
             headers: { etag: this.etag } as HeadersInit,
           });
         }
-      )
+      ),
+      http.get("http://localhost:1337/small.pmtiles", ({ request }) => {
+        this.lastRequestHeaders = request.headers;
+        this.lastCredentials = request.credentials;
+        this.numRequests++;
+        const range = request.headers.get("range")?.substr(6).split("-");
+        if (!range) {
+          throw new Error("invalid range");
+        }
+        const rangeEnd = +range[1];
+        if (rangeEnd >= serverBuffer.length) {
+          return new HttpResponse(null, {
+            status: 416,
+            headers: {
+              "Content-Range": `bytes */${serverBuffer.length}`,
+            },
+          });
+        }
+        const rangeStart = +range[0];
+        const body = serverBuffer.slice(rangeStart, rangeEnd + 1);
+        return new HttpResponse(body, {
+          status: 206,
+          headers: { etag: this.etag } as HeadersInit,
+        });
+      })
     );
     server.listen({ onUnhandledRequest: "error" });
   }

js/test/v3.test.ts

@@ -6,6 +6,7 @@ import { mockServer } from "./utils";
 import {
   BufferPosition,
   Entry,
+  FetchSource,
   PMTiles,
   RangeResponse,
   SharedPromiseCache,
@@ -407,3 +408,36 @@ describe("pmtiles v3", () => {
     });
   });
 });
+
+describe("FetchSource", () => {
+  test("customHeaders are sent with requests", async () => {
+    mockServer.reset();
+    const source = new FetchSource(
+      "http://localhost:1337/example.pmtiles",
+      new Headers({ "X-Custom-Header": "test-value" }),
+      "include"
+    );
+    await source.getBytes(0, 100);
+    assert.strictEqual(
+      mockServer.lastRequestHeaders?.get("x-custom-header"),
+      "test-value"
+    );
+    assert.strictEqual(mockServer.lastCredentials, "include");
+  });
+
+  test("customHeaders are preserved on 416 retry", async () => {
+    mockServer.reset();
+    const source = new FetchSource(
+      "http://localhost:1337/small.pmtiles",
+      new Headers({ "X-Custom-Header": "retry-value" }),
+      "include"
+    );
+    await source.getBytes(0, 16384);
+    assert.strictEqual(mockServer.numRequests, 2);
+    assert.strictEqual(
+      mockServer.lastRequestHeaders?.get("x-custom-header"),
+      "retry-value",
+      "include"
+    );
+  });
+});