box-ui-elements-mlh/src/elements/content-sidebar/__tests__/AddTaskButton.test.js

Box UI Elements

import React from 'react';
import { mount } from 'enzyme';
import { AddTaskButtonComponent as AddTaskButton } from '../AddTaskButton';

describe('elements/content-sidebar/AddTaskButton', () => {
    /* 
    1. Pushing the open state into history keeps the sidebar open upon resize and refresh
    2. Preventing the sidebar from closing keeps the task modal open upon edit and resize 
    */

    test('should call history.replace state with force open state when task menu items are clicked', () => {
        const historyMock = { replace: jest.fn() };
        const wrapper = mount(<AddTaskButton history={historyMock} />);

        const button = wrapper.find('Button');
        button.simulate('click');

        const menuItem = wrapper.find('MenuItem').first();
        menuItem.simulate('click');

        expect(historyMock.replace).toHaveBeenCalledWith({ state: { open: true } });
    });

    test('should set state.isTaskFormOpen to false and call onTaskModalClose when task modal is closed', () => {
        const onTaskModalCloseMock = jest.fn();
        const mockButtonRef = {
            current: {
                focus: jest.fn(),
            },
        };
        const wrapper = shallow(<AddTaskButton onTaskModalClose={onTaskModalCloseMock} />);
        wrapper.instance().buttonRef = mockButtonRef;
        wrapper.setState({ isTaskFormOpen: true });

        wrapper.instance().handleModalClose();

        expect(wrapper.state('isTaskFormOpen')).toBe(false);
        expect(mockButtonRef.current.focus).toHaveBeenCalledTimes(1);
        expect(onTaskModalCloseMock).toHaveBeenCalledTimes(1);
    });
});