/* eslint-disable no-undef, react/no-multi-comp */ import React from 'react'; import { render, mount } from 'enzyme'; import { renderToJson } from 'enzyme-to-json'; import KeyCode from 'rc-util/lib/KeyCode'; import Menu, { MenuItem, MenuItemGroup, SubMenu, Divider } from '../src'; describe('Menu', () => { describe('render', () => { function createMenu(props) { return ( 1 2 3 4 5 6 ); } ['vertical', 'horizontal', 'inline'].forEach((mode) => { it(`renders ${mode} menu correctly`, () => { const wrapper = render(createMenu({ mode })); expect(renderToJson(wrapper)).toMatchSnapshot(); }); }); }); describe('render role listbox', () => { function createMenu() { return ( 1 2 3 ); } it('renders menu correctly', () => { const wrapper = render(createMenu()); expect(renderToJson(wrapper)).toMatchSnapshot(); }); }); it('set activeKey', () => { const wrapper = mount( 1 2 , ); expect(wrapper.find('MenuItem').first().props().active).toBe(true); expect(wrapper.find('MenuItem').last().props().active).toBe(false); }); it('active first item', () => { const wrapper = mount( 1 2 , ); expect(wrapper.find('MenuItem').first().props().active).toBe(true); }); it('select multiple items', () => { const wrapper = mount( 1 2 , ); wrapper.find('MenuItem').first().simulate('click'); wrapper.find('MenuItem').last().simulate('click'); expect(wrapper.find('.rc-menu-item-selected').length).toBe(2); }); it('can be controlled by selectedKeys', () => { const wrapper = mount( 1 2 , ); expect(wrapper.find('li').first().props().className).toContain('-selected'); wrapper.setProps({ selectedKeys: ['2'] }); expect(wrapper.find('li').last().props().className).toContain('-selected'); }); it('select default item', () => { const wrapper = mount( 1 2 , ); expect(wrapper.find('li').first().props().className).toContain('-selected'); }); it('can be controlled by openKeys', () => { const wrapper = mount( 1 2 , ); expect(wrapper.find('ul').first().props().className).not.toContain('-hidden'); wrapper.setProps({ openKeys: ['g2'] }); expect(wrapper.find('ul').last().props().className).not.toContain('-hidden'); }); it('open default submenu', () => { const wrapper = mount( 1 2 , ); expect(wrapper.find('ul').first().props().className).not.toContain('-hidden'); }); it('fires select event', () => { const handleSelect = jest.fn(); const wrapper = mount( 1 2 , ); wrapper.find('MenuItem').first().simulate('click'); expect(handleSelect.mock.calls[0][0].key).toBe('1'); }); it('fires click event', () => { const handleClick = jest.fn(); const wrapper = mount( 1 2 , ); wrapper.find('MenuItem').first().simulate('click'); expect(handleClick.mock.calls[0][0].key).toBe('1'); }); it('fires deselect event', () => { const handleDeselect = jest.fn(); const wrapper = mount( 1 2 , ); wrapper.find('MenuItem').first().simulate('click').simulate('click'); expect(handleDeselect.mock.calls[0][0].key).toBe('1'); }); it('active by mouse enter', () => { const wrapper = mount( item disabled item2 , ); let menuItem = wrapper.find('MenuItem').last(); menuItem.simulate('mouseEnter'); menuItem = wrapper.find('MenuItem').last(); expect(menuItem.props().active).toBe(true); }); it('active by key down', () => { const wrapper = mount( 1 2 , ); wrapper.simulate('keyDown', { keyCode: KeyCode.DOWN }); expect(wrapper.find('MenuItem').at(1).props().active).toBe(true); }); it('keydown works when children change', () => { class App extends React.Component { state = { items: [1, 2, 3], } render() { return ( {this.state.items.map(i => {i}, )} ); } } const wrapper = mount(); wrapper.setState({ items: [0, 1] }); wrapper.find('Menu').simulate('keyDown', { keyCode: KeyCode.DOWN }); expect(wrapper.find('MenuItem').at(0).props().active).toBe(true); wrapper.find('Menu').simulate('keyDown', { keyCode: KeyCode.DOWN }); expect(wrapper.find('MenuItem').at(1).props().active).toBe(true); }); it('active first item when children changes', () => { class App extends React.Component { state = { items: ['foo'], } render() { return ( {this.state.items.map(item => {item}, )} ); } } const wrapper = mount(); wrapper.setState({ items: ['bar', 'foo'] }); expect( wrapper.find('li').first().hasClass('rc-menu-item-active'), ).toBe(true); }); });