Orders Example Test

1import { describe, expect, it } from "vitest";
2import { query } from "./clickhouse.js";
3import { Cursor } from "../src/readers/core.js";
4import { type OrderRow, readOrderRow } from "../src/examples/orders.js";
5import { readRows } from "../src/readers/rows.js";
6
7/**
8 * Runs the `orders` example end to end (UUID / Decimal / Enum). These types are
9 * awkward or lossy as JSON — UUID and Enum as names, Decimal as a float that
10 * can't represent every value exactly — so the rows go in as raw SQL `VALUES`
11 * instead of `JSONEachRow`.
12 */
13describe("example: orders (UUID / Decimal / Enum via raw VALUES)", () => {
14  it("creates, populates, and reads back through readOrderRow", async () => {
15    const t = "rb_example_orders";
16    await query(`DROP TABLE IF EXISTS ${t}`);
17    await query(
18      `CREATE TABLE ${t} (` +
19        `id UInt8, ` +
20        `uid UUID, ` +
21        `price Decimal64(2), ` +
22        `status Enum8('new' = 1, 'shipped' = 2, 'done' = 3)` +
23        `) ENGINE = Memory`,
24    );
25    try {
26      await query(
27        `INSERT INTO ${t} VALUES ` +
28          `(1, '61f0c404-5cb3-11e7-907b-a6006ad3dba0', 12.34, 'new'), ` +
29          `(2, '00000000-0000-0000-0000-000000000000', 0.00, 'shipped'), ` +
30          `(3, 'ffffffff-ffff-ffff-ffff-ffffffffffff', -9.99, 'done')`,
31      );
32
33      const r = new Cursor(
34        await query(
35          `SELECT id, uid, price, status FROM ${t} ORDER BY id FORMAT RowBinary`,
36        ),
37      );
38      const out: OrderRow[] = readRows(readOrderRow)(r);
39      expect(out).toEqual([
40        {
41          id: 1,
42          uid: "61f0c404-5cb3-11e7-907b-a6006ad3dba0",
43          price: [1234n, 2],
44          status: 1,
45        },
46        {
47          id: 2,
48          uid: "00000000-0000-0000-0000-000000000000",
49          price: [0n, 2],
50          status: 2,
51        },
52        {
53          id: 3,
54          uid: "ffffffff-ffff-ffff-ffff-ffffffffffff",
55          price: [-999n, 2],
56          status: 3,
57        },
58      ]);
59      expect(r.pos).toBe(r.buf.length);
60    } finally {
61      await query(`DROP TABLE IF EXISTS ${t}`);
62    }
63  });
64});