mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-05 17:59:57 +08:00
46 lines
1.3 KiB
JavaScript
46 lines
1.3 KiB
JavaScript
import { calcRangeKeys } from '../utils/dictUtil';
|
|
|
|
describe('Tree util', () => {
|
|
describe('calcRangeKeys', () => {
|
|
const treeData = [
|
|
{ key: '0-0', children: [{ key: '0-0-0' }, { key: '0-0-1' }] },
|
|
{ key: '0-1', children: [{ key: '0-1-0' }, { key: '0-1-1' }] },
|
|
{
|
|
key: '0-2',
|
|
children: [
|
|
{ key: '0-2-0', children: [{ key: '0-2-0-0' }, { key: '0-2-0-1' }, { key: '0-2-0-2' }] },
|
|
],
|
|
},
|
|
];
|
|
|
|
it('calc range keys', () => {
|
|
const keys = calcRangeKeys({
|
|
treeData,
|
|
expandedKeys: ['0-0', '0-2', '0-2-0'],
|
|
startKey: '0-2-0-1',
|
|
endKey: '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());
|
|
});
|
|
|
|
it('return startKey when startKey === endKey', () => {
|
|
const keys = calcRangeKeys({
|
|
treeData,
|
|
expandedKeys: ['0-0', '0-2', '0-2-0'],
|
|
startKey: '0-0-0',
|
|
endKey: '0-0-0',
|
|
});
|
|
expect(keys).toEqual(['0-0-0']);
|
|
});
|
|
|
|
it('return empty array without startKey and endKey', () => {
|
|
const keys = calcRangeKeys({
|
|
treeData,
|
|
expandedKeys: ['0-0', '0-2', '0-2-0'],
|
|
});
|
|
expect(keys).toEqual([]);
|
|
});
|
|
});
|
|
});
|