ant-design/components/calendar/__tests__/select.test.tsx
2024-01-02 15:37:00 +08:00

98 lines
2.9 KiB
TypeScript

import Dayjs from 'dayjs';
import 'dayjs/locale/zh-cn';
import React from 'react';
import { resetWarned } from 'rc-util/lib/warning';
import Calendar from '..';
import { fireEvent, render, waitFakeTimer } from '../../../tests/utils';
describe('Calendar.onSelect', () => {
beforeEach(() => {
resetWarned();
jest.useFakeTimers().setSystemTime(new Date('2000-02-01'));
});
afterEach(() => {
jest.clearAllTimers();
jest.useRealTimers();
});
it('source of year select', async () => {
const onSelect = jest.fn();
const { container } = render(<Calendar onSelect={onSelect} />);
fireEvent.mouseDown(container.querySelector('.ant-select-selector')!);
await waitFakeTimer();
fireEvent.click(container.querySelector('.ant-select-item-option')!);
await waitFakeTimer();
expect(onSelect).toHaveBeenCalledWith(expect.anything(), { source: 'year' });
});
it('source of month select', async () => {
const onSelect = jest.fn();
const { container } = render(<Calendar onSelect={onSelect} />);
fireEvent.mouseDown(container.querySelectorAll('.ant-select-selector')[1]!);
await waitFakeTimer();
fireEvent.click(container.querySelector('.ant-select-item-option')!);
await waitFakeTimer();
expect(onSelect).toHaveBeenCalledWith(expect.anything(), { source: 'month' });
});
it('source of customize', async () => {
const onSelect = jest.fn();
const { container } = render(
<Calendar
onSelect={onSelect}
headerRender={({ onChange }) => (
<button
className="bamboo"
type="button"
onClick={() => {
onChange(Dayjs('1999-01-01'));
}}
>
Trigger
</button>
)}
/>,
);
fireEvent.click(container.querySelector('.bamboo')!);
await waitFakeTimer();
expect(onSelect).toHaveBeenCalledWith(expect.anything(), { source: 'customize' });
});
it('source of date', () => {
const onSelect = jest.fn();
const { container } = render(<Calendar onSelect={onSelect} />);
fireEvent.click(container.querySelector('.ant-picker-cell')!);
expect(onSelect).toHaveBeenCalledWith(expect.anything(), { source: 'date' });
});
it('source of date with month panel', async () => {
const onSelect = jest.fn();
const onPanelChange = jest.fn();
const { container } = render(<Calendar onSelect={onSelect} onPanelChange={onPanelChange} />);
// Default is month radio
fireEvent.click(container.querySelector('.ant-picker-cell')!);
expect(onSelect).toHaveBeenCalledWith(expect.anything(), { source: 'date' });
// Click year radio
fireEvent.click(container.querySelectorAll('.ant-radio-button-input')[1]!);
expect(onPanelChange).toHaveBeenCalledWith(expect.anything(), 'year');
fireEvent.click(container.querySelector('.ant-picker-cell')!);
expect(onSelect).toHaveBeenCalledWith(expect.anything(), { source: 'month' });
});
});