downcast.js 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718
  1. /**
  2. * @license Copyright (c) 2003-2018, CKSource - Frederico Knabben. All rights reserved.
  3. * For licensing, see LICENSE.md.
  4. */
  5. import VirtualTestEditor from '@ckeditor/ckeditor5-core/tests/_utils/virtualtesteditor';
  6. import { getData as getViewData } from '@ckeditor/ckeditor5-engine/src/dev-utils/view';
  7. import { setData as setModelData } from '@ckeditor/ckeditor5-engine/src/dev-utils/model';
  8. import {
  9. downcastAttributeChange,
  10. downcastInsertCell,
  11. downcastInsertRow,
  12. downcastInsertTable
  13. } from '../../src/converters/downcast';
  14. import { formatModelTable, formattedViewTable, modelTable, viewTable } from '../_utils/utils';
  15. describe( 'downcast converters', () => {
  16. let editor, model, doc, root, viewDocument;
  17. beforeEach( () => {
  18. return VirtualTestEditor.create()
  19. .then( newEditor => {
  20. editor = newEditor;
  21. model = editor.model;
  22. doc = model.document;
  23. root = doc.getRoot( 'main' );
  24. viewDocument = editor.editing.view;
  25. const conversion = editor.conversion;
  26. const schema = model.schema;
  27. schema.register( 'table', {
  28. allowWhere: '$block',
  29. allowAttributes: [ 'headingRows', 'headingColumns' ],
  30. isBlock: true,
  31. isObject: true
  32. } );
  33. schema.register( 'tableRow', {
  34. allowIn: 'table',
  35. allowAttributes: [],
  36. isBlock: true,
  37. isLimit: true
  38. } );
  39. schema.register( 'tableCell', {
  40. allowIn: 'tableRow',
  41. allowContentOf: '$block',
  42. allowAttributes: [ 'colspan', 'rowspan' ],
  43. isBlock: true,
  44. isLimit: true
  45. } );
  46. conversion.for( 'downcast' ).add( downcastInsertTable() );
  47. conversion.attributeToAttribute( { model: 'colspan', view: 'colspan' } );
  48. conversion.attributeToAttribute( { model: 'rowspan', view: 'rowspan' } );
  49. // Insert conversion
  50. conversion.for( 'downcast' ).add( downcastInsertRow() );
  51. conversion.for( 'downcast' ).add( downcastInsertCell() );
  52. conversion.for( 'downcast' ).add( downcastAttributeChange( 'headingRows' ) );
  53. conversion.for( 'downcast' ).add( downcastAttributeChange( 'headingColumns' ) );
  54. } );
  55. } );
  56. describe( 'downcastInsertTable()', () => {
  57. it( 'should create table with tbody', () => {
  58. setModelData( model,
  59. '<table>' +
  60. '<tableRow><tableCell></tableCell></tableRow>' +
  61. '</table>'
  62. );
  63. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  64. '<table>' +
  65. '<tbody>' +
  66. '<tr><td></td></tr>' +
  67. '</tbody>' +
  68. '</table>'
  69. );
  70. } );
  71. it( 'should create table with tbody and thead', () => {
  72. setModelData( model,
  73. '<table headingRows="1">' +
  74. '<tableRow><tableCell>1</tableCell></tableRow>' +
  75. '<tableRow><tableCell>2</tableCell></tableRow>' +
  76. '</table>'
  77. );
  78. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  79. '<table>' +
  80. '<thead>' +
  81. '<tr><th>1</th></tr>' +
  82. '</thead>' +
  83. '<tbody>' +
  84. '<tr><td>2</td></tr>' +
  85. '</tbody>' +
  86. '</table>'
  87. );
  88. } );
  89. it( 'should create table with thead', () => {
  90. setModelData( model,
  91. '<table headingRows="2">' +
  92. '<tableRow><tableCell>1</tableCell></tableRow>' +
  93. '<tableRow><tableCell>2</tableCell></tableRow>' +
  94. '</table>'
  95. );
  96. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  97. '<table>' +
  98. '<thead>' +
  99. '<tr><th>1</th></tr>' +
  100. '<tr><th>2</th></tr>' +
  101. '</thead>' +
  102. '</table>'
  103. );
  104. } );
  105. it( 'should create table with heading columns and rows', () => {
  106. setModelData( model,
  107. '<table headingColumns="3" headingRows="1">' +
  108. '<tableRow>' +
  109. '<tableCell>11</tableCell><tableCell>12</tableCell><tableCell>13</tableCell><tableCell>14</tableCell>' +
  110. '</tableRow>' +
  111. '<tableRow>' +
  112. '<tableCell>21</tableCell><tableCell>22</tableCell><tableCell>23</tableCell><tableCell>24</tableCell>' +
  113. '</tableRow>' +
  114. '</table>'
  115. );
  116. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  117. '<table>' +
  118. '<thead>' +
  119. '<tr><th>11</th><th>12</th><th>13</th><th>14</th></tr>' +
  120. '</thead>' +
  121. '<tbody>' +
  122. '<tr><th>21</th><th>22</th><th>23</th><td>24</td></tr>' +
  123. '</tbody>' +
  124. '</table>'
  125. );
  126. } );
  127. it( 'should be possible to overwrite', () => {
  128. editor.conversion.elementToElement( { model: 'tableRow', view: 'tr', priority: 'high' } );
  129. editor.conversion.elementToElement( { model: 'tableCell', view: 'td', priority: 'high' } );
  130. editor.conversion.for( 'downcast' ).add( dispatcher => {
  131. dispatcher.on( 'insert:table', ( evt, data, conversionApi ) => {
  132. conversionApi.consumable.consume( data.item, 'insert' );
  133. const tableElement = conversionApi.writer.createContainerElement( 'table', { foo: 'bar' } );
  134. const viewPosition = conversionApi.mapper.toViewPosition( data.range.start );
  135. conversionApi.mapper.bindElements( data.item, tableElement );
  136. conversionApi.writer.insert( viewPosition, tableElement );
  137. }, { priority: 'high' } );
  138. } );
  139. setModelData( model,
  140. '<table>' +
  141. '<tableRow><tableCell></tableCell></tableRow>' +
  142. '</table>'
  143. );
  144. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  145. '<table foo="bar">' +
  146. '<tr><td></td></tr>' +
  147. '</table>'
  148. );
  149. } );
  150. describe( 'headingColumns attribute', () => {
  151. it( 'should mark heading columns table cells', () => {
  152. setModelData( model,
  153. '<table headingColumns="2">' +
  154. '<tableRow><tableCell>11</tableCell><tableCell>12</tableCell><tableCell>13</tableCell></tableRow>' +
  155. '<tableRow><tableCell>21</tableCell><tableCell>22</tableCell><tableCell>23</tableCell></tableRow>' +
  156. '</table>'
  157. );
  158. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  159. '<table>' +
  160. '<tbody>' +
  161. '<tr><th>11</th><th>12</th><td>13</td></tr>' +
  162. '<tr><th>21</th><th>22</th><td>23</td></tr>' +
  163. '</tbody>' +
  164. '</table>'
  165. );
  166. } );
  167. it( 'should mark heading columns table cells when one has colspan attribute', () => {
  168. setModelData( model,
  169. '<table headingColumns="3">' +
  170. '<tableRow>' +
  171. '<tableCell>11</tableCell><tableCell>12</tableCell><tableCell>13</tableCell><tableCell>14</tableCell>' +
  172. '</tableRow>' +
  173. '<tableRow><tableCell colspan="2">21</tableCell><tableCell>23</tableCell><tableCell>24</tableCell></tableRow>' +
  174. '</table>'
  175. );
  176. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  177. '<table>' +
  178. '<tbody>' +
  179. '<tr><th>11</th><th>12</th><th>13</th><td>14</td></tr>' +
  180. '<tr><th colspan="2">21</th><th>23</th><td>24</td></tr>' +
  181. '</tbody>' +
  182. '</table>'
  183. );
  184. } );
  185. it( 'should work with colspan and rowspan attributes on table cells', () => {
  186. // The table in this test looks like a table below:
  187. //
  188. // Row headings | Normal cells
  189. // |
  190. // +----+----+----+----+
  191. // | 11 | 12 | 13 | 14 |
  192. // | +----+ +----+
  193. // | | 22 | | 24 |
  194. // |----+----+ +----+
  195. // | 31 | | 34 |
  196. // | +----+----+
  197. // | | 43 | 44 |
  198. // +----+----+----+----+
  199. setModelData( model,
  200. '<table headingColumns="3">' +
  201. '<tableRow>' +
  202. '<tableCell rowspan="2">11</tableCell>' +
  203. '<tableCell>12</tableCell>' +
  204. '<tableCell rowspan="3">13</tableCell>' +
  205. '<tableCell>14</tableCell>' +
  206. '</tableRow>' +
  207. '<tableRow><tableCell>22</tableCell><tableCell>24</tableCell></tableRow>' +
  208. '<tableRow><tableCell colspan="2" rowspan="2">31</tableCell><tableCell>34</tableCell></tableRow>' +
  209. '<tableRow><tableCell>43</tableCell><tableCell>44</tableCell></tableRow>' +
  210. '</table>'
  211. );
  212. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  213. '<table>' +
  214. '<tbody>' +
  215. '<tr><th rowspan="2">11</th><th>12</th><th rowspan="3">13</th><td>14</td></tr>' +
  216. '<tr><th>22</th><td>24</td></tr>' +
  217. '<tr><th colspan="2" rowspan="2">31</th><td>34</td></tr>' +
  218. '<tr><th>43</th><td>44</td></tr>' +
  219. '</tbody>' +
  220. '</table>'
  221. );
  222. } );
  223. } );
  224. } );
  225. describe( 'downcastInsertRow()', () => {
  226. it( 'should react to changed rows', () => {
  227. setModelData( model, modelTable( [
  228. [ '11', '12' ]
  229. ] ) );
  230. const table = root.getChild( 0 );
  231. model.change( writer => {
  232. const row = writer.createElement( 'tableRow' );
  233. writer.insert( row, table, 1 );
  234. writer.insertElement( 'tableCell', row, 'end' );
  235. writer.insertElement( 'tableCell', row, 'end' );
  236. } );
  237. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  238. [ '11', '12' ],
  239. [ '', '' ]
  240. ] ) );
  241. } );
  242. it( 'should insert row on proper index', () => {
  243. setModelData( model, modelTable( [
  244. [ '11', '12' ],
  245. [ '21', '22' ],
  246. [ '31', '32' ]
  247. ] ) );
  248. const table = root.getChild( 0 );
  249. model.change( writer => {
  250. const row = writer.createElement( 'tableRow' );
  251. writer.insert( row, table, 1 );
  252. writer.insertElement( 'tableCell', row, 'end' );
  253. writer.insertElement( 'tableCell', row, 'end' );
  254. } );
  255. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  256. [ '11', '12' ],
  257. [ '', '' ],
  258. [ '21', '22' ],
  259. [ '31', '32' ]
  260. ] ) );
  261. } );
  262. it( 'should insert row on proper index when table has heading rows defined - insert in body', () => {
  263. setModelData( model, modelTable( [
  264. [ '11', '12' ],
  265. [ '21', '22' ],
  266. [ '31', '32' ]
  267. ], { headingRows: 1 } ) );
  268. const table = root.getChild( 0 );
  269. model.change( writer => {
  270. const row = writer.createElement( 'tableRow' );
  271. writer.insert( row, table, 1 );
  272. writer.insertElement( 'tableCell', row, 'end' );
  273. writer.insertElement( 'tableCell', row, 'end' );
  274. } );
  275. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  276. [ '11', '12' ],
  277. [ '', '' ],
  278. [ '21', '22' ],
  279. [ '31', '32' ]
  280. ], { headingRows: 1 } ) );
  281. } );
  282. it( 'should insert row on proper index when table has heading rows defined - insert in heading', () => {
  283. setModelData( model, modelTable( [
  284. [ '11', '12' ],
  285. [ '21', '22' ],
  286. [ '31', '32' ]
  287. ], { headingRows: 2 } ) );
  288. const table = root.getChild( 0 );
  289. model.change( writer => {
  290. const row = writer.createElement( 'tableRow' );
  291. writer.insert( row, table, 1 );
  292. writer.insertElement( 'tableCell', row, 'end' );
  293. writer.insertElement( 'tableCell', row, 'end' );
  294. } );
  295. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  296. [ '11', '12' ],
  297. [ '', '' ],
  298. [ '21', '22' ],
  299. [ '31', '32' ]
  300. ], { headingRows: 3 } ) );
  301. } );
  302. it( 'should react to changed rows when previous rows\' cells has rowspans', () => {
  303. setModelData( model, modelTable( [
  304. [ { rowspan: 3, contents: '11' }, '12' ],
  305. [ '22' ]
  306. ] ) );
  307. const table = root.getChild( 0 );
  308. model.change( writer => {
  309. const row = writer.createElement( 'tableRow' );
  310. writer.insert( row, table, 2 );
  311. writer.insertElement( 'tableCell', row, 'end' );
  312. } );
  313. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  314. [ { rowspan: 3, contents: '11' }, '12' ],
  315. [ '22' ],
  316. [ '' ]
  317. ] ) );
  318. } );
  319. it( 'should properly create row headings', () => {
  320. setModelData( model, modelTable( [
  321. [ { rowspan: 3, contents: '11' }, '12' ],
  322. [ '22' ]
  323. ], { headingColumns: 1 } ) );
  324. const table = root.getChild( 0 );
  325. model.change( writer => {
  326. const firstRow = writer.createElement( 'tableRow' );
  327. writer.insert( firstRow, table, 2 );
  328. writer.insert( writer.createElement( 'tableCell' ), firstRow, 'end' );
  329. const secondRow = writer.createElement( 'tableRow' );
  330. writer.insert( secondRow, table, 3 );
  331. writer.insert( writer.createElement( 'tableCell' ), secondRow, 'end' );
  332. writer.insert( writer.createElement( 'tableCell' ), secondRow, 'end' );
  333. } );
  334. expect( formatModelTable( getViewData( viewDocument, { withoutSelection: true } ) ) ).to.equal( formattedViewTable( [
  335. [ { rowspan: 3, contents: '11', isHeading: true }, '12' ],
  336. [ '22' ],
  337. [ '' ],
  338. [ { contents: '', isHeading: true }, '' ]
  339. ] ) );
  340. } );
  341. } );
  342. describe( 'downcastInsertCell()', () => {
  343. it( 'should add tableCell on proper index in tr', () => {
  344. setModelData( model, modelTable( [
  345. [ '11', '12' ]
  346. ] ) );
  347. const table = root.getChild( 0 );
  348. model.change( writer => {
  349. const row = table.getChild( 0 );
  350. writer.insertElement( 'tableCell', row, 1 );
  351. } );
  352. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  353. [ '11', '', '12' ]
  354. ] ) );
  355. } );
  356. it( 'should add tableCell on proper index in tr when previous have colspans', () => {
  357. setModelData( model, modelTable( [
  358. [ { colspan: 2, contents: '11' }, '13' ]
  359. ] ) );
  360. const table = root.getChild( 0 );
  361. model.change( writer => {
  362. const row = table.getChild( 0 );
  363. writer.insertElement( 'tableCell', row, 1 );
  364. } );
  365. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  366. [ { colspan: 2, contents: '11' }, '', '13' ]
  367. ] ) );
  368. } );
  369. it( 'should add tableCell on proper index in tr when previous row have rowspans', () => {
  370. setModelData( model, modelTable( [
  371. [ { rowspan: 2, contents: '11' }, '13' ],
  372. [ '22', '23' ]
  373. ] ) );
  374. const table = root.getChild( 0 );
  375. model.change( writer => {
  376. writer.insertElement( 'tableCell', table.getChild( 0 ), 1 );
  377. writer.insertElement( 'tableCell', table.getChild( 1 ), 0 );
  378. } );
  379. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  380. [ { rowspan: 2, contents: '11' }, '', '13' ],
  381. [ '', '22', '23' ]
  382. ] ) );
  383. } );
  384. it( 'split cell simulation - simple', () => {
  385. setModelData( model, modelTable( [
  386. [ '11', '12' ],
  387. [ '21', '22' ]
  388. ] ) );
  389. const table = root.getChild( 0 );
  390. model.change( writer => {
  391. const firstRow = table.getChild( 0 );
  392. const secondRow = table.getChild( 1 );
  393. writer.insertElement( 'tableCell', firstRow, 1 );
  394. writer.setAttribute( 'colspan', 2, secondRow.getChild( 0 ) );
  395. } );
  396. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  397. [ '11', '', '12' ],
  398. [ { colspan: 2, contents: '21' }, '22' ]
  399. ] ) );
  400. } );
  401. it( 'merge simulation - simple', () => {
  402. setModelData( model, modelTable( [
  403. [ '11', '12' ],
  404. [ '21', '22' ]
  405. ] ) );
  406. const table = root.getChild( 0 );
  407. model.change( writer => {
  408. const firstRow = table.getChild( 0 );
  409. writer.setAttribute( 'colspan', 2, firstRow.getChild( 0 ) );
  410. writer.remove( firstRow.getChild( 1 ) );
  411. } );
  412. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  413. [ { colspan: 2, contents: '11' } ],
  414. [ '21', '22' ]
  415. ] ) );
  416. } );
  417. } );
  418. describe( 'downcastAttributeChange()', () => {
  419. it( 'should work for adding heading rows', () => {
  420. setModelData( model, modelTable( [
  421. [ '11', '12' ],
  422. [ '21', '22' ],
  423. [ '31', '32' ]
  424. ] ) );
  425. const table = root.getChild( 0 );
  426. model.change( writer => {
  427. writer.setAttribute( 'headingRows', 2, table );
  428. } );
  429. expect( formatModelTable( getViewData( viewDocument, { withoutSelection: true } ) ) ).to.equal( formattedViewTable( [
  430. [ '11', '12' ],
  431. [ '21', '22' ],
  432. [ '31', '32' ]
  433. ], { headingRows: 2 } ) );
  434. } );
  435. it( 'should work for changing number of heading rows to a bigger number', () => {
  436. setModelData( model, modelTable( [
  437. [ '11', '12' ],
  438. [ '21', '22' ],
  439. [ '31', '32' ]
  440. ], { headingRows: 1 } ) );
  441. const table = root.getChild( 0 );
  442. model.change( writer => {
  443. writer.setAttribute( 'headingRows', 2, table );
  444. } );
  445. expect( formatModelTable( getViewData( viewDocument, { withoutSelection: true } ) ) ).to.equal( formattedViewTable( [
  446. [ '11', '12' ],
  447. [ '21', '22' ],
  448. [ '31', '32' ]
  449. ], { headingRows: 2 } ) );
  450. } );
  451. it( 'should work for changing number of heading rows to a smaller number', () => {
  452. setModelData( model, modelTable( [
  453. [ '11', '12' ],
  454. [ '21', '22' ],
  455. [ '31', '32' ],
  456. [ '41', '42' ]
  457. ], { headingRows: 3 } ) );
  458. const table = root.getChild( 0 );
  459. model.change( writer => {
  460. writer.setAttribute( 'headingRows', 2, table );
  461. } );
  462. expect( formatModelTable( getViewData( viewDocument, { withoutSelection: true } ) ) ).to.equal( formattedViewTable( [
  463. [ '11', '12' ],
  464. [ '21', '22' ],
  465. [ '31', '32' ],
  466. [ '41', '42' ]
  467. ], { headingRows: 2 } ) );
  468. } );
  469. it( 'should work for removing heading rows', () => {
  470. setModelData( model, modelTable( [
  471. [ '11', '12' ],
  472. [ '21', '22' ]
  473. ], { headingRows: 2 } ) );
  474. const table = root.getChild( 0 );
  475. model.change( writer => {
  476. writer.removeAttribute( 'headingRows', table );
  477. } );
  478. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  479. [ '11', '12' ],
  480. [ '21', '22' ]
  481. ] ) );
  482. } );
  483. it( 'should work for making heading rows without tbody', () => {
  484. setModelData( model, modelTable( [
  485. [ '11', '12' ],
  486. [ '21', '22' ]
  487. ] ) );
  488. const table = root.getChild( 0 );
  489. model.change( writer => {
  490. writer.setAttribute( 'headingRows', 2, table );
  491. } );
  492. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  493. [ '11', '12' ],
  494. [ '21', '22' ]
  495. ], { headingRows: 2 } ) );
  496. } );
  497. it( 'should work for adding heading columns', () => {
  498. setModelData( model, modelTable( [
  499. [ '11', '12' ],
  500. [ '21', '22' ]
  501. ] ) );
  502. const table = root.getChild( 0 );
  503. model.change( writer => {
  504. writer.setAttribute( 'headingColumns', 1, table );
  505. } );
  506. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  507. [ { isHeading: true, contents: '11' }, '12' ],
  508. [ { isHeading: true, contents: '21' }, '22' ]
  509. ] ) );
  510. } );
  511. it( 'should work for changing heading columns to a bigger number', () => {
  512. setModelData( model, modelTable( [
  513. [ '11', '12', '13', '14' ],
  514. [ '21', '22', '23', '24' ]
  515. ], { headingColumns: 1 } ) );
  516. const table = root.getChild( 0 );
  517. model.change( writer => {
  518. writer.setAttribute( 'headingColumns', 3, table );
  519. } );
  520. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  521. [ { isHeading: true, contents: '11' }, { isHeading: true, contents: '12' }, { isHeading: true, contents: '13' }, '14' ],
  522. [ { isHeading: true, contents: '21' }, { isHeading: true, contents: '22' }, { isHeading: true, contents: '23' }, '24' ]
  523. ] ) );
  524. } );
  525. it( 'should work for removing heading columns', () => {
  526. setModelData( model, modelTable( [
  527. [ '11', '12' ],
  528. [ '21', '22' ]
  529. ], { headingColumns: 1 } ) );
  530. const table = root.getChild( 0 );
  531. model.change( writer => {
  532. writer.removeAttribute( 'headingColumns', table );
  533. } );
  534. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal( viewTable( [
  535. [ '11', '12' ],
  536. [ '21', '22' ]
  537. ] ) );
  538. } );
  539. it( 'should be possible to overwrite', () => {
  540. editor.conversion.attributeToAttribute( { model: 'headingColumns', view: 'headingColumns', priority: 'high' } );
  541. setModelData( model, modelTable( [ [ '11' ] ] ) );
  542. const table = root.getChild( 0 );
  543. model.change( writer => {
  544. writer.setAttribute( 'headingColumns', 1, table );
  545. } );
  546. expect( getViewData( viewDocument, { withoutSelection: true } ) ).to.equal(
  547. '<table headingColumns="1"><tbody><tr><td>11</td></tr></tbody></table>'
  548. );
  549. } );
  550. it( 'should work with adding table cells', () => {
  551. setModelData( model, modelTable( [
  552. [ { rowspan: 2, contents: '11' }, '12', '13', '14' ],
  553. [ '22', '23', '24' ],
  554. [ { colspan: 2, contents: '31' }, '33', '34' ]
  555. ], { headingColumns: 2 } ) );
  556. const table = root.getChild( 0 );
  557. model.change( writer => {
  558. // Inserting column in heading columns so update table's attribute also
  559. writer.setAttribute( 'headingColumns', 3, table );
  560. writer.insertElement( 'tableCell', table.getChild( 0 ), 2 );
  561. writer.insertElement( 'tableCell', table.getChild( 1 ), 1 );
  562. writer.insertElement( 'tableCell', table.getChild( 2 ), 1 );
  563. } );
  564. expect( formatModelTable( getViewData( viewDocument, { withoutSelection: true } ) ) ).to.equal( formattedViewTable( [
  565. [
  566. { isHeading: true, rowspan: 2, contents: '11' },
  567. { isHeading: true, contents: '12' },
  568. { isHeading: true, contents: '' },
  569. '13',
  570. '14'
  571. ],
  572. [
  573. { isHeading: true, contents: '22' },
  574. { isHeading: true, contents: '' },
  575. '23',
  576. '24'
  577. ],
  578. [
  579. { isHeading: true, colspan: 2, contents: '31' },
  580. { isHeading: true, contents: '' },
  581. '33',
  582. '34'
  583. ]
  584. ] ) );
  585. } );
  586. } );
  587. } );