PMTiles/js/test/v2.test.ts

import { test } from "node:test";
import assert from "node:assert";

import {
  unshift,
  getUint24,
  getUint48,
  deriveLeaf,
  queryLeafdir,
  queryTile,
  parseEntry,
  EntryV2,
  createDirectory,
} from "../v2";

test("stub data", () => {
  let dataview = new DataView(createDirectory([
    { z: 5, x: 1000, y: 2000, offset: 1000, length: 2000, is_dir: false },
    {
      z: 14,
      x: 16383,
      y: 16383,
      offset: 999999,
      length: 999,
      is_dir: false,
    },
  ]));
  var z_raw = dataview.getUint8(17 + 0);
  var x = getUint24(dataview, 17 + 1);
  var y = getUint24(dataview, 17 + 4);
  var offset = getUint48(dataview, 17 + 7);
  var length = dataview.getUint32(17 + 13, true);
  assert.strictEqual(z_raw,14);
  assert.strictEqual(x,16383);
  assert.strictEqual(y,16383);
});

test("get entry", () => {
  let view = new DataView(createDirectory([
    { z: 5, x: 1000, y: 2000, offset: 1000, length: 2000, is_dir: false },
    {
      z: 14,
      x: 16383,
      y: 16383,
      offset: 999999,
      length: 999,
      is_dir: false,
    },
  ]));
  let entry = queryTile(view, 14, 16383, 16383);
  assert.strictEqual(entry!.z,14);
  assert.strictEqual(entry!.x,16383);
  assert.strictEqual(entry!.y,16383);
  assert.strictEqual(entry!.offset,999999);
  assert.strictEqual(entry!.length,999);
  assert.strictEqual(entry!.is_dir,false);
  assert.strictEqual(queryLeafdir(view, 14, 16383, 16383),null);
});

test("get leafdir", () => {
  let view = new DataView(createDirectory([
    {
      z: 14,
      x: 16383,
      y: 16383,
      offset: 999999,
      length: 999,
      is_dir: true,
    },
  ]));
  let entry = queryLeafdir(view, 14, 16383, 16383);
  assert.strictEqual(entry!.z,14);
  assert.strictEqual(entry!.x,16383);
  assert.strictEqual(entry!.y,16383);
  assert.strictEqual(entry!.offset,999999);
  assert.strictEqual(entry!.length,999);
  assert.strictEqual(entry!.is_dir,true);
  assert.strictEqual(queryTile(view, 14, 16383, 16383),null);
});

test("derive the leaf level", () => {
  let view = new DataView(createDirectory([
    {
      z: 6,
      x: 3,
      y: 3,
      offset: 0,
      length: 0,
      is_dir: true,
    },
  ]));
  let leaf = deriveLeaf(view,{z:7,x:6,y:6});
  assert.strictEqual(leaf!.z,6);
  assert.strictEqual(leaf!.x,3);
  assert.strictEqual(leaf!.y,3);
  view = new DataView(createDirectory([
    {
      z: 6,
      x: 3,
      y: 3,
      offset: 0,
      length: 0,
      is_dir: false,
    },
  ]));
  leaf = deriveLeaf(view,{z:7,x:6,y:6});
  assert.strictEqual(leaf,null);
});

test("convert spec v1 directory to spec v2 directory", () => {
  let view = new DataView(createDirectory([
    {
      z: 7,
      x: 3,
      y: 3,
      offset: 3,
      length: 3,
      is_dir: true,
    },
    {
      z: 6,
      x: 2,
      y: 2,
      offset: 2,
      length: 2,
      is_dir: false,
    },
    {
      z: 6,
      x: 2,
      y: 1,
      offset: 1,
      length: 1,
      is_dir: false,
    },
  ]));
  let entry = queryLeafdir(view, 7, 3, 3);
  assert.strictEqual(entry!.offset,3);
  entry = queryTile(view, 6, 2, 2);
  assert.strictEqual(entry!.offset,2);
  entry = queryTile(view, 6, 2, 1);
  assert.strictEqual(entry!.offset,1);

  entry = parseEntry(view, 0);
  assert.strictEqual(entry!.offset,1);
  entry = parseEntry(view, 1);
  assert.strictEqual(entry!.offset,2);
  entry = parseEntry(view, 2);
  assert.strictEqual(entry!.offset,3);
});