Multi Polygon Test

1import { describe, expect, it } from "vitest";
2import { query } from "./clickhouse.js";
3import { NeedMoreData, Cursor } from "../src/readers/core.js";
4import { readMultiPolygon } from "../src/readers/geo.js";
5
6async function reader(expr: string): Promise<Cursor> {
7  return new Cursor(await query(`SELECT ${expr} FORMAT RowBinary`));
8}
9
10describe("readMultiPolygon", () => {
11  it("decodes a MultiPolygon as an array of polygons", async () => {
12    const r = await reader(
13      "CAST([[[(0, 0), (1, 0), (1, 1)]]] AS MultiPolygon)",
14    );
15    expect(readMultiPolygon(r)).toEqual([
16      [
17        [
18          [0, 0],
19          [1, 0],
20          [1, 1],
21        ],
22      ],
23    ]);
24  });
25
26  describe("advance() edge cases", () => {
27    it("throws NeedMoreData for every incomplete prefix (0 .. full.length-1)", async () => {
28      const full = await query(
29        "SELECT CAST([[[(0, 0), (1, 0), (1, 1)]]] AS MultiPolygon) FORMAT RowBinary",
30      );
31      for (let len = 0; len < full.length; len++) {
32        const r = new Cursor(full.subarray(0, len));
33        let thrown: unknown;
34        try {
35          readMultiPolygon(r);
36        } catch (e) {
37          thrown = e;
38        }
39        expect(thrown, `prefix length ${len} of ${full.length}`).toBe(
40          NeedMoreData,
41        );
42      }
43    });
44  });
45});