/* eslint no-use-before-define: "off" */ import React from 'react'; import { act } from 'react-dom/test-utils'; import Transfer from '..'; import { fireEvent, render } from '../../../tests/utils'; 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(container: HTMLElement, index: number) { const items = Array.from( container // Menu .querySelector('.ant-dropdown-menu')! // Items .querySelectorAll('li.ant-dropdown-menu-item'), ); fireEvent.click(items[index]); } it('select all', () => { jest.useFakeTimers(); const onSelectChange = jest.fn(); const { container } = render(); fireEvent.mouseEnter(container.querySelector('.ant-transfer-list-header-dropdown')!); act(() => { jest.runAllTimers(); }); clickItem(container, 0); expect(onSelectChange).toHaveBeenCalledWith(['b', 'c', 'd', 'e'], []); jest.useRealTimers(); }); it('select current page', () => { jest.useFakeTimers(); const onSelectChange = jest.fn(); const { container } = render(); fireEvent.mouseEnter(container.querySelector('.ant-transfer-list-header-dropdown')!); act(() => { jest.runAllTimers(); }); clickItem(container, 1); expect(onSelectChange).toHaveBeenCalledWith(['b', 'c', 'd'], []); jest.useRealTimers(); }); it('should hide checkbox and dropdown icon when showSelectAll={false}', () => { const { container } = render(); expect(container.querySelector('.ant-transfer-list-header-dropdown')).toBeFalsy(); expect( container.querySelector('.ant-transfer-list-header .ant-transfer-list-checkbox'), ).toBeFalsy(); }); describe('select invert', () => { [ { name: 'with pagination', props: listProps, index: 2, keys: ['c', 'd'] }, { name: 'without pagination', props: { ...listProps, pagination: null as any }, index: 1, keys: ['c', 'd', 'e'], }, ].forEach(({ name, props, index, keys }) => { it(name, () => { jest.useFakeTimers(); const onSelectChange = jest.fn(); const { container } = render(); fireEvent.mouseEnter(container.querySelector('.ant-transfer-list-header-dropdown')!); act(() => { jest.runAllTimers(); }); clickItem(container, index); expect(onSelectChange).toHaveBeenCalledWith(keys, []); jest.useRealTimers(); }); }); }); describe('oneWay to remove', () => { [ { name: 'with pagination', props: listProps }, { name: 'without pagination', props: { ...listProps, pagination: null as any } }, ].forEach(({ name, props }) => { it(name, () => { jest.useFakeTimers(); const onChange = jest.fn(); const { container } = render( , ); // Right dropdown fireEvent.mouseEnter(container.querySelectorAll('.ant-transfer-list-header-dropdown')[1]!); act(() => { jest.runAllTimers(); }); clickItem(container, 0); expect(onChange).toHaveBeenCalledWith([], 'left', ['b', 'c']); jest.useRealTimers(); }); }); }); });