ant-design/components/date-picker/__tests__/focus.test.js

import React from 'react';
import { mount } from 'enzyme';
import moment from 'moment';
import MockDate from 'mockdate';
import DatePicker from '..';
import {
  selectDate,
  openPanel,
} from './utils';

const { MonthPicker, WeekPicker, RangePicker } = DatePicker;

describe('DatePicker', () => {
  beforeEach(() => {
    MockDate.set(moment('2016-11-22'));
  });

  afterEach(() => {
    MockDate.reset();
  });

  it('should focus trigger input after select date in DatePicker', () => {
    const wrapper = mount(
      <DatePicker />
    );
    openPanel(wrapper);
    selectDate(wrapper, moment('2016-11-23'));
    expect(wrapper.find('.ant-calendar-picker-input').getDOMNode()).toBe(document.activeElement);
  });

  it('should focus trigger input after select date in RangePicker', () => {
    const wrapper = mount(
      <RangePicker />
    );
    openPanel(wrapper);
    selectDate(wrapper, moment('2016-11-23'), 0);
    selectDate(wrapper, moment('2016-11-28'), 1);
    expect(wrapper.find('.ant-calendar-picker').getDOMNode()).toBe(document.activeElement);
  });

  it('should focus trigger input after select date in MonthPicker', () => {
    const wrapper = mount(
      <MonthPicker />
    );
    openPanel(wrapper);
    wrapper.find('.ant-calendar-month-panel-month').first().simulate('click');
    wrapper.find('.ant-calendar-month-panel-cell').at(6).hasClass('ant-calendar-month-panel-selected-cell');
    expect(wrapper.find('.ant-calendar-picker-input').getDOMNode()).toBe(document.activeElement);
  });

  it('should focus trigger input after select date in WeekPicker', () => {
    const wrapper = mount(
      <WeekPicker />
    );
    openPanel(wrapper);
    selectDate(wrapper, moment('2016-11-23'));
    expect(wrapper.find('.ant-calendar-picker-input').getDOMNode()).toBe(document.activeElement);
  });
});