ant-design/components/date-picker/__tests__/utils.ts

import type { render } from '../../../tests/utils';
import { fireEvent, screen } from '../../../tests/utils';

export function openPicker(wrapper: ReturnType<typeof render>, index = 0) {
  fireEvent.mouseDown(wrapper.container?.querySelectorAll('input')?.[index]!);
  fireEvent.focus(wrapper.container?.querySelectorAll('input')?.[index]!);
}

export function closePicker(wrapper: ReturnType<typeof render>, index = 0) {
  fireEvent.blur(wrapper.container?.querySelectorAll('input')[index]!);
}

export function selectCell(wrapper: ReturnType<typeof render>, text: string | number, index = 0) {
  let matchCell: HTMLTableCellElement | null = null;
  const tds = wrapper.container?.querySelectorAll('table')?.[index]?.querySelectorAll('td');
  tds.forEach((td) => {
    if (td.querySelector('div')?.innerHTML === String(text) && td.className.includes('-in-view')) {
      matchCell = td;
      fireEvent.click(td);
    }
  });
  /* istanbul ignore next */
  if (!matchCell) {
    throw new Error('Cell not match in picker panel.');
  }
  return matchCell;
}

export const closeCircleByRole = { role: 'img', options: { name: 'close-circle' } } as const;

export function expectCloseCircle(shouldExist: boolean) {
  const { role, options } = closeCircleByRole;
  const count = shouldExist ? 1 : 0;
  return expect(screen.queryAllByRole(role, options).length).toStrictEqual(count);
}