downcast.js 24 KB

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