import { describe, it, expect } from 'vitest';
import { buildInsertTableWithDataRequests } from './insertTableWithData.js';
describe('buildInsertTableWithDataRequests', () => {
describe('table structure', () => {
it('should create an insertTable request with correct dimensions', () => {
const requests = buildInsertTableWithDataRequests(
[
['A', 'B'],
['C', 'D'],
],
1,
false
);
const tableReq = requests.find((r) => r.insertTable);
expect(tableReq).toBeDefined();
expect(tableReq!.insertTable!.rows).toBe(2);
expect(tableReq!.insertTable!.columns).toBe(2);
expect(tableReq!.insertTable!.location!.index).toBe(1);
});
it('should create a single-cell table', () => {
const requests = buildInsertTableWithDataRequests([['Only']], 1, false);
const tableReq = requests.find((r) => r.insertTable);
expect(tableReq!.insertTable!.rows).toBe(1);
expect(tableReq!.insertTable!.columns).toBe(1);
const insertTexts = requests.filter((r) => r.insertText);
expect(insertTexts).toHaveLength(1);
expect(insertTexts[0].insertText!.text).toBe('Only');
});
it('should pad ragged rows to the widest row', () => {
const requests = buildInsertTableWithDataRequests([['A', 'B', 'C'], ['D']], 1, false);
const tableReq = requests.find((r) => r.insertTable);
expect(tableReq!.insertTable!.rows).toBe(2);
expect(tableReq!.insertTable!.columns).toBe(3);
// Only non-empty cells get insertText: A, B, C, D
const insertTexts = requests.filter((r) => r.insertText);
expect(insertTexts).toHaveLength(4);
});
});
describe('cell index math', () => {
// For a 2x2 table at T=1, numCols=2:
// baseCellIndex(1, r, c, 2) = 1 + 4 + r * (1 + 4) + 2*c = 5 + 5r + 2c
// cell(0,0) base = 5, cell(0,1) base = 7
// cell(1,0) base = 10, cell(1,1) base = 12
it('should insert text at correct indices for a 2x2 table', () => {
const requests = buildInsertTableWithDataRequests(
[
['A', 'B'],
['C', 'D'],
],
1,
false
);
const insertTexts = requests.filter((r) => r.insertText);
expect(insertTexts).toHaveLength(4);
// cell(0,0): base=5, cumulative=0 → index=5
expect(insertTexts[0].insertText!.text).toBe('A');
expect(insertTexts[0].insertText!.location!.index).toBe(5);
// cell(0,1): base=7, cumulative=1 (len of "A") → index=8
expect(insertTexts[1].insertText!.text).toBe('B');
expect(insertTexts[1].insertText!.location!.index).toBe(8);
// cell(1,0): base=10, cumulative=2 (len of "A"+"B") → index=12
expect(insertTexts[2].insertText!.text).toBe('C');
expect(insertTexts[2].insertText!.location!.index).toBe(12);
// cell(1,1): base=12, cumulative=3 (len of "A"+"B"+"C") → index=15
expect(insertTexts[3].insertText!.text).toBe('D');
expect(insertTexts[3].insertText!.location!.index).toBe(15);
});
it('should handle multi-character cell text correctly', () => {
const requests = buildInsertTableWithDataRequests(
[
['Hello', 'World'],
['Foo', 'Bar'],
],
1,
false
);
const insertTexts = requests.filter((r) => r.insertText);
// cell(0,0): base=5, cumulative=0 → 5
expect(insertTexts[0].insertText!.location!.index).toBe(5);
// cell(0,1): base=7, cumulative=5 ("Hello") → 12
expect(insertTexts[1].insertText!.location!.index).toBe(12);
// cell(1,0): base=10, cumulative=10 ("Hello"+"World") → 20
expect(insertTexts[2].insertText!.location!.index).toBe(20);
// cell(1,1): base=12, cumulative=13 ("Hello"+"World"+"Foo") → 25
expect(insertTexts[3].insertText!.location!.index).toBe(25);
});
it('should work with a different starting index', () => {
const requests = buildInsertTableWithDataRequests([['X']], 50, false);
const tableReq = requests.find((r) => r.insertTable);
expect(tableReq!.insertTable!.location!.index).toBe(50);
const insertTexts = requests.filter((r) => r.insertText);
// cell(0,0): 50 + 4 + 0 + 0 = 54
expect(insertTexts[0].insertText!.location!.index).toBe(54);
});
it('should handle a 3x3 table correctly', () => {
const requests = buildInsertTableWithDataRequests(
[
['a', 'b', 'c'],
['d', 'e', 'f'],
['g', 'h', 'i'],
],
1,
false
);
const insertTexts = requests.filter((r) => r.insertText);
expect(insertTexts).toHaveLength(9);
// For 3x3 at T=1: baseCellIndex = 1 + 4 + r*(1+6) + 2c = 5 + 7r + 2c
// row 0: bases = 5, 7, 9
// row 1: bases = 12, 14, 16
// row 2: bases = 19, 21, 23
// cell(0,0): base=5, cum=0 → 5
expect(insertTexts[0].insertText!.location!.index).toBe(5);
// cell(0,1): base=7, cum=1 → 8
expect(insertTexts[1].insertText!.location!.index).toBe(8);
// cell(0,2): base=9, cum=2 → 11
expect(insertTexts[2].insertText!.location!.index).toBe(11);
// cell(1,0): base=12, cum=3 → 15
expect(insertTexts[3].insertText!.location!.index).toBe(15);
// cell(1,1): base=14, cum=4 → 18
expect(insertTexts[4].insertText!.location!.index).toBe(18);
// cell(1,2): base=16, cum=5 → 21
expect(insertTexts[5].insertText!.location!.index).toBe(21);
// cell(2,0): base=19, cum=6 → 25
expect(insertTexts[6].insertText!.location!.index).toBe(25);
// cell(2,1): base=21, cum=7 → 28
expect(insertTexts[7].insertText!.location!.index).toBe(28);
// cell(2,2): base=23, cum=8 → 31
expect(insertTexts[8].insertText!.location!.index).toBe(31);
});
});
describe('empty cells', () => {
it('should skip insertText for empty string cells', () => {
const requests = buildInsertTableWithDataRequests(
[
['A', ''],
['', 'D'],
],
1,
false
);
const insertTexts = requests.filter((r) => r.insertText);
expect(insertTexts).toHaveLength(2);
expect(insertTexts[0].insertText!.text).toBe('A');
expect(insertTexts[1].insertText!.text).toBe('D');
});
it('should compute correct indices when cells are skipped', () => {
// 2x2 at T=1, but cell(0,1) is empty
const requests = buildInsertTableWithDataRequests(
[
['A', ''],
['C', 'D'],
],
1,
false
);
const insertTexts = requests.filter((r) => r.insertText);
expect(insertTexts).toHaveLength(3);
// cell(0,0): base=5, cum=0 → 5
expect(insertTexts[0].insertText!.location!.index).toBe(5);
// cell(0,1) skipped
// cell(1,0): base=10, cum=1 ("A") → 11
expect(insertTexts[1].insertText!.location!.index).toBe(11);
// cell(1,1): base=12, cum=2 ("A"+"C") → 14
expect(insertTexts[2].insertText!.location!.index).toBe(14);
});
});
describe('header row bolding', () => {
it('should add updateTextStyle requests for header cells when hasHeaderRow=true', () => {
const requests = buildInsertTableWithDataRequests(
[
['Name', 'Age'],
['Alice', '30'],
],
1,
true
);
const styleReqs = requests.filter((r) => r.updateTextStyle);
expect(styleReqs).toHaveLength(2);
// "Name" bold: range starts at 5 (base for cell(0,0), cum=0), ends at 5+4=9
expect(styleReqs[0].updateTextStyle!.textStyle!.bold).toBe(true);
expect(styleReqs[0].updateTextStyle!.range!.startIndex).toBe(5);
expect(styleReqs[0].updateTextStyle!.range!.endIndex).toBe(9);
// "Age" bold: base for cell(0,1)=7, cum=4 ("Name") → 11, ends at 11+3=14
expect(styleReqs[1].updateTextStyle!.textStyle!.bold).toBe(true);
expect(styleReqs[1].updateTextStyle!.range!.startIndex).toBe(11);
expect(styleReqs[1].updateTextStyle!.range!.endIndex).toBe(14);
});
it('should not add style requests when hasHeaderRow=false', () => {
const requests = buildInsertTableWithDataRequests(
[
['Name', 'Age'],
['Alice', '30'],
],
1,
false
);
const styleReqs = requests.filter((r) => r.updateTextStyle);
expect(styleReqs).toHaveLength(0);
});
it('should not bold empty header cells', () => {
const requests = buildInsertTableWithDataRequests(
[
['Name', ''],
['Alice', '30'],
],
1,
true
);
const styleReqs = requests.filter((r) => r.updateTextStyle);
expect(styleReqs).toHaveLength(1);
expect(styleReqs[0].updateTextStyle!.range!.startIndex).toBe(5);
});
it('should not bold non-header rows', () => {
const requests = buildInsertTableWithDataRequests(
[
['H1', 'H2'],
['A', 'B'],
['C', 'D'],
],
1,
true
);
const styleReqs = requests.filter((r) => r.updateTextStyle);
// Only 2 bold requests (for H1 and H2), not for A, B, C, D
expect(styleReqs).toHaveLength(2);
});
});
describe('tabId propagation', () => {
it('should include tabId in all request locations when provided', () => {
const tabId = 'tab123';
const requests = buildInsertTableWithDataRequests(
[
['A', 'B'],
['C', 'D'],
],
1,
true,
tabId
);
const tableReq = requests.find((r) => r.insertTable);
expect((tableReq!.insertTable!.location as any).tabId).toBe(tabId);
const insertTexts = requests.filter((r) => r.insertText);
for (const req of insertTexts) {
expect((req.insertText!.location as any).tabId).toBe(tabId);
}
const styleReqs = requests.filter((r) => r.updateTextStyle);
for (const req of styleReqs) {
expect(req.updateTextStyle!.range!.tabId).toBe(tabId);
}
});
it('should not include tabId when not provided', () => {
const requests = buildInsertTableWithDataRequests([['A']], 1, false);
const tableReq = requests.find((r) => r.insertTable);
expect((tableReq!.insertTable!.location as any).tabId).toBeUndefined();
const insertTexts = requests.filter((r) => r.insertText);
expect((insertTexts[0].insertText!.location as any).tabId).toBeUndefined();
});
});
describe('error handling', () => {
it('should throw on empty data array', () => {
expect(() => buildInsertTableWithDataRequests([], 1, false)).toThrow();
});
it('should throw on data with only empty rows', () => {
expect(() => buildInsertTableWithDataRequests([[]], 1, false)).toThrow();
});
});
});
