index.spec.js

import React from 'react';
import { shallow, mount } from 'enzyme';
import MenuBar from '.';

const menuOptions = (onClick, onOptionClick = jest.fn()) => [
  {
    title: 'file',
    options: [
      { title: 'open', onClick, className: 'testButton' },
      {
        title: 'find',
        onOptionClick,
        className: 'testOption',
        options: [{ title: 'test', onClick }],
      },
    ],
  },
];

describe('MenuBar', () => {
  it('renders', () => {
    const wrapper = shallow(<MenuBar />);
    expect(wrapper.find('menu').length).toBeTruthy();
  });

  it('renders button and standard menu if passed props', () => {
    const options = menuOptions(jest.fn());
    const wrapper = mount(<MenuBar className="MenuBar" options={options} />);
    expect(wrapper.find('AbstractButton').length).toBeTruthy();
    expect(wrapper.find('StandardMenu').length).toBeTruthy();
  });

  it('cant call onclick on button with option', () => {
    const func = jest.fn();
    const options = menuOptions(jest.fn(), func);
    const wrapper = mount(<MenuBar className="MenuBar" options={options} />);
    expect(wrapper.find('.testOption').length).toBeTruthy();
    wrapper
      .find('.testButton')
      .at(0)
      .find('button')
      .at(0)
      .simulate('click');
    expect(func).not.toHaveBeenCalled();
  });

  it('can call onclick prop child option', () => {
    const func = jest.fn();
    const options = menuOptions(func);
    const wrapper = mount(<MenuBar className="MenuBar" options={options} />);
    expect(wrapper.find('.testButton').length).toBeTruthy();
    wrapper
      .find('.testButton')
      .at(0)
      .find('button')
      .at(0)
      .simulate('click');
    expect(func).toHaveBeenCalled();
  });
});