Profiles Example Test

1import { describe, expect, it } from "vitest";
2import { query } from "./clickhouse.js";
3import { Cursor } from "../src/readers/core.js";
4import { type ProfileRow, readProfileRow } from "../src/examples/profiles.js";
5import { readRows } from "../src/readers/rows.js";
6
7/**
8 * Runs the `profiles` example end to end (Array + Nullable). Populated via
9 * `JSONEachRow`; the empty array and the NULL score are the single-byte edge
10 * cases the reader has to get right.
11 */
12describe("example: profiles (Array + Nullable via JSONEachRow)", () => {
13  it("creates, populates, and reads back through readProfileRow", async () => {
14    const t = "rb_example_profiles";
15    await query(`DROP TABLE IF EXISTS ${t}`);
16    await query(
17      `CREATE TABLE ${t} (id UInt32, tags Array(String), score Nullable(Int32)) ENGINE = Memory`,
18    );
19    try {
20      const rows = [
21        { id: 1, tags: ["a", "b"], score: 10 },
22        { id: 2, tags: [], score: null },
23        { id: 3, tags: ["solo"], score: -5 },
24      ];
25      await query(
26        `INSERT INTO ${t} FORMAT JSONEachRow\n` +
27          rows.map((r) => JSON.stringify(r)).join("\n"),
28      );
29
30      const r = new Cursor(
31        await query(
32          `SELECT id, tags, score FROM ${t} ORDER BY id FORMAT RowBinary`,
33        ),
34      );
35      const out: ProfileRow[] = readRows(readProfileRow)(r);
36      expect(out).toEqual([
37        { id: 1, tags: ["a", "b"], score: 10 },
38        { id: 2, tags: [], score: null },
39        { id: 3, tags: ["solo"], score: -5 },
40      ]);
41      expect(r.pos).toBe(r.buf.length);
42    } finally {
43      await query(`DROP TABLE IF EXISTS ${t}`);
44    }
45  });
46});