Interval Test

1import { describe, expect, it } from "vitest";
2import { query } from "./clickhouse.js";
3import { NeedMoreData, Cursor } from "../src/readers/core.js";
4import { readInterval } from "../src/readers/interval.js";
5
6async function reader(expr: string): Promise<Cursor> {
7  return new Cursor(await query(`SELECT ${expr} FORMAT RowBinary`));
8}
9
10// All 11 Interval* types share one Int64 wire; the unit lives in the type name.
11describe("readInterval", () => {
12  it("decodes a positive count (IntervalSecond)", async () => {
13    const r = await reader("toIntervalSecond(5)");
14    expect(readInterval(r)).toBe(5n);
15    expect(r.pos).toBe(8);
16  });
17
18  it("decodes a negative count (IntervalDay)", async () => {
19    expect(readInterval(await reader("toIntervalDay(-3)"))).toBe(-3n);
20  });
21
22  it("decodes a large count (IntervalNanosecond)", async () => {
23    expect(readInterval(await reader("toIntervalNanosecond(1000000000)"))).toBe(
24      1000000000n,
25    );
26  });
27
28  describe("advance() edge cases", () => {
29    it("throws NeedMoreData for every incomplete prefix (0 .. full.length-1)", async () => {
30      const full = await query("SELECT toIntervalSecond(5) FORMAT RowBinary");
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          readInterval(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});