dbx/packages/app-tests/dataGridBooleanColumn.test.ts

import { strict as assert } from "node:assert";
import { readFileSync } from "node:fs";
import { test } from "vitest";
import { BOOLEAN_CELL_EDITOR_VALUES, booleanCellEditorValue, isBooleanCellValue, isBooleanColumnType, normalizeBooleanCellValue, parseBooleanCellEditorValue } from "../../apps/desktop/src/lib/dataGrid/dataGridBooleanColumn.ts";
import { resolveDataGridColumnsByResultIndex } from "../../apps/desktop/src/lib/dataGrid/dataGridColumnMetadata.ts";
import type { ColumnInfo } from "../../apps/desktop/src/types/database.ts";

function column(name: string, dataType: string): ColumnInfo {
  return {
    name,
    data_type: dataType,
    is_nullable: true,
    column_default: null,
    is_primary_key: false,
    extra: null,
  };
}

test("detects boolean types using database semantics", () => {
  assert.equal(isBooleanColumnType("boolean"), true);
  assert.equal(isBooleanColumnType("bool", "postgres"), true);
  assert.equal(isBooleanColumnType("bit", "sqlserver"), true);
  assert.equal(isBooleanColumnType("boolean", "mysql"), true);
  assert.equal(isBooleanColumnType("  BOOLEAN ", "postgres"), true);
});

test("does not treat database bit strings or MySQL integer aliases as boolean", () => {
  assert.equal(isBooleanColumnType("bit", "postgres"), false);
  assert.equal(isBooleanColumnType("bit(1)", "postgres"), false);
  assert.equal(isBooleanColumnType("bit varying", "postgres"), false);
  assert.equal(isBooleanColumnType("varbit", "postgres"), false);
  assert.equal(isBooleanColumnType("bit", "opengauss"), false);
  assert.equal(isBooleanColumnType("bit", undefined), false);
  assert.equal(isBooleanColumnType("bit", "mysql"), false);
  assert.equal(isBooleanColumnType("bit(1)", "mysql"), false);
  assert.equal(isBooleanColumnType("bit(8)", "mysql"), false);
  assert.equal(isBooleanColumnType("tinyint(1)", "mysql"), false);
  assert.equal(isBooleanColumnType(undefined, "mysql"), false);
});

test("normalizes raw cell values to a tri-state boolean", () => {
  assert.equal(normalizeBooleanCellValue(true), true);
  assert.equal(normalizeBooleanCellValue(false), false);
  assert.equal(normalizeBooleanCellValue(1), true);
  assert.equal(normalizeBooleanCellValue(0), false);
  assert.equal(normalizeBooleanCellValue("true"), true);
  assert.equal(normalizeBooleanCellValue("false"), false);
  assert.equal(normalizeBooleanCellValue("t"), true);
  assert.equal(normalizeBooleanCellValue("0"), false);
  assert.equal(normalizeBooleanCellValue(null), null);
  assert.equal(normalizeBooleanCellValue(undefined), null);
  assert.equal(normalizeBooleanCellValue("maybe"), null);
});

test("opens boolean editors only for recognized boolean cell values", () => {
  assert.equal(isBooleanCellValue(true), true);
  assert.equal(isBooleanCellValue(0), true);
  assert.equal(isBooleanCellValue("false"), true);
  assert.equal(isBooleanCellValue(null), true);
  assert.equal(isBooleanCellValue(undefined), false);
  assert.equal(isBooleanCellValue("maybe"), false);
  assert.equal(isBooleanCellValue({}), false);
});

test("normalizes boolean values for the enum-style editor", () => {
  assert.deepEqual(BOOLEAN_CELL_EDITOR_VALUES, ["true", "false"]);
  assert.equal(booleanCellEditorValue(true), "true");
  assert.equal(booleanCellEditorValue(1), "true");
  assert.equal(booleanCellEditorValue(false), "false");
  assert.equal(booleanCellEditorValue("0"), "false");
  assert.equal(booleanCellEditorValue(null), "");
  assert.equal(booleanCellEditorValue("maybe"), "");
});

test("parses explicit enum-style boolean selections", () => {
  assert.equal(parseBooleanCellEditorValue("true"), true);
  assert.equal(parseBooleanCellEditorValue("false"), false);
  assert.equal(parseBooleanCellEditorValue(null), null);
  assert.equal(parseBooleanCellEditorValue("maybe"), undefined);
});

test("indexes table metadata once and resolves source-column aliases", () => {
  const enabled = column("Enabled", "boolean");
  const displayName = column("DisplayName", "varchar");
  const resolved = resolveDataGridColumnsByResultIndex({
    resultColumns: ["enabled_alias", "DisplayName", "missing"],
    sourceColumns: ["enabled", undefined, undefined],
    tableColumns: [enabled, displayName],
  });

  assert.equal(resolved[0], enabled);
  assert.equal(resolved[1], displayName);
  assert.equal(resolved[2], undefined);
});

test("uses the enum editor for boolean cells without checkbox rendering or click cycling", () => {
  const gridSource = readFileSync("apps/desktop/src/components/grid/DataGrid.vue", "utf8");
  const rendererSource = readFileSync("apps/desktop/src/lib/dataGrid/canvasDataGridRenderer.ts", "utf8");

  assert.match(gridSource, /v-else-if="isBooleanGridCell\([^\n]+"[\s\S]*?v-model="booleanEditorModelValue"[\s\S]*?:values="BOOLEAN_CELL_EDITOR_VALUES"/);
  assert.match(gridSource, /@commit="commitBooleanGridEdit"/);
  assert.doesNotMatch(gridSource, /cycleBooleanCellValue|tryCycleBooleanCheckboxOnCanvasMouseDown|booleanCellChecked/);
  assert.doesNotMatch(rendererSource, /drawBooleanCheckbox|BOOLEAN_CHECKBOX_SIZE|columnIsBoolean/);
});

test("uses the indexed metadata lookup in grid hot paths", () => {
  const source = readFileSync("apps/desktop/src/components/grid/DataGrid.vue", "utf8");
  const start = source.indexOf("function tableColumnForGridColumn");
  const end = source.indexOf("function resultColumnInfoForGridColumn", start);
  const lookup = source.slice(start, end);

  assert.ok(start >= 0 && end > start);
  assert.match(lookup, /tableColumnsByResultIndex\.value\[columnIndex\]/);
  assert.doesNotMatch(lookup, /\.find\(/);
});