tableutils.js 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963
  1. /**
  2. * @license Copyright (c) 2003-2020, CKSource - Frederico Knabben. All rights reserved.
  3. * For licensing, see LICENSE.md or https://ckeditor.com/legal/ckeditor-oss-license
  4. */
  5. import ModelTestEditor from '@ckeditor/ckeditor5-core/tests/_utils/modeltesteditor';
  6. import { getData, setData } from '@ckeditor/ckeditor5-engine/src/dev-utils/model';
  7. import { defaultConversion, defaultSchema, modelTable } from './_utils/utils';
  8. import TableUtils from '../src/tableutils';
  9. import { assertEqualMarkup } from '@ckeditor/ckeditor5-utils/tests/_utils/utils';
  10. describe( 'TableUtils', () => {
  11. let editor, model, root, tableUtils;
  12. beforeEach( () => {
  13. return ModelTestEditor.create( {
  14. plugins: [ TableUtils ]
  15. } ).then( newEditor => {
  16. editor = newEditor;
  17. model = editor.model;
  18. root = model.document.getRoot( 'main' );
  19. tableUtils = editor.plugins.get( TableUtils );
  20. defaultSchema( model.schema );
  21. defaultConversion( editor.conversion );
  22. } );
  23. } );
  24. afterEach( () => {
  25. return editor.destroy();
  26. } );
  27. describe( '#pluginName', () => {
  28. it( 'should provide plugin name', () => {
  29. expect( TableUtils.pluginName ).to.equal( 'TableUtils' );
  30. } );
  31. } );
  32. describe( 'getCellLocation()', () => {
  33. it( 'should return proper table cell location', () => {
  34. setData( model, modelTable( [
  35. [ { rowspan: 2, colspan: 2, contents: '00[]' }, '02' ],
  36. [ '12' ]
  37. ] ) );
  38. expect( tableUtils.getCellLocation( root.getNodeByPath( [ 0, 0, 0 ] ) ) ).to.deep.equal( { row: 0, column: 0 } );
  39. expect( tableUtils.getCellLocation( root.getNodeByPath( [ 0, 0, 1 ] ) ) ).to.deep.equal( { row: 0, column: 2 } );
  40. expect( tableUtils.getCellLocation( root.getNodeByPath( [ 0, 1, 0 ] ) ) ).to.deep.equal( { row: 1, column: 2 } );
  41. } );
  42. } );
  43. describe( 'insertRows()', () => {
  44. it( 'should insert row in given table at given index', () => {
  45. setData( model, modelTable( [
  46. [ '11[]', '12' ],
  47. [ '21', '22' ]
  48. ] ) );
  49. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 1 } );
  50. assertEqualMarkup( getData( model ), modelTable( [
  51. [ '11[]', '12' ],
  52. [ '', '' ],
  53. [ '21', '22' ]
  54. ] ) );
  55. } );
  56. it( 'should insert row in given table at default index', () => {
  57. setData( model, modelTable( [
  58. [ '11[]', '12' ],
  59. [ '21', '22' ]
  60. ] ) );
  61. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ) );
  62. assertEqualMarkup( getData( model ), modelTable( [
  63. [ '', '' ],
  64. [ '11[]', '12' ],
  65. [ '21', '22' ]
  66. ] ) );
  67. } );
  68. it( 'should update table heading rows attribute when inserting row in headings section', () => {
  69. setData( model, modelTable( [
  70. [ '11[]', '12' ],
  71. [ '21', '22' ],
  72. [ '31', '32' ]
  73. ], { headingRows: 2 } ) );
  74. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 1 } );
  75. assertEqualMarkup( getData( model ), modelTable( [
  76. [ '11[]', '12' ],
  77. [ '', '' ],
  78. [ '21', '22' ],
  79. [ '31', '32' ]
  80. ], { headingRows: 3 } ) );
  81. } );
  82. it( 'should not update table heading rows attribute when inserting row after headings section', () => {
  83. setData( model, modelTable( [
  84. [ '11[]', '12' ],
  85. [ '21', '22' ],
  86. [ '31', '32' ]
  87. ], { headingRows: 2 } ) );
  88. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 2 } );
  89. assertEqualMarkup( getData( model ), modelTable( [
  90. [ '11[]', '12' ],
  91. [ '21', '22' ],
  92. [ '', '' ],
  93. [ '31', '32' ]
  94. ], { headingRows: 2 } ) );
  95. } );
  96. it( 'should expand rowspan of a cell that overlaps inserted rows', () => {
  97. setData( model, modelTable( [
  98. [ { colspan: 2, contents: '11[]' }, '13', '14' ],
  99. [ { colspan: 2, rowspan: 4, contents: '21' }, '23', '24' ],
  100. [ '33', '34' ]
  101. ], { headingColumns: 3, headingRows: 1 } ) );
  102. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 2, rows: 3 } );
  103. assertEqualMarkup( getData( model ), modelTable( [
  104. [ { colspan: 2, contents: '11[]' }, '13', '14' ],
  105. [ { colspan: 2, rowspan: 7, contents: '21' }, '23', '24' ],
  106. [ '', '' ],
  107. [ '', '' ],
  108. [ '', '' ],
  109. [ '33', '34' ]
  110. ], { headingColumns: 3, headingRows: 1 } ) );
  111. } );
  112. it( 'should not expand rowspan of a cell that does not overlaps inserted rows', () => {
  113. setData( model, modelTable( [
  114. [ { rowspan: 2, contents: '11[]' }, '12', '13' ],
  115. [ '22', '23' ],
  116. [ '31', '32', '33' ]
  117. ], { headingColumns: 3, headingRows: 1 } ) );
  118. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 2, rows: 3 } );
  119. assertEqualMarkup( getData( model ), modelTable( [
  120. [ { rowspan: 2, contents: '11[]' }, '12', '13' ],
  121. [ '22', '23' ],
  122. [ '', '', '' ],
  123. [ '', '', '' ],
  124. [ '', '', '' ],
  125. [ '31', '32', '33' ]
  126. ], { headingColumns: 3, headingRows: 1 } ) );
  127. } );
  128. it( 'should properly calculate columns if next row has colspans', () => {
  129. setData( model, modelTable( [
  130. [ { rowspan: 2, contents: '11[]' }, '12', '13' ],
  131. [ '22', '23' ],
  132. [ { colspan: 3, contents: '31' } ]
  133. ], { headingColumns: 3, headingRows: 1 } ) );
  134. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 2, rows: 3 } );
  135. assertEqualMarkup( getData( model ), modelTable( [
  136. [ { rowspan: 2, contents: '11[]' }, '12', '13' ],
  137. [ '22', '23' ],
  138. [ '', '', '' ],
  139. [ '', '', '' ],
  140. [ '', '', '' ],
  141. [ { colspan: 3, contents: '31' } ]
  142. ], { headingColumns: 3, headingRows: 1 } ) );
  143. } );
  144. it( 'should insert rows at the end of a table', () => {
  145. setData( model, modelTable( [
  146. [ '11[]', '12' ],
  147. [ '21', '22' ]
  148. ] ) );
  149. tableUtils.insertRows( root.getNodeByPath( [ 0 ] ), { at: 2, rows: 3 } );
  150. assertEqualMarkup( getData( model ), modelTable( [
  151. [ '11[]', '12' ],
  152. [ '21', '22' ],
  153. [ '', '' ],
  154. [ '', '' ],
  155. [ '', '' ]
  156. ] ) );
  157. } );
  158. } );
  159. describe( 'insertColumns()', () => {
  160. it( 'should insert column in given table at given index', () => {
  161. setData( model, modelTable( [
  162. [ '11[]', '12' ],
  163. [ '21', '22' ]
  164. ] ) );
  165. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 1 } );
  166. assertEqualMarkup( getData( model ), modelTable( [
  167. [ '11[]', '', '12' ],
  168. [ '21', '', '22' ]
  169. ] ) );
  170. } );
  171. it( 'should insert column in given table with default values', () => {
  172. setData( model, modelTable( [
  173. [ '11[]', '12' ],
  174. [ '21', '22' ]
  175. ] ) );
  176. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ) );
  177. assertEqualMarkup( getData( model ), modelTable( [
  178. [ '', '11[]', '12' ],
  179. [ '', '21', '22' ]
  180. ] ) );
  181. } );
  182. it( 'should insert column in given table at default index', () => {
  183. setData( model, modelTable( [
  184. [ '11[]', '12' ],
  185. [ '21', '22' ]
  186. ] ) );
  187. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ) );
  188. assertEqualMarkup( getData( model ), modelTable( [
  189. [ '', '11[]', '12' ],
  190. [ '', '21', '22' ]
  191. ] ) );
  192. } );
  193. it( 'should insert columns at the end of a row', () => {
  194. setData( model, modelTable( [
  195. [ '00[]', '01' ],
  196. [ { colspan: 2, contents: '10' } ],
  197. [ '20', { rowspan: 2, contents: '21' } ],
  198. [ '30' ]
  199. ] ) );
  200. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 2, columns: 2 } );
  201. assertEqualMarkup( getData( model ), modelTable( [
  202. [ '00[]', '01', '', '' ],
  203. [ { colspan: 2, contents: '10' }, '', '' ],
  204. [ '20', { rowspan: 2, contents: '21' }, '', '' ],
  205. [ '30', '', '' ]
  206. ] ) );
  207. } );
  208. it( 'should insert columns at the beginning of a row', () => {
  209. setData( model, modelTable( [
  210. [ '00[]', '01' ],
  211. [ { colspan: 2, contents: '10' } ],
  212. [ '20', { rowspan: 2, contents: '21' } ],
  213. [ { rowspan: 2, contents: '30' } ],
  214. [ '41' ]
  215. ] ) );
  216. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 0, columns: 2 } );
  217. assertEqualMarkup( getData( model ), modelTable( [
  218. [ '', '', '00[]', '01' ],
  219. [ '', '', { colspan: 2, contents: '10' } ],
  220. [ '', '', '20', { rowspan: 2, contents: '21' } ],
  221. [ '', '', { rowspan: 2, contents: '30' } ],
  222. [ '', '', '41' ]
  223. ] ) );
  224. } );
  225. it( 'should properly insert column at beginning of row-col-spanned cell', () => {
  226. setData( model, modelTable( [
  227. [ '11', '12', '13' ],
  228. [ '21', { colspan: 2, rowspan: 2, contents: '22[]' } ],
  229. [ '31' ],
  230. [ '41', '42', '43' ]
  231. ] ) );
  232. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 1, columns: 1 } );
  233. assertEqualMarkup( getData( model ), modelTable( [
  234. [ '11', '', '12', '13' ],
  235. [ '21', '', { colspan: 2, rowspan: 2, contents: '22[]' } ],
  236. [ '31', '' ],
  237. [ '41', '', '42', '43' ]
  238. ] ) );
  239. } );
  240. it( 'should update table heading columns attribute when inserting column in headings section', () => {
  241. setData( model, modelTable( [
  242. [ '11[]', '12' ],
  243. [ '21', '22' ],
  244. [ '31', '32' ]
  245. ], { headingColumns: 2 } ) );
  246. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 1 } );
  247. assertEqualMarkup( getData( model ), modelTable( [
  248. [ '11[]', '', '12' ],
  249. [ '21', '', '22' ],
  250. [ '31', '', '32' ]
  251. ], { headingColumns: 3 } ) );
  252. } );
  253. it( 'should not update table heading columns attribute when inserting column after headings section', () => {
  254. setData( model, modelTable( [
  255. [ '11[]', '12', '13' ],
  256. [ '21', '22', '23' ],
  257. [ '31', '32', '33' ]
  258. ], { headingColumns: 2 } ) );
  259. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 2 } );
  260. assertEqualMarkup( getData( model ), modelTable( [
  261. [ '11[]', '12', '', '13' ],
  262. [ '21', '22', '', '23' ],
  263. [ '31', '32', '', '33' ]
  264. ], { headingColumns: 2 } ) );
  265. } );
  266. it( 'should expand spanned columns', () => {
  267. setData( model, modelTable( [
  268. [ '00[]', '01' ],
  269. [ { colspan: 2, contents: '10' } ],
  270. [ '20', '21' ]
  271. ], { headingColumns: 2 } ) );
  272. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 1 } );
  273. assertEqualMarkup( getData( model ), modelTable( [
  274. [ '00[]', '', '01' ],
  275. [ { colspan: 3, contents: '10' } ],
  276. [ '20', '', '21' ]
  277. ], { headingColumns: 3 } ) );
  278. } );
  279. it( 'should skip wide spanned columns', () => {
  280. setData( model, modelTable( [
  281. [ '11[]', '12', '13', '14', '15' ],
  282. [ '21', '22', { colspan: 2, contents: '23' }, '25' ],
  283. [ { colspan: 4, contents: '31' }, { colspan: 2, contents: '34' } ]
  284. ], { headingColumns: 4 } ) );
  285. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 2, columns: 2 } );
  286. assertEqualMarkup( getData( model ), modelTable( [
  287. [ '11[]', '12', '', '', '13', '14', '15' ],
  288. [ '21', '22', '', '', { colspan: 2, contents: '23' }, '25' ],
  289. [ { colspan: 6, contents: '31' }, { colspan: 2, contents: '34' } ]
  290. ], { headingColumns: 6 } ) );
  291. } );
  292. it( 'should skip row & column spanned cells', () => {
  293. setData( model, modelTable( [
  294. [ { colspan: 2, rowspan: 2, contents: '00[]' }, '02' ],
  295. [ '12' ],
  296. [ '20', '21', '22' ]
  297. ], { headingColumns: 2 } ) );
  298. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 1, columns: 2 } );
  299. assertEqualMarkup( getData( model ), modelTable( [
  300. [ { colspan: 4, rowspan: 2, contents: '00[]' }, '02' ],
  301. [ '12' ],
  302. [ '20', '', '', '21', '22' ]
  303. ], { headingColumns: 4 } ) );
  304. } );
  305. it( 'should properly insert column while table has rowspanned cells', () => {
  306. setData( model, modelTable( [
  307. [ { rowspan: 4, contents: '00[]' }, { rowspan: 2, contents: '01' }, '02' ],
  308. [ '12' ],
  309. [ { rowspan: 2, contents: '21' }, '22' ],
  310. [ '32' ]
  311. ], { headingColumns: 2 } ) );
  312. tableUtils.insertColumns( root.getNodeByPath( [ 0 ] ), { at: 1, columns: 1 } );
  313. assertEqualMarkup( getData( model ), modelTable( [
  314. [ { rowspan: 4, contents: '00[]' }, '', { rowspan: 2, contents: '01' }, '02' ],
  315. [ '', '12' ],
  316. [ '', { rowspan: 2, contents: '21' }, '22' ],
  317. [ '', '32' ]
  318. ], { headingColumns: 3 } ) );
  319. } );
  320. } );
  321. describe( 'splitCellVertically()', () => {
  322. it( 'should split table cell to given table cells number', () => {
  323. setData( model, modelTable( [
  324. [ '00', '01', '02' ],
  325. [ '10', '[]11', '12' ],
  326. [ '20', { colspan: 2, contents: '21' } ],
  327. [ { colspan: 2, contents: '30' }, '32' ]
  328. ] ) );
  329. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 1 ] ), 3 );
  330. assertEqualMarkup( getData( model ), modelTable( [
  331. [ '00', { colspan: 3, contents: '01' }, '02' ],
  332. [ '10', '[]11', '', '', '12' ],
  333. [ '20', { colspan: 4, contents: '21' } ],
  334. [ { colspan: 4, contents: '30' }, '32' ]
  335. ] ) );
  336. } );
  337. it( 'should split table cell for two table cells as a default', () => {
  338. setData( model, modelTable( [
  339. [ '00', '01', '02' ],
  340. [ '10', '[]11', '12' ],
  341. [ '20', { colspan: 2, contents: '21' } ],
  342. [ { colspan: 2, contents: '30' }, '32' ]
  343. ] ) );
  344. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 1 ] ) );
  345. assertEqualMarkup( getData( model ), modelTable( [
  346. [ '00', { colspan: 2, contents: '01' }, '02' ],
  347. [ '10', '[]11', '', '12' ],
  348. [ '20', { colspan: 3, contents: '21' } ],
  349. [ { colspan: 3, contents: '30' }, '32' ]
  350. ] ) );
  351. } );
  352. it( 'should split table cell if split is equal to colspan', () => {
  353. setData( model, modelTable( [
  354. [ '00', '01', '02' ],
  355. [ '10', '11', '12' ],
  356. [ '20', { colspan: 2, contents: '21[]' } ],
  357. [ { colspan: 2, contents: '30' }, '32' ]
  358. ] ) );
  359. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 2, 1 ] ), 2 );
  360. assertEqualMarkup( getData( model ), modelTable( [
  361. [ '00', '01', '02' ],
  362. [ '10', '11', '12' ],
  363. [ '20', '21[]', '' ],
  364. [ { colspan: 2, contents: '30' }, '32' ]
  365. ] ) );
  366. } );
  367. it( 'should properly split table cell if split is uneven', () => {
  368. setData( model, modelTable( [
  369. [ '00', '01', '02' ],
  370. [ { colspan: 3, contents: '10[]' } ]
  371. ] ) );
  372. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 0 ] ), 2 );
  373. assertEqualMarkup( getData( model ), modelTable( [
  374. [ '00', '01', '02' ],
  375. [ { colspan: 2, contents: '10[]' }, '' ]
  376. ] ) );
  377. } );
  378. it( 'should properly set colspan of inserted cells', () => {
  379. setData( model, modelTable( [
  380. [ '00', '01', '02', '03' ],
  381. [ { colspan: 4, contents: '10[]' } ]
  382. ] ) );
  383. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 0 ] ), 2 );
  384. assertEqualMarkup( getData( model ), modelTable( [
  385. [ '00', '01', '02', '03' ],
  386. [ { colspan: 2, contents: '10[]' }, { colspan: 2, contents: '' } ]
  387. ] ) );
  388. } );
  389. it( 'should keep rowspan attribute for newly inserted cells', () => {
  390. setData( model, modelTable( [
  391. [ '00', '01', '02', '03', '04', '05' ],
  392. [ { colspan: 5, rowspan: 2, contents: '10[]' }, '15' ],
  393. [ '25' ]
  394. ] ) );
  395. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 0 ] ), 2 );
  396. assertEqualMarkup( getData( model ), modelTable( [
  397. [ '00', '01', '02', '03', '04', '05' ],
  398. [ { colspan: 3, rowspan: 2, contents: '10[]' }, { colspan: 2, rowspan: 2, contents: '' }, '15' ],
  399. [ '25' ]
  400. ] ) );
  401. } );
  402. it( 'should keep rowspan attribute of for newly inserted cells if number of cells is bigger then curren colspan', () => {
  403. setData( model, modelTable( [
  404. [ '00', '01', '02' ],
  405. [ { colspan: 2, rowspan: 2, contents: '10[]' }, '12' ],
  406. [ '22' ]
  407. ] ) );
  408. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 0 ] ), 3 );
  409. assertEqualMarkup( getData( model ), modelTable( [
  410. [ { colspan: 2, contents: '00' }, '01', '02' ],
  411. [ { rowspan: 2, contents: '10[]' }, { rowspan: 2, contents: '' }, { rowspan: 2, contents: '' }, '12' ],
  412. [ '22' ]
  413. ] ) );
  414. } );
  415. it( 'should properly break a cell if it has colspan and number of created cells is bigger then colspan', () => {
  416. setData( model, modelTable( [
  417. [ '00', '01', '02', '03' ],
  418. [ { colspan: 4, contents: '10[]' } ]
  419. ] ) );
  420. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 0 ] ), 6 );
  421. assertEqualMarkup( getData( model ), modelTable( [
  422. [ { colspan: 3, contents: '00' }, '01', '02', '03' ],
  423. [ '10[]', '', '', '', '', '' ]
  424. ] ) );
  425. } );
  426. it( 'should update heading columns is split cell is in heading section', () => {
  427. setData( model, modelTable( [
  428. [ '00', '01' ],
  429. [ '10[]', '11' ]
  430. ], { headingColumns: 1 } ) );
  431. tableUtils.splitCellVertically( root.getNodeByPath( [ 0, 1, 0 ] ), 3 );
  432. assertEqualMarkup( getData( model ), modelTable( [
  433. [ { colspan: 3, contents: '00' }, '01' ],
  434. [ '10[]', '', '', '11' ]
  435. ], { headingColumns: 3 } ) );
  436. } );
  437. } );
  438. describe( 'splitCellHorizontally()', () => {
  439. it( 'should split table cell to default table cells number', () => {
  440. setData( model, modelTable( [
  441. [ '00', '01', '02' ],
  442. [ '10', '[]11', '12' ],
  443. [ '20', '21', '22' ]
  444. ] ) );
  445. tableUtils.splitCellHorizontally( root.getNodeByPath( [ 0, 1, 1 ] ) );
  446. assertEqualMarkup( getData( model ), modelTable( [
  447. [ '00', '01', '02' ],
  448. [ { rowspan: 2, contents: '10' }, '[]11', { rowspan: 2, contents: '12' } ],
  449. [ '' ],
  450. [ '20', '21', '22' ]
  451. ] ) );
  452. } );
  453. it( 'should split table cell to given table cells number', () => {
  454. setData( model, modelTable( [
  455. [ '00', '01', '02' ],
  456. [ '10', '[]11', '12' ],
  457. [ '20', '21', '22' ]
  458. ] ) );
  459. tableUtils.splitCellHorizontally( root.getNodeByPath( [ 0, 1, 1 ] ), 4 );
  460. assertEqualMarkup( getData( model ), modelTable( [
  461. [ '00', '01', '02' ],
  462. [ { rowspan: 4, contents: '10' }, '[]11', { rowspan: 4, contents: '12' } ],
  463. [ '' ],
  464. [ '' ],
  465. [ '' ],
  466. [ '20', '21', '22' ]
  467. ] ) );
  468. } );
  469. it( 'should properly update rowspanned cells overlapping selected cell', () => {
  470. setData( model, modelTable( [
  471. [ { rowspan: 2, contents: '00' }, '01', { rowspan: 3, contents: '02' } ],
  472. [ '[]11' ],
  473. [ '20', '21' ]
  474. ] ) );
  475. tableUtils.splitCellHorizontally( root.getNodeByPath( [ 0, 1, 0 ] ), 3 );
  476. assertEqualMarkup( getData( model ), modelTable( [
  477. [ { rowspan: 4, contents: '00' }, '01', { rowspan: 5, contents: '02' } ],
  478. [ '[]11' ],
  479. [ '' ],
  480. [ '' ],
  481. [ '20', '21' ]
  482. ] ) );
  483. } );
  484. it( 'should split rowspanned cell', () => {
  485. setData( model, modelTable( [
  486. [ '00', { rowspan: 2, contents: '01[]' } ],
  487. [ '10' ],
  488. [ '20', '21' ]
  489. ] ) );
  490. const tableCell = root.getNodeByPath( [ 0, 0, 1 ] );
  491. tableUtils.splitCellHorizontally( tableCell, 2 );
  492. assertEqualMarkup( getData( model ), modelTable( [
  493. [ '00', '01[]' ],
  494. [ '10', '' ],
  495. [ '20', '21' ]
  496. ] ) );
  497. } );
  498. it( 'should copy colspan while splitting rowspanned cell', () => {
  499. setData( model, modelTable( [
  500. [ '00', { rowspan: 2, colspan: 2, contents: '01[]' } ],
  501. [ '10' ],
  502. [ '20', '21', '22' ]
  503. ] ) );
  504. const tableCell = root.getNodeByPath( [ 0, 0, 1 ] );
  505. tableUtils.splitCellHorizontally( tableCell, 2 );
  506. assertEqualMarkup( getData( model ), modelTable( [
  507. [ '00', { colspan: 2, contents: '01[]' } ],
  508. [ '10', { colspan: 2, contents: '' } ],
  509. [ '20', '21', '22' ]
  510. ] ) );
  511. } );
  512. it( 'should evenly distribute rowspan attribute', () => {
  513. setData( model, modelTable( [
  514. [ '00', { rowspan: 7, contents: '01[]' } ],
  515. [ '10' ],
  516. [ '20' ],
  517. [ '30' ],
  518. [ '40' ],
  519. [ '50' ],
  520. [ '60' ],
  521. [ '70', '71' ]
  522. ] ) );
  523. const tableCell = root.getNodeByPath( [ 0, 0, 1 ] );
  524. tableUtils.splitCellHorizontally( tableCell, 3 );
  525. assertEqualMarkup( getData( model ), modelTable( [
  526. [ '00', { rowspan: 3, contents: '01[]' } ],
  527. [ '10' ],
  528. [ '20' ],
  529. [ '30', { rowspan: 2, contents: '' } ],
  530. [ '40' ],
  531. [ '50', { rowspan: 2, contents: '' } ],
  532. [ '60' ],
  533. [ '70', '71' ]
  534. ] ) );
  535. } );
  536. it( 'should split rowspanned cell and updated other cells rowspan when splitting to bigger number of cells', () => {
  537. setData( model, modelTable( [
  538. [ '00', { rowspan: 2, contents: '01[]' } ],
  539. [ '10' ],
  540. [ '20', '21' ]
  541. ] ) );
  542. const tableCell = root.getNodeByPath( [ 0, 0, 1 ] );
  543. tableUtils.splitCellHorizontally( tableCell, 3 );
  544. assertEqualMarkup( getData( model ), modelTable( [
  545. [ { rowspan: 2, contents: '00' }, '01[]' ],
  546. [ '' ],
  547. [ '10', '' ],
  548. [ '20', '21' ]
  549. ] ) );
  550. } );
  551. it( 'should split rowspanned & colspaned cell', () => {
  552. setData( model, modelTable( [
  553. [ '00', { colspan: 2, contents: '01[]' } ],
  554. [ '10', '11' ]
  555. ] ) );
  556. const tableCell = root.getNodeByPath( [ 0, 0, 1 ] );
  557. tableUtils.splitCellHorizontally( tableCell, 3 );
  558. assertEqualMarkup( getData( model ), modelTable( [
  559. [ { rowspan: 3, contents: '00' }, { colspan: 2, contents: '01[]' } ],
  560. [ { colspan: 2, contents: '' } ],
  561. [ { colspan: 2, contents: '' } ],
  562. [ '10', '11' ]
  563. ] ) );
  564. } );
  565. it( 'should split table cell from a heading section', () => {
  566. setData( model, modelTable( [
  567. [ '00[]', '01', '02' ],
  568. [ '10', '11', '12' ],
  569. [ '20', '21', '22' ]
  570. ], { headingRows: 1 } ) );
  571. tableUtils.splitCellHorizontally( root.getNodeByPath( [ 0, 0, 0 ] ), 3 );
  572. assertEqualMarkup( getData( model ), modelTable( [
  573. [ '00[]', { rowspan: 3, contents: '01' }, { rowspan: 3, contents: '02' } ],
  574. [ '' ],
  575. [ '' ],
  576. [ '10', '11', '12' ],
  577. [ '20', '21', '22' ]
  578. ], { headingRows: 3 } ) );
  579. } );
  580. } );
  581. describe( 'getColumns()', () => {
  582. it( 'should return proper number of columns', () => {
  583. setData( model, modelTable( [
  584. [ '00', { colspan: 3, contents: '01' }, '04' ]
  585. ] ) );
  586. expect( tableUtils.getColumns( root.getNodeByPath( [ 0 ] ) ) ).to.equal( 5 );
  587. } );
  588. } );
  589. describe( 'getRows()', () => {
  590. it( 'should return proper number of columns for simple table', () => {
  591. setData( model, modelTable( [
  592. [ '00', '01' ],
  593. [ '10', '11' ]
  594. ] ) );
  595. expect( tableUtils.getRows( root.getNodeByPath( [ 0 ] ) ) ).to.equal( 2 );
  596. } );
  597. it( 'should return proper number of columns for a table with header', () => {
  598. setData( model, modelTable( [
  599. [ '00', '01' ],
  600. [ '10', '11' ]
  601. ], { headingRows: 1 } ) );
  602. expect( tableUtils.getRows( root.getNodeByPath( [ 0 ] ) ) ).to.equal( 2 );
  603. } );
  604. it( 'should return proper number of columns for rowspan table', () => {
  605. setData( model, modelTable( [
  606. [ '00', '01' ],
  607. [ { rowspan: 2, contents: '10' }, '11' ],
  608. [ '21' ]
  609. ] ) );
  610. expect( tableUtils.getRows( root.getNodeByPath( [ 0 ] ) ) ).to.equal( 3 );
  611. } );
  612. } );
  613. describe( 'removeRows()', () => {
  614. describe( 'single row', () => {
  615. it( 'should remove a given row from a table start', () => {
  616. setData( model, modelTable( [
  617. [ '00', '01' ],
  618. [ '10', '11' ],
  619. [ '20', '21' ]
  620. ] ) );
  621. tableUtils.removeRows( root.getChild( 0 ), { at: 0 } );
  622. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  623. [ '10', '11' ],
  624. [ '20', '21' ]
  625. ] ) );
  626. } );
  627. it( 'should remove last row', () => {
  628. setData( model, modelTable( [
  629. [ '00', '01' ],
  630. [ '10', '11' ]
  631. ] ) );
  632. tableUtils.removeRows( root.getChild( 0 ), { at: 1 } );
  633. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  634. [ '00', '01' ]
  635. ] ) );
  636. } );
  637. it( 'should change heading rows if removing a heading row', () => {
  638. setData( model, modelTable( [
  639. [ '00', '01' ],
  640. [ '10', '11' ],
  641. [ '20', '21' ]
  642. ], { headingRows: 2 } ) );
  643. tableUtils.removeRows( root.getChild( 0 ), { at: 1 } );
  644. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  645. [ '00', '01' ],
  646. [ '20', '21' ]
  647. ], { headingRows: 1 } ) );
  648. } );
  649. it( 'should decrease rowspan of table cells from previous rows', () => {
  650. setData( model, modelTable( [
  651. [ { rowspan: 4, contents: '00' }, { rowspan: 3, contents: '01' }, { rowspan: 2, contents: '02' }, '03', '04' ],
  652. [ { rowspan: 2, contents: '13' }, '14' ],
  653. [ '22', '23', '24' ],
  654. [ '30', '31', '32', '33', '34' ]
  655. ] ) );
  656. tableUtils.removeRows( root.getChild( 0 ), { at: 2 } );
  657. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  658. [ { rowspan: 3, contents: '00' }, { rowspan: 2, contents: '01' }, { rowspan: 2, contents: '02' }, '03', '04' ],
  659. [ '13', '14' ],
  660. [ '30', '31', '32', '33', '34' ]
  661. ] ) );
  662. } );
  663. it( 'should move rowspaned cells to row below removing it\'s row', () => {
  664. setData( model, modelTable( [
  665. [ { rowspan: 3, contents: '00' }, { rowspan: 2, contents: '01' }, '02' ],
  666. [ '12' ],
  667. [ '21', '22' ],
  668. [ '30', '31', '32' ]
  669. ] ) );
  670. tableUtils.removeRows( root.getChild( 0 ), { at: 0 } );
  671. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  672. [ { rowspan: 2, contents: '00' }, '01', '12' ],
  673. [ '21', '22' ],
  674. [ '30', '31', '32' ]
  675. ] ) );
  676. } );
  677. } );
  678. describe( 'many rows', () => {
  679. it( 'should properly remove middle rows', () => {
  680. setData( model, modelTable( [
  681. [ '00', '01' ],
  682. [ '10', '11' ],
  683. [ '20', '21' ],
  684. [ '30', '31' ]
  685. ] ) );
  686. tableUtils.removeRows( root.getChild( 0 ), { at: 1, rows: 2 } );
  687. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  688. [ '00', '01' ],
  689. [ '30', '31' ]
  690. ] ) );
  691. } );
  692. it( 'should properly remove tailing rows', () => {
  693. setData( model, modelTable( [
  694. [ '00', '01' ],
  695. [ '10', '11' ],
  696. [ '20', '21' ],
  697. [ '30', '31' ]
  698. ] ) );
  699. tableUtils.removeRows( root.getChild( 0 ), { at: 2, rows: 2 } );
  700. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  701. [ '00', '01' ],
  702. [ '10', '11' ]
  703. ] ) );
  704. } );
  705. it( 'should properly remove beginning rows', () => {
  706. setData( model, modelTable( [
  707. [ '00', '01' ],
  708. [ '10', '11' ],
  709. [ '20', '21' ],
  710. [ '30', '31' ]
  711. ] ) );
  712. tableUtils.removeRows( root.getChild( 0 ), { at: 0, rows: 2 } );
  713. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  714. [ '20', '21' ],
  715. [ '30', '31' ]
  716. ] ) );
  717. } );
  718. it( 'should support removing multiple headings (removed rows in heading section)', () => {
  719. setData( model, modelTable( [
  720. [ '00', '01' ],
  721. [ '10', '11' ],
  722. [ '20', '21' ],
  723. [ '30', '31' ]
  724. ], { headingRows: 3 } ) );
  725. tableUtils.removeRows( root.getChild( 0 ), { at: 0, rows: 2 } );
  726. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  727. [ '20', '21' ],
  728. [ '30', '31' ]
  729. ], { headingRows: 1 } ) );
  730. } );
  731. it( 'should support removing multiple headings (removed rows in heading and body section)', () => {
  732. setData( model, modelTable( [
  733. [ '00', '01' ],
  734. [ '10', '11' ],
  735. [ '20', '21' ],
  736. [ '30', '31' ],
  737. [ '40', '41' ]
  738. ], { headingRows: 3 } ) );
  739. tableUtils.removeRows( root.getChild( 0 ), { at: 1, rows: 3 } );
  740. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  741. [ '00', '01' ],
  742. [ '40', '41' ]
  743. ], { headingRows: 1 } ) );
  744. } );
  745. it( 'should support removing mixed heading and cell rows', () => {
  746. setData( model, modelTable( [
  747. [ '00', '01' ],
  748. [ '10', '11' ],
  749. [ '20', '21' ]
  750. ], { headingRows: 1 } ) );
  751. tableUtils.removeRows( root.getChild( 0 ), { at: 0, rows: 2 } );
  752. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  753. [ '20', '21' ]
  754. ] ) );
  755. } );
  756. it( 'should properly calculate truncated rowspans', () => {
  757. setData( model, modelTable( [
  758. [ '00', { contents: '01', rowspan: 3 } ],
  759. [ '10' ],
  760. [ '20' ]
  761. ] ) );
  762. tableUtils.removeRows( root.getChild( 0 ), { at: 0, rows: 2 } );
  763. assertEqualMarkup( getData( model, { withoutSelection: true } ), modelTable( [
  764. [ '20', '01' ]
  765. ] ) );
  766. } );
  767. it( 'should create one undo step (1 batch)', () => {
  768. setData( model, modelTable( [
  769. [ '00', '01' ],
  770. [ '10', '11' ],
  771. [ '20', '21' ],
  772. [ '30', '31' ]
  773. ], { headingRows: 3 } ) );
  774. const createdBatches = new Set();
  775. model.on( 'applyOperation', ( evt, args ) => {
  776. const operation = args[ 0 ];
  777. createdBatches.add( operation.batch );
  778. } );
  779. tableUtils.removeRows( root.getChild( 0 ), { at: 0, rows: 2 } );
  780. expect( createdBatches.size ).to.equal( 1 );
  781. } );
  782. } );
  783. } );
  784. } );