mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-09 12:59:12 +08:00
130 lines
3.3 KiB
JavaScript
130 lines
3.3 KiB
JavaScript
/* eslint no-use-before-define: "off" */
|
|
import React from 'react';
|
|
import { act } from 'react-dom/test-utils';
|
|
import { mount } from 'enzyme';
|
|
import Transfer from '..';
|
|
|
|
const listProps = {
|
|
dataSource: [
|
|
{
|
|
key: 'a',
|
|
title: 'a',
|
|
disabled: true,
|
|
},
|
|
{
|
|
key: 'b',
|
|
title: 'b',
|
|
},
|
|
{
|
|
key: 'c',
|
|
title: 'c',
|
|
},
|
|
{
|
|
key: 'd',
|
|
title: 'd',
|
|
},
|
|
{
|
|
key: 'e',
|
|
title: 'e',
|
|
},
|
|
],
|
|
selectedKeys: ['b'],
|
|
targetKeys: [],
|
|
pagination: { pageSize: 4 },
|
|
};
|
|
|
|
describe('Transfer.Dropdown', () => {
|
|
function clickItem(wrapper, index) {
|
|
wrapper.find('li.ant-dropdown-menu-item').at(index).simulate('click');
|
|
}
|
|
|
|
it('select all', () => {
|
|
jest.useFakeTimers();
|
|
|
|
const onSelectChange = jest.fn();
|
|
const wrapper = mount(<Transfer {...listProps} onSelectChange={onSelectChange} />);
|
|
wrapper.find('.ant-transfer-list-header-dropdown').first().simulate('mouseenter');
|
|
act(() => {
|
|
jest.runAllTimers();
|
|
});
|
|
wrapper.update();
|
|
|
|
clickItem(wrapper.find('.ant-dropdown-menu').first(), 0);
|
|
expect(onSelectChange).toHaveBeenCalledWith(['b', 'c', 'd', 'e'], []);
|
|
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
it('select current page', () => {
|
|
jest.useFakeTimers();
|
|
|
|
const onSelectChange = jest.fn();
|
|
const wrapper = mount(<Transfer {...listProps} onSelectChange={onSelectChange} />);
|
|
wrapper.find('.ant-transfer-list-header-dropdown').first().simulate('mouseenter');
|
|
act(() => {
|
|
jest.runAllTimers();
|
|
});
|
|
wrapper.update();
|
|
|
|
clickItem(wrapper.find('.ant-dropdown-menu').first(), 1);
|
|
expect(onSelectChange).toHaveBeenCalledWith(['b', 'c', 'd'], []);
|
|
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
describe('select invert', () => {
|
|
[
|
|
{ name: 'with pagination', props: listProps, index: 2, keys: ['c', 'd'] },
|
|
{
|
|
name: 'without pagination',
|
|
props: { ...listProps, pagination: null },
|
|
index: 1,
|
|
keys: ['c', 'd', 'e'],
|
|
},
|
|
].forEach(({ name, props, index, keys }) => {
|
|
it(name, () => {
|
|
jest.useFakeTimers();
|
|
|
|
const onSelectChange = jest.fn();
|
|
const wrapper = mount(<Transfer {...props} onSelectChange={onSelectChange} />);
|
|
wrapper.find('.ant-transfer-list-header-dropdown').first().simulate('mouseenter');
|
|
act(() => {
|
|
jest.runAllTimers();
|
|
});
|
|
wrapper.update();
|
|
|
|
clickItem(wrapper.find('.ant-dropdown-menu').first(), index);
|
|
expect(onSelectChange).toHaveBeenCalledWith(keys, []);
|
|
|
|
jest.useRealTimers();
|
|
});
|
|
});
|
|
});
|
|
|
|
describe('oneWay to remove', () => {
|
|
[
|
|
{ name: 'with pagination', props: listProps },
|
|
{ name: 'without pagination', props: { ...listProps, pagination: null } },
|
|
].forEach(({ name, props }) => {
|
|
it(name, () => {
|
|
jest.useFakeTimers();
|
|
|
|
const onChange = jest.fn();
|
|
const wrapper = mount(
|
|
<Transfer {...props} targetKeys={['b', 'c']} oneWay onChange={onChange} />,
|
|
);
|
|
wrapper.find('.ant-transfer-list-header-dropdown').last().simulate('mouseenter');
|
|
act(() => {
|
|
jest.runAllTimers();
|
|
});
|
|
wrapper.update();
|
|
|
|
clickItem(wrapper.find('.ant-dropdown-menu').first(), 0);
|
|
expect(onChange).toHaveBeenCalledWith([], 'left', ['b', 'c']);
|
|
|
|
jest.useRealTimers();
|
|
});
|
|
});
|
|
});
|
|
});
|