Time Interval Write Test

1import { describe, expect, it } from "vitest";
2import { query } from "./clickhouse.js";
3import { encode } from "./encode.js";
4import {
5  writeTime,
6  writeTime64,
7  parseTime,
8  parseTime64,
9} from "../src/writers/time.js";
10import { writeInterval } from "../src/writers/interval.js";
11
12describe("writeTime", () => {
13  it("encodes 12:34:56", async () =>
14    expect(encode(writeTime, parseTime("12:34:56"))).toEqual(
15      await query(
16        "SELECT CAST('12:34:56' AS Time) SETTINGS enable_time_time64_type = 1 FORMAT RowBinary",
17      ),
18    ));
19  it("encodes a negative -01:02:03", async () =>
20    expect(encode(writeTime, parseTime("-01:02:03"))).toEqual(
21      await query(
22        "SELECT CAST('-01:02:03' AS Time) SETTINGS enable_time_time64_type = 1 FORMAT RowBinary",
23      ),
24    ));
25});
26
27describe("writeTime64", () => {
28  it("encodes Time64(3)", async () =>
29    expect(encode(writeTime64, parseTime64("12:34:56.789", 3))).toEqual(
30      await query(
31        "SELECT toTime64('12:34:56.789', 3) SETTINGS enable_time_time64_type = 1 FORMAT RowBinary",
32      ),
33    ));
34  it("encodes a negative Time64(6)", async () =>
35    expect(encode(writeTime64, parseTime64("-01:02:03.000004", 6))).toEqual(
36      await query(
37        "SELECT toTime64('-01:02:03.000004', 6) SETTINGS enable_time_time64_type = 1 FORMAT RowBinary",
38      ),
39    ));
40});
41
42describe("parseTime64", () => {
43  it("scales whole seconds and the fraction to ticks", () =>
44    expect(parseTime64("12:34:56.789", 3)).toEqual([
45      (12n * 3600n + 34n * 60n + 56n) * 1000n + 789n,
46      3,
47    ]));
48  it("handles a negative value with a short fraction", () =>
49    expect(parseTime64("-01:02:03.5", 1)).toEqual([
50      -((1n * 3600n + 2n * 60n + 3n) * 10n + 5n),
51      1,
52    ]));
53});
54
55describe("writeInterval", () => {
56  it("encodes toIntervalDay(7)", async () =>
57    expect(encode(writeInterval, 7n)).toEqual(
58      await query("SELECT toIntervalDay(7) FORMAT RowBinary"),
59    ));
60  it("encodes a negative toIntervalSecond(-90)", async () =>
61    expect(encode(writeInterval, -90n)).toEqual(
62      await query("SELECT toIntervalSecond(-90) FORMAT RowBinary"),
63    ));
64  it("encodes toIntervalYear(2)", async () =>
65    expect(encode(writeInterval, 2n)).toEqual(
66      await query("SELECT toIntervalYear(2) FORMAT RowBinary"),
67    ));
68});