tree.spec.js 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802
  1. import { createVue, destroyVM } from '../util';
  2. const DELAY = 10;
  3. describe('Tree', () => {
  4. let vm;
  5. afterEach(() => {
  6. destroyVM(vm);
  7. });
  8. const getTreeVm = (props, options) => {
  9. return createVue(Object.assign({
  10. template: `
  11. <el-tree ref="tree" :data="data" ${ props }></el-tree>
  12. `,
  13. data() {
  14. return {
  15. defaultExpandedKeys: [],
  16. defaultCheckedKeys: [],
  17. clickedNode: null,
  18. count: 1,
  19. data: [{
  20. id: 1,
  21. label: '一级 1',
  22. children: [{
  23. id: 11,
  24. label: '二级 1-1',
  25. children: [{
  26. id: 111,
  27. label: '三级 1-1'
  28. }]
  29. }]
  30. }, {
  31. id: 2,
  32. label: '一级 2',
  33. children: [{
  34. id: 21,
  35. label: '二级 2-1'
  36. }, {
  37. id: 22,
  38. label: '二级 2-2'
  39. }]
  40. }, {
  41. id: 3,
  42. label: '一级 3',
  43. children: [{
  44. id: 31,
  45. label: '二级 3-1'
  46. }, {
  47. id: 32,
  48. label: '二级 3-2'
  49. }]
  50. }],
  51. defaultProps: {
  52. children: 'children',
  53. label: 'label'
  54. }
  55. };
  56. }
  57. }, options), true);
  58. };
  59. const getDisableTreeVm = (props, options) => {
  60. return createVue(Object.assign({
  61. template: `
  62. <el-tree ref="tree" :data="data" ${ props }></el-tree>
  63. `,
  64. data() {
  65. return {
  66. defaultExpandedKeys: [],
  67. defaultCheckedKeys: [],
  68. clickedNode: null,
  69. count: 1,
  70. data: [{
  71. id: 1,
  72. label: '一级 1',
  73. children: [{
  74. id: 11,
  75. label: '二级 1-1',
  76. children: [{
  77. id: 111,
  78. label: '三级 1-1',
  79. disabled: true
  80. }]
  81. }]
  82. }, {
  83. id: 2,
  84. label: '一级 2',
  85. children: [{
  86. id: 21,
  87. label: '二级 2-1'
  88. }, {
  89. id: 22,
  90. label: '二级 2-2'
  91. }]
  92. }, {
  93. id: 3,
  94. label: '一级 3',
  95. children: [{
  96. id: 31,
  97. label: '二级 3-1'
  98. }, {
  99. id: 32,
  100. label: '二级 3-2'
  101. }]
  102. }],
  103. defaultProps: {
  104. children: 'children',
  105. label: 'label',
  106. disabled: 'disabled'
  107. }
  108. };
  109. }
  110. }, options), true);
  111. };
  112. const ALL_NODE_COUNT = 9;
  113. it('create', () => {
  114. vm = getTreeVm(':props="defaultProps" default-expand-all');
  115. expect(document.querySelector('.el-tree')).to.exist;
  116. expect(document.querySelectorAll('.el-tree > .el-tree-node').length).to.equal(3);
  117. expect(document.querySelectorAll('.el-tree .el-tree-node').length).to.equal(ALL_NODE_COUNT);
  118. vm.data[1].children = [{ label: '二级 2-1' }];
  119. const tree = vm.$children[0];
  120. expect(tree.children).to.deep.equal(vm.data);
  121. });
  122. it('click node', done => {
  123. vm = getTreeVm(':props="defaultProps" @node-click="handleNodeClick"', {
  124. methods: {
  125. handleNodeClick(data) {
  126. this.clickedNode = data;
  127. }
  128. }
  129. });
  130. const firstNode = vm.$el.querySelector('.el-tree-node__content');
  131. firstNode.click();
  132. expect(vm.clickedNode.label).to.equal('一级 1');
  133. setTimeout(() => {
  134. expect(vm.$el.querySelector('.el-tree-node').classList.contains('is-expanded')).to.true;
  135. firstNode.click();
  136. setTimeout(() => {
  137. expect(vm.$el.querySelector('.el-tree-node').classList.contains('is-expanded')).to.false;
  138. done();
  139. }, DELAY);
  140. }, DELAY);
  141. });
  142. it('current change', done => {
  143. vm = getTreeVm(':props="defaultProps" @current-change="handleCurrentChange"', {
  144. methods: {
  145. handleCurrentChange(data) {
  146. this.currentNode = data;
  147. }
  148. }
  149. });
  150. const firstNode = vm.$el.querySelector('.el-tree-node__content');
  151. firstNode.click();
  152. expect(vm.currentNode.label).to.equal('一级 1');
  153. done();
  154. });
  155. it('emptyText', (done) => {
  156. vm = getTreeVm(':props="defaultProps"');
  157. vm.data = [];
  158. vm.$nextTick(() => {
  159. expect(vm.$el.querySelectorAll('.el-tree__empty-block').length).to.equal(1);
  160. done();
  161. });
  162. });
  163. it('highlight current', done => {
  164. vm = getTreeVm(':props="defaultProps" highlight-current');
  165. const firstNode = document.querySelector('.el-tree-node');
  166. firstNode.click();
  167. vm.$nextTick(() => {
  168. expect(firstNode.className.indexOf('is-current') !== -1);
  169. done();
  170. });
  171. });
  172. it('expandOnNodeClick', done => {
  173. vm = getTreeVm(':props="defaultProps" :expand-on-node-click="false"');
  174. const firstNode = document.querySelector('.el-tree-node');
  175. firstNode.click();
  176. vm.$nextTick(() => {
  177. expect(firstNode.className.indexOf('is-expanded') === -1);
  178. done();
  179. });
  180. });
  181. it('current-node-key', done => {
  182. vm = getTreeVm(':props="defaultProps" :current-node-key="1"');
  183. const firstNode = document.querySelector('.el-tree-node');
  184. firstNode.click();
  185. vm.$nextTick(() => {
  186. expect(firstNode.classList.contains('is-current')).to.true;
  187. done();
  188. });
  189. });
  190. it('defaultExpandAll', () => {
  191. vm = getTreeVm(':props="defaultProps" default-expand-all');
  192. expect(vm.$el.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(ALL_NODE_COUNT);
  193. });
  194. it('defaultExpandedKeys', () => {
  195. vm = getTreeVm(':props="defaultProps" :default-expanded-keys="defaultExpandedKeys" node-key="id"', {
  196. created() {
  197. this.defaultExpandedKeys = [1, 3];
  198. }
  199. });
  200. expect(vm.$el.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(2);
  201. });
  202. it('defaultExpandedKeys set', (done) => {
  203. vm = getTreeVm(':props="defaultProps" :default-expanded-keys="defaultExpandedKeys" node-key="id"', {
  204. created() {
  205. this.defaultExpandedKeys = [1, 3];
  206. }
  207. });
  208. expect(vm.$el.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(2);
  209. vm.defaultExpandedKeys = [2];
  210. vm.data = JSON.parse(JSON.stringify(vm.data));
  211. setTimeout(() => {
  212. expect(vm.$el.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(1);
  213. done();
  214. }, 50);
  215. });
  216. it('filter-node-method', (done) => {
  217. vm = getTreeVm(':props="defaultProps" :filter-node-method="filterNode"', {
  218. methods: {
  219. filterNode(value, data) {
  220. if (!value) return true;
  221. return data.label.indexOf(value) !== -1;
  222. }
  223. }
  224. });
  225. const tree = vm.$refs.tree;
  226. tree.filter('2-1');
  227. setTimeout(() => {
  228. expect(tree.$el.querySelectorAll('.el-tree-node.is-hidden').length).to.equal(3);
  229. done();
  230. }, 100);
  231. });
  232. it('autoExpandParent = true', () => {
  233. vm = getTreeVm(':props="defaultProps" :default-expanded-keys="defaultExpandedKeys" node-key="id"', {
  234. created() {
  235. this.defaultExpandedKeys = [111];
  236. }
  237. });
  238. expect(vm.$el.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(3);
  239. });
  240. it('autoExpandParent = false', (done) => {
  241. vm = getTreeVm(':props="defaultProps" :default-expanded-keys="defaultExpandedKeys" node-key="id" :auto-expand-parent="false"', {
  242. created() {
  243. this.defaultExpandedKeys = [11];
  244. }
  245. });
  246. expect(vm.$el.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(0);
  247. const firstNode = document.querySelector('.el-tree-node__content');
  248. firstNode.querySelector('.el-tree-node__expand-icon').click();
  249. setTimeout(() => {
  250. expect(document.querySelectorAll('.el-tree-node.is-expanded').length).to.equal(2);
  251. done();
  252. }, 100);
  253. });
  254. it('defaultCheckedKeys & check-strictly = false', () => {
  255. vm = getTreeVm(':props="defaultProps" default-expand-all show-checkbox :default-checked-keys="defaultCheckedKeys" node-key="id"', {
  256. created() {
  257. this.defaultCheckedKeys = [1];
  258. }
  259. });
  260. expect(vm.$el.querySelectorAll('.el-checkbox .is-checked').length).to.equal(3);
  261. });
  262. it('defaultCheckedKeys & check-strictly', () => {
  263. vm = getTreeVm(':props="defaultProps" default-expand-all show-checkbox :default-checked-keys="defaultCheckedKeys" node-key="id" check-strictly', {
  264. created() {
  265. this.defaultCheckedKeys = [1];
  266. }
  267. });
  268. expect(vm.$el.querySelectorAll('.el-checkbox .is-checked').length).to.equal(1);
  269. });
  270. it('show checkbox', done => {
  271. vm = getTreeVm(':props="defaultProps" show-checkbox');
  272. const tree = vm.$children[0];
  273. const secondNode = document.querySelectorAll('.el-tree-node__content')[1];
  274. const nodeCheckbox = secondNode.querySelector('.el-checkbox');
  275. expect(nodeCheckbox).to.be.exist;
  276. nodeCheckbox.click();
  277. setTimeout(() => {
  278. expect(tree.getCheckedNodes().length).to.equal(3);
  279. expect(tree.getCheckedNodes(true).length).to.equal(2);
  280. secondNode.querySelector('.el-tree-node__expand-icon').click();
  281. setTimeout(() => {
  282. const firstLeaf = secondNode.nextElementSibling.querySelector('.el-tree-node__content');
  283. const leafCheckbox = firstLeaf.querySelector('.el-checkbox');
  284. leafCheckbox.click();
  285. setTimeout(() => {
  286. expect(tree.getCheckedNodes().length).to.equal(1);
  287. done();
  288. }, 10);
  289. }, 10);
  290. }, 10);
  291. });
  292. it('setCheckedNodes', (done) => {
  293. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  294. const tree = vm.$children[0];
  295. const secondNode = document.querySelectorAll('.el-tree-node__content')[1];
  296. const nodeCheckbox = secondNode.querySelector('.el-checkbox');
  297. nodeCheckbox.click();
  298. setTimeout(() => {
  299. expect(tree.getCheckedNodes().length).to.equal(3);
  300. expect(tree.getCheckedNodes(true).length).to.equal(2);
  301. setTimeout(() => {
  302. tree.setCheckedNodes([]);
  303. expect(tree.getCheckedNodes().length).to.equal(0);
  304. done();
  305. }, 10);
  306. }, 10);
  307. });
  308. it('setCheckedKeys', () => {
  309. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  310. const tree = vm.$children[0];
  311. tree.setCheckedKeys([111]);
  312. expect(tree.getCheckedNodes().length).to.equal(3);
  313. expect(tree.getCheckedKeys().length).to.equal(3);
  314. tree.setCheckedKeys([1]);
  315. setTimeout(function() {
  316. expect(tree.getCheckedNodes().length).to.equal(3);
  317. expect(tree.getCheckedKeys().length).to.equal(3);
  318. }, 0);
  319. tree.setCheckedKeys([2]);
  320. setTimeout(function() {
  321. expect(tree.getCheckedNodes().length).to.equal(3);
  322. expect(tree.getCheckedKeys().length).to.equal(3);
  323. }, 0);
  324. tree.setCheckedKeys([21]);
  325. expect(tree.getCheckedNodes().length).to.equal(1);
  326. expect(tree.getCheckedKeys().length).to.equal(1);
  327. });
  328. it('setCheckedKeys with checkStrictly', () => {
  329. vm = getTreeVm(':props="defaultProps" checkStrictly show-checkbox node-key="id"');
  330. const tree = vm.$children[0];
  331. tree.setCheckedKeys([111]);
  332. expect(tree.getCheckedNodes().length).to.equal(1);
  333. expect(tree.getCheckedKeys().length).to.equal(1);
  334. tree.setCheckedKeys([1]);
  335. setTimeout(function() {
  336. expect(tree.getCheckedNodes().length).to.equal(1);
  337. expect(tree.getCheckedKeys().length).to.equal(1);
  338. }, 0);
  339. tree.setCheckedKeys([2]);
  340. setTimeout(function() {
  341. expect(tree.getCheckedNodes().length).to.equal(1);
  342. expect(tree.getCheckedKeys().length).to.equal(1);
  343. }, 0);
  344. tree.setCheckedKeys([21, 22]);
  345. expect(tree.getCheckedNodes().length).to.equal(2);
  346. expect(tree.getCheckedKeys().length).to.equal(2);
  347. });
  348. it('method setChecked', () => {
  349. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  350. const tree = vm.$children[0];
  351. tree.setChecked(111, true, true);
  352. expect(tree.getCheckedNodes().length).to.equal(3);
  353. expect(tree.getCheckedKeys().length).to.equal(3);
  354. tree.setChecked(vm.data[0], false, true);
  355. expect(tree.getCheckedNodes().length).to.equal(0);
  356. expect(tree.getCheckedKeys().length).to.equal(0);
  357. });
  358. it('setCheckedKeys with leafOnly=false', () => {
  359. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  360. const tree = vm.$children[0];
  361. tree.setCheckedKeys([1, 11, 111, 2], false);
  362. setTimeout(function() {
  363. expect(tree.getCheckedNodes().length).to.equal(6);
  364. expect(tree.getCheckedKeys().length).to.equal(6);
  365. }, 0);
  366. });
  367. it('setCheckedKeys with leafOnly=true', () => {
  368. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  369. const tree = vm.$children[0];
  370. tree.setCheckedKeys([2], true);
  371. setTimeout(function() {
  372. expect(tree.getCheckedNodes().length).to.equal(2);
  373. expect(tree.getCheckedKeys().length).to.equal(2);
  374. }, 0);
  375. });
  376. it('setCurrentKey', (done) => {
  377. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  378. const tree = vm.$children[0];
  379. tree.setCurrentKey(111);
  380. vm.$nextTick(() => {
  381. expect(tree.store.currentNode.data.id).to.equal(111);
  382. done();
  383. });
  384. });
  385. it('setCurrentNode', (done) => {
  386. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  387. const tree = vm.$children[0];
  388. tree.setCurrentNode({
  389. id: 111,
  390. label: '三级 1-1'
  391. });
  392. vm.$nextTick(() => {
  393. expect(tree.store.currentNode.data.id).to.equal(111);
  394. done();
  395. });
  396. });
  397. it('getCurrentKey', (done) => {
  398. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  399. const tree = vm.$children[0];
  400. tree.setCurrentKey(111);
  401. vm.$nextTick(() => {
  402. expect(tree.getCurrentKey()).to.equal(111);
  403. done();
  404. });
  405. });
  406. it('getCurrentNode', (done) => {
  407. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id"');
  408. const tree = vm.$children[0];
  409. tree.setCurrentKey(111);
  410. vm.$nextTick(() => {
  411. expect(tree.getCurrentNode().id).to.equal(111);
  412. done();
  413. });
  414. });
  415. it('getNode', () => {
  416. vm = getTreeVm(':props="defaultProps" node-key="id"');
  417. const tree = vm.$children[0];
  418. const node = tree.getNode(111);
  419. expect(node.data.id).to.equal(111);
  420. });
  421. it('remove', (done) => {
  422. vm = getTreeVm(':props="defaultProps" node-key="id"');
  423. const tree = vm.$children[0];
  424. tree.remove(1);
  425. vm.$nextTick(() => {
  426. expect(vm.data[0].id).to.equal(2);
  427. expect(tree.getNode(1)).to.equal(null);
  428. done();
  429. });
  430. });
  431. it('append', () => {
  432. vm = getTreeVm(':props="defaultProps" node-key="id"');
  433. const tree = vm.$children[0];
  434. const nodeData = { id: 88, label: '88' };
  435. tree.append(nodeData, tree.getNode(1));
  436. expect(vm.data[0].children.length).to.equal(2);
  437. expect(tree.getNode(88).data).to.equal(nodeData);
  438. });
  439. it('insertBefore', () => {
  440. vm = getTreeVm(':props="defaultProps" node-key="id"');
  441. const tree = vm.$children[0];
  442. const nodeData = { id: 88, label: '88' };
  443. tree.insertBefore(nodeData, tree.getNode(11));
  444. expect(vm.data[0].children.length).to.equal(2);
  445. expect(vm.data[0].children[0]).to.equal(nodeData);
  446. expect(tree.getNode(88).data).to.equal(nodeData);
  447. });
  448. it('insertAfter', () => {
  449. vm = getTreeVm(':props="defaultProps" node-key="id"');
  450. const tree = vm.$children[0];
  451. const nodeData = { id: 88, label: '88' };
  452. tree.insertAfter(nodeData, tree.getNode(11));
  453. expect(vm.data[0].children.length).to.equal(2);
  454. expect(vm.data[0].children[1]).to.equal(nodeData);
  455. expect(tree.getNode(88).data).to.equal(nodeData);
  456. });
  457. it('set disabled checkbox', done => {
  458. vm = getDisableTreeVm(':props="defaultProps" show-checkbox node-key="id" default-expand-all');
  459. const node = document.querySelectorAll('.el-tree-node__content')[2];
  460. const nodeCheckbox = node.querySelector('.el-checkbox input');
  461. vm.$nextTick(() => {
  462. expect(nodeCheckbox.disabled).to.equal(true);
  463. done();
  464. });
  465. });
  466. it('check strictly', (done) => {
  467. vm = getTreeVm(':props="defaultProps" show-checkbox check-strictly default-expand-all');
  468. const tree = vm.$children[0];
  469. const secondNode = document.querySelectorAll('.el-tree-node__content')[3];
  470. const nodeCheckbox = secondNode.querySelector('.el-checkbox');
  471. nodeCheckbox.click();
  472. expect(tree.getCheckedNodes().length).to.equal(1);
  473. expect(tree.getCheckedNodes(true).length).to.equal(0);
  474. const firstLeaf = secondNode.nextElementSibling.querySelector('.el-tree-node__content');
  475. const leafCheckbox = firstLeaf.querySelector('.el-checkbox');
  476. vm.$nextTick(() => {
  477. leafCheckbox.click();
  478. expect(tree.getCheckedNodes().length).to.equal(2);
  479. done();
  480. });
  481. });
  482. it('render content', () => {
  483. vm = getTreeVm(':props="defaultProps" :render-content="renderContent"', {
  484. methods: {
  485. renderContent(h, node) {
  486. return (
  487. <span class="custom-content">
  488. <el-button>{ node.node.label }</el-button>
  489. </span>
  490. );
  491. }
  492. }
  493. });
  494. const firstNode = document.querySelector('.el-tree-node__content');
  495. expect(firstNode.querySelector('.custom-content')).to.exist;
  496. const button = firstNode.querySelector('.custom-content .el-button');
  497. expect(button).to.exist;
  498. expect(button.textContent).to.equal('一级 1');
  499. });
  500. it('scoped slot', () => {
  501. vm = createVue({
  502. template: `
  503. <el-tree ref="tree" :data="data">
  504. <div slot-scope="scope" class="custom-tree-template">
  505. <span>{{ scope.node.label }}</span>
  506. <el-button></el-button>
  507. </div>
  508. </el-tree>
  509. `,
  510. data() {
  511. return {
  512. data: [{
  513. id: 1,
  514. label: '一级 1',
  515. children: [{
  516. id: 11,
  517. label: '二级 1-1',
  518. children: [{
  519. id: 111,
  520. label: '三级 1-1'
  521. }]
  522. }]
  523. }, {
  524. id: 2,
  525. label: '一级 2',
  526. children: [{
  527. id: 21,
  528. label: '二级 2-1'
  529. }, {
  530. id: 22,
  531. label: '二级 2-2'
  532. }]
  533. }, {
  534. id: 3,
  535. label: '一级 3',
  536. children: [{
  537. id: 31,
  538. label: '二级 3-1'
  539. }, {
  540. id: 32,
  541. label: '二级 3-2'
  542. }]
  543. }]
  544. };
  545. }
  546. }, true);
  547. const firstNode = document.querySelector('.custom-tree-template');
  548. expect(firstNode).to.exist;
  549. const span = firstNode.querySelector('span');
  550. const button = firstNode.querySelector('.el-button');
  551. expect(span).to.exist;
  552. expect(span.innerText).to.equal('一级 1');
  553. expect(button).to.exist;
  554. });
  555. it('load node', done => {
  556. vm = getTreeVm(':props="defaultProps" lazy :load="loadNode" show-checkbox', {
  557. methods: {
  558. loadNode(node, resolve) {
  559. if (node.level === 0) {
  560. return resolve([{ label: 'region1' }, { label: 'region2' }]);
  561. }
  562. if (node.level > 4) return resolve([]);
  563. setTimeout(() => {
  564. resolve([{
  565. label: 'zone' + this.count++
  566. }, {
  567. label: 'zone' + this.count++
  568. }]);
  569. }, 50);
  570. }
  571. }
  572. });
  573. const nodes = document.querySelectorAll('.el-tree-node__content');
  574. expect(nodes.length).to.equal(2);
  575. nodes[0].click();
  576. setTimeout(() => {
  577. expect(document.querySelectorAll('.el-tree-node__content').length).to.equal(4);
  578. done();
  579. }, 100);
  580. });
  581. it('lazy defaultChecked', done => {
  582. vm = getTreeVm(':props="defaultProps" node-key="id" lazy :load="loadNode" show-checkbox', {
  583. methods: {
  584. loadNode(node, resolve) {
  585. if (node.level === 0) {
  586. return resolve([{ label: 'region1', id: this.count++ }, { label: 'region2', id: this.count++ }]);
  587. }
  588. if (node.level > 4) return resolve([]);
  589. setTimeout(() => {
  590. resolve([{
  591. label: 'zone' + this.count,
  592. id: this.count++
  593. }, {
  594. label: 'zone' + this.count,
  595. id: this.count++
  596. }]);
  597. }, 50);
  598. }
  599. }
  600. });
  601. const tree = vm.$children[0];
  602. const firstNode = document.querySelector('.el-tree-node__content');
  603. const initStatus = firstNode.querySelector('.is-indeterminate');
  604. expect(initStatus).to.equal(null);
  605. tree.store.setCheckedKeys([3]);
  606. firstNode.querySelector('.el-tree-node__expand-icon').click();
  607. setTimeout(() => {
  608. const clickStatus = firstNode.querySelector('.is-indeterminate');
  609. expect(clickStatus).to.not.equal(null);
  610. const child = document.querySelectorAll('.el-tree-node__content')[1];
  611. expect(child.querySelector('input').checked).to.equal(true);
  612. done();
  613. }, 300);
  614. });
  615. it('lazy expandOnChecked', done => {
  616. vm = getTreeVm(':props="defaultProps" node-key="id" lazy :load="loadNode" show-checkbox check-descendants', {
  617. methods: {
  618. loadNode(node, resolve) {
  619. if (node.level === 0) {
  620. return resolve([{ label: 'region1', id: this.count++ }, { label: 'region2', id: this.count++ }]);
  621. }
  622. if (node.level > 2) return resolve([]);
  623. setTimeout(() => {
  624. resolve([{
  625. label: 'zone' + this.count,
  626. id: this.count++
  627. }, {
  628. label: 'zone' + this.count,
  629. id: this.count++
  630. }]);
  631. }, 10);
  632. }
  633. }
  634. });
  635. const tree = vm.$children[0];
  636. tree.store.setCheckedKeys([1]);
  637. setTimeout(() => {
  638. const checkedKeys = tree.getCheckedKeys();
  639. expect(checkedKeys.length).to.equal(7);
  640. done();
  641. }, 400);
  642. });
  643. it('lazy without expandOnChecked', done => {
  644. vm = getTreeVm(':props="defaultProps" node-key="id" lazy :load="loadNode" show-checkbox', {
  645. methods: {
  646. loadNode(node, resolve) {
  647. if (node.level === 0) {
  648. return resolve([{ label: 'region1', id: this.count++ }, { label: 'region2', id: this.count++ }]);
  649. }
  650. if (node.level > 4) return resolve([]);
  651. setTimeout(() => {
  652. resolve([{
  653. label: 'zone' + this.count,
  654. id: this.count++
  655. }, {
  656. label: 'zone' + this.count,
  657. id: this.count++
  658. }]);
  659. }, 50);
  660. }
  661. }
  662. });
  663. const tree = vm.$children[0];
  664. tree.store.setCheckedKeys([1]);
  665. setTimeout(() => {
  666. const nodes = document.querySelectorAll('.el-tree-node__content');
  667. expect(nodes[0].querySelector('input').checked).to.equal(true);
  668. expect(nodes.length).to.equal(2);
  669. done();
  670. }, 300);
  671. });
  672. it('accordion', done => {
  673. vm = getTreeVm(':props="defaultProps" accordion');
  674. const firstNode = vm.$el.querySelector('.el-tree-node__content');
  675. const secondNode = vm.$el.querySelector('.el-tree-node:nth-child(2) .el-tree-node__content');
  676. firstNode.click();
  677. setTimeout(() => {
  678. expect(vm.$el.querySelector('.el-tree-node').classList.contains('is-expanded')).to.true;
  679. secondNode.click();
  680. setTimeout(() => {
  681. expect(vm.$el.querySelector('.el-tree-node').classList.contains('is-expanded')).to.false;
  682. done();
  683. }, DELAY);
  684. }, DELAY);
  685. });
  686. it('handleNodeOpen & handleNodeClose', (done) => {
  687. vm = getTreeVm(':props="defaultProps" lazy :load="loadNode" @node-expand="handleNodeOpen" @node-collapse="handleNodeClose"', {
  688. methods: {
  689. loadNode(node, resolve) {
  690. if (node.level === 0) {
  691. return resolve([{label: 'region1'}, {label: 'region2'}]);
  692. }
  693. if (node.level > 4) return resolve([]);
  694. setTimeout(() => {
  695. resolve([{
  696. label: 'zone' + this.count++
  697. }, {
  698. label: 'zone' + this.count++
  699. }]);
  700. }, 50);
  701. },
  702. handleNodeOpen(data, node, vm) {
  703. this.currentNode = data;
  704. this.nodeExpended = true;
  705. },
  706. handleNodeClose(data, node, vm) {
  707. this.currentNode = data;
  708. this.nodeExpended = false;
  709. }
  710. }
  711. });
  712. const firstNode = document.querySelector('.el-tree-node__content');
  713. expect(firstNode.nextElementSibling).to.not.exist;
  714. firstNode.click();
  715. setTimeout(() => {
  716. expect(vm.nodeExpended).to.equal(true);
  717. expect(vm.currentNode.label).to.equal('region1');
  718. firstNode.click();
  719. setTimeout(() => {
  720. expect(vm.nodeExpended).to.equal(false);
  721. expect(vm.currentNode.label).to.equal('region1');
  722. done();
  723. }, 100);
  724. }, 100);
  725. });
  726. it('updateKeyChildren', (done) => {
  727. vm = getTreeVm(':props="defaultProps" show-checkbox node-key="id" default-expand-all');
  728. const tree = vm.$children[0];
  729. tree.updateKeyChildren(1, [{
  730. id: 111,
  731. label: '三级 1-1'
  732. }]);
  733. const node = document.querySelectorAll('.el-tree-node__content')[2];
  734. const label = node.querySelector('.el-tree-node__label');
  735. vm.$nextTick(() => {
  736. expect(tree.store.nodesMap['11']).to.equal(undefined);
  737. expect(tree.store.nodesMap['1'].childNodes[0].data.id).to.equal(111);
  738. expect(label.textContent).to.equal('三级 1-1');
  739. done();
  740. });
  741. });
  742. });