import React from 'react';
import { mount } from 'enzyme';
import CreatePollMenu from './CreatePollMenu';

describe('CreatePollMenu', () => {
  it('should match the snapshot', () => {
    const component = mount(<CreatePollMenu />);
    expect(component.getElement()).toMatchSnapshot();
  });

  const user = {
    data: {
      name: 'Tom Tom',
      email: 'tom@example.com',
      avatarSmallUrl:
        'https://6a5edc300520d4037dd6-0732807511066685711db213ddc1d2df.ssl.cf2.rackcdn.com/snw8ceng29ch3zbs4pvro3t2jxswf8yk',
      accessToken: 'abc123',
    },
    loading: false,
    isEligibleForFreeTrial: true,
  };

  it('should call the function passed into onClickCreatePoll', () => {
    const mockFunction = jest.fn();
    const component = mount(<CreatePollMenu onClickCreatePoll={mockFunction} user={user} />);
    const button = component.find('.Menu span.Button');
    button.simulate('click');
    const menuItem = component.find('span.Menu-item[data-ga-action="clickedHomepageCreateDatepoll"]');
    menuItem.simulate('click');
    expect(mockFunction).toHaveBeenCalled();
  });

  it('should call the function passed into onClickCreateOneOnOne', () => {
    const mockFunction = jest.fn();
    const component = mount(<CreatePollMenu onClickCreateOneOnOne={mockFunction} user={user} />);
    const button = component.find('.Menu span.Button');
    button.simulate('click');
    const menuItem = component.find('span.Menu-item[data-ga-action="clickedHomepageCreateD11"]');
    menuItem.simulate('click');
    expect(mockFunction).toHaveBeenCalled();
  });

  it('should call the function passed into onClickCreateSurvey', () => {
    const mockFunction = jest.fn();
    const component = mount(<CreatePollMenu onClickCreateSurvey={mockFunction} user={user} />);
    const button = component.find('.Menu span.Button');
    button.simulate('click');
    const menuItem = component.find('span.Menu-item[data-ga-action="clickedHomepageCreateTextpoll"]');
    menuItem.simulate('click');
    expect(mockFunction).toHaveBeenCalled();
  });
});