ahbeng/NUSMods

import { shallow } from 'enzyme';

import SemesterSwitcher from './SemesterSwitcher';

describe('<SemesterSwitcher />', () => {
  test('simulates click events', () => {
    const onButtonClick = jest.fn();
    const wrapper = shallow(<SemesterSwitcher semester={2} onSelectSemester={onButtonClick} />);
    const buttons = wrapper.find('button');
    buttons.at(0).simulate('click');
    expect(onButtonClick).toBeCalledWith(1);
    buttons.at(1).simulate('click');
    expect(onButtonClick).toBeCalledWith(3);
  });

  describe('semester switching buttons are disabled appropriately', () => {
    test('left button', () => {
      const onButtonClick = jest.fn();
      const wrapper = shallow(<SemesterSwitcher semester={1} onSelectSemester={onButtonClick} />);
      const buttons = wrapper.find('button');
      buttons.at(0).simulate('click');
      expect(onButtonClick).not.toBeCalled();
      buttons.at(1).simulate('click');
      expect(onButtonClick).toBeCalledWith(2);
    });

    test('right button', () => {
      const onButtonClick = jest.fn();
      const wrapper = shallow(<SemesterSwitcher semester={4} onSelectSemester={onButtonClick} />);
      const buttons = wrapper.find('button');
      buttons.at(1).simulate('click');
      expect(onButtonClick).not.toBeCalled();
      buttons.at(0).simulate('click');
      expect(onButtonClick).toBeCalledWith(3);
    });
  });
});