@react-native-community/datetimepicker/jest/index.js

DateTimePicker component for React Native

/**
 * @format
 * @flow strict-local
 */
import * as androidUtils from '../src/androidUtils';
import {DATE_SET_ACTION, DISMISS_ACTION} from '../src/constants';
import type {PresentPickerCallback} from '../src/androidUtils';

export const mockAndroidDialogDateChange = (datePickedByUser: Date) => {
  jest.spyOn(androidUtils, 'getOpenPicker').mockImplementation(() => {
    // $FlowFixMe[missing-local-annot]
    async function fakeDateTimePickerAndroidOpener({
      value: timestampFromPickerValueProp,
    }) {
      const pickedDate = new Date(timestampFromPickerValueProp);
      pickedDate.setTime(datePickedByUser.getTime());

      return {
        action: DATE_SET_ACTION,
        timestamp: pickedDate.getTime(),
        utcOffset: 0,
      };
    }
    return (fakeDateTimePickerAndroidOpener: PresentPickerCallback);
  });
};

export const mockAndroidDialogDismissal = () => {
  jest.spyOn(androidUtils, 'getOpenPicker').mockImplementation(() => {
    async function fakeDateTimePickerAndroidOpener() {
      return {
        action: DISMISS_ACTION,
      };
    }
    // $FlowExpectedError - the typings actually don't 100% reflect the native module behavior
    return (fakeDateTimePickerAndroidOpener: PresentPickerCallback);
  });
};