Files
2026-07-13 13:09:14 +08:00

111 lines
3.4 KiB
TypeScript
Vendored

import { strict as assert } from "node:assert";
import { test } from "vitest";
import {
allCellsSelectionRange,
columnSelectionRange,
extractColumnsSelection,
extractSelection,
formatSelectionAsCsv,
formatSelectionAsJson,
formatSelectionAsSqlInList,
formatSelectionAsTsv,
isCellInSelection,
normalizeSelectionRange,
parseClipboardTable,
rowSelectionRange,
} from "../../apps/desktop/src/lib/dataGrid/gridSelection.ts";
test("normalizes a dragged cell range in either direction", () => {
const range = normalizeSelectionRange({ rowIndex: 4, colIndex: 3 }, { rowIndex: 1, colIndex: 0 });
assert.deepEqual(range, {
startRow: 1,
endRow: 4,
startCol: 0,
endCol: 3,
});
assert.equal(isCellInSelection(2, 1, range), true);
assert.equal(isCellInSelection(5, 1, range), false);
});
test("extracts selection rows and columns from a rectangular range", () => {
const selection = extractSelection(
["id", "name", "active"],
[
[1, "Ada", true],
[2, "Linus", false],
[3, null, true],
],
{ startRow: 0, endRow: 1, startCol: 1, endCol: 2 },
);
assert.deepEqual(selection.columns, ["name", "active"]);
assert.deepEqual(selection.rows, [
["Ada", true],
["Linus", false],
]);
});
test("extracts non-contiguous columns in display order", () => {
const selection = extractColumnsSelection(
["id", "name", "active", "email"],
[
[1, "Ada", true, "ada@example.com"],
[2, "Linus", false, "linus@example.com"],
],
[3, 1, 3],
);
assert.deepEqual(selection.columns, ["name", "email"]);
assert.deepEqual(selection.rows, [
["Ada", "ada@example.com"],
["Linus", "linus@example.com"],
]);
});
test("builds whole row, column, and table selection ranges", () => {
assert.deepEqual(rowSelectionRange(2, 4), { startRow: 2, endRow: 2, startCol: 0, endCol: 3 });
assert.deepEqual(rowSelectionRange(4, 4, 2), { startRow: 2, endRow: 4, startCol: 0, endCol: 3 });
assert.deepEqual(columnSelectionRange(5, 1), { startRow: 0, endRow: 4, startCol: 1, endCol: 1 });
assert.deepEqual(columnSelectionRange(5, 3, 1), { startRow: 0, endRow: 4, startCol: 1, endCol: 3 });
assert.deepEqual(allCellsSelectionRange(3, 2), { startRow: 0, endRow: 2, startCol: 0, endCol: 1 });
assert.equal(rowSelectionRange(0, 0), null);
assert.equal(columnSelectionRange(0, 0), null);
assert.equal(allCellsSelectionRange(0, 2), null);
});
test("formats selected cells as TSV, CSV, JSON, and SQL values", () => {
const selection = {
columns: ["name", "note"],
rows: [
["Ada", "math"],
["Bob", 'quote "here"'],
["O'Hara", null],
],
};
assert.equal(formatSelectionAsTsv(selection), 'Ada\tmath\nBob\tquote "here"\nO\'Hara\tNULL');
assert.equal(formatSelectionAsCsv(selection), '"name","note"\n"Ada","math"\n"Bob","quote ""here"""\n"O\'Hara","NULL"');
assert.equal(
formatSelectionAsJson(selection),
JSON.stringify(
[
{ name: "Ada", note: "math" },
{ name: "Bob", note: 'quote "here"' },
{ name: "O'Hara", note: null },
],
null,
2,
),
);
assert.equal(formatSelectionAsSqlInList(selection), "('Ada', 'math', 'Bob', 'quote \"here\"', 'O''Hara', NULL)");
});
test("parses clipboard table text with spreadsheet line endings", () => {
assert.deepEqual(parseClipboardTable("0"), [["0"]]);
assert.deepEqual(parseClipboardTable("a\tb\r\nc\td\r\n"), [
["a", "b"],
["c", "d"],
]);
});