mirror of
https://gitee.com/ant-design/ant-design.git
synced 2024-12-05 21:48:19 +08:00
36 lines
1.0 KiB
JavaScript
36 lines
1.0 KiB
JavaScript
import React from 'react';
|
|
import { mount } from 'enzyme';
|
|
import Tree from '../index';
|
|
import { calcRangeKeys } from '../util';
|
|
|
|
const { TreeNode } = Tree;
|
|
|
|
describe('Tree util', () => {
|
|
it('calc range keys', () => {
|
|
const wrapper = mount(
|
|
<Tree>
|
|
<TreeNode key="0-0">
|
|
<TreeNode key="0-0-0" />
|
|
<TreeNode key="0-0-1" />
|
|
</TreeNode>
|
|
<TreeNode key="0-1">
|
|
<TreeNode key="0-1-0" />
|
|
<TreeNode key="0-1-1" />
|
|
</TreeNode>
|
|
<TreeNode key="0-2">
|
|
<TreeNode key="0-2-0">
|
|
<TreeNode key="0-2-0-0" />
|
|
<TreeNode key="0-2-0-1" />
|
|
<TreeNode key="0-2-0-2" />
|
|
</TreeNode>
|
|
</TreeNode>
|
|
</Tree>,
|
|
);
|
|
|
|
const { children } = wrapper.find(Tree).props();
|
|
const keys = calcRangeKeys(children, ['0-0', '0-2', '0-2-0'], '0-2-0-1', '0-0-0');
|
|
const target = ['0-0-0', '0-0-1', '0-1', '0-2', '0-2-0', '0-2-0-0', '0-2-0-1'];
|
|
expect(keys.sort()).toEqual(target.sort());
|
|
});
|
|
});
|