import { Fun } from '@ephox/katamari';
import * as Structs from 'ephox/snooker/api/Structs';
import TableGrid from 'ephox/snooker/model/TableGrid';
import { UnitTest, assert } from '@ephox/bedrock';
import { Element } from '@ephox/sugar';

UnitTest.test('TableGrid.subgrid test', function () {
  const r = Structs.rowcells;
  const en = (fakeElement: any, isNew: boolean) => Structs.elementnew(fakeElement as any as Element, isNew);

  const check = function (expected: { colspan: number, rowspan: number }, row: number, column: number, grid: Structs.RowCells[]) {
    const actual = TableGrid.subgrid(grid, row, column, Fun.tripleEquals);
    assert.eq(expected.rowspan, actual.rowspan());
    assert.eq(expected.colspan, actual.colspan());
  };

  const world = [
    r([ en('a', false), en('a', false), en('a', false) ], 'thead'),
    r([ en('b', false), en('b', false), en('c', false) ], 'tbody'),
    r([ en('d', false), en('e', false), en('c', false) ], 'tfoot')
  ];

  check({ colspan: 3, rowspan: 1 }, 0, 0, world);
  check({ colspan: 2, rowspan: 1 }, 0, 1, world);
  check({ colspan: 2, rowspan: 1 }, 1, 0, world);
  check({ colspan: 1, rowspan: 1 }, 2, 0, world);
});