Date Time64 P9 Test

1import { describe, expect, it } from "vitest";
2import { query } from "./clickhouse.js";
3import { NeedMoreData, Cursor } from "../src/readers/core.js";
4import { readDateTime64P9 } from "../src/readers/datetime.js";
5
6async function reader(expr: string): Promise<Cursor> {
7  return new Cursor(await query(`SELECT ${expr} FORMAT RowBinary`));
8}
9
10describe("readDateTime64P9", () => {
11  it("decodes nanoseconds to [Date (whole seconds), nanoseconds]", async () => {
12    const r = await reader(
13      "toDateTime64('2021-01-01 00:00:00.123456789', 9, 'UTC')",
14    );
15    const [date, nanos] = readDateTime64P9(r);
16    expect(date.toISOString()).toBe("2021-01-01T00:00:00.000Z");
17    expect(nanos).toBe(123456789);
18    expect(r.pos).toBe(8);
19  });
20
21  describe("advance() edge cases", () => {
22    it("throws NeedMoreData for every incomplete prefix (0 .. full.length-1)", async () => {
23      const full = await query(
24        "SELECT toDateTime64('2021-01-01 00:00:00.123456789', 9, 'UTC') FORMAT RowBinary",
25      );
26      for (let len = 0; len < full.length; len++) {
27        const r = new Cursor(full.subarray(0, len));
28        let thrown: unknown;
29        try {
30          readDateTime64P9(r);
31        } catch (e) {
32          thrown = e;
33        }
34        expect(thrown, `prefix length ${len} of ${full.length}`).toBe(
35          NeedMoreData,
36        );
37      }
38    });
39  });
40});