import React from 'react';
import type { GetRef } from '../../_util/type';
import mountTest from '../../../tests/shared/mountTest';
import rtlTest from '../../../tests/shared/rtlTest';
import { act, fireEvent, render, waitFakeTimer } from '../../../tests/utils';
import Tooltip from '../../tooltip';
import Badge from '../index';
describe('Badge', () => {
mountTest(Badge);
rtlTest(Badge);
rtlTest(() => (
head
));
beforeEach(() => {
jest.useFakeTimers();
});
afterEach(() => {
jest.useRealTimers();
});
it('no strict warning', async () => {
const errSpy = jest.spyOn(console, 'error').mockImplementation(() => {});
const Comp = () => {
const [count, setCount] = React.useState(9999);
return (
<>
Badge
>
);
};
const { container } = render();
fireEvent.click(container.querySelector('button')!);
await waitFakeTimer();
expect(errSpy).not.toHaveBeenCalled();
errSpy.mockRestore();
});
it('badge dot not scaling count > 9', () => {
const { container } = render();
expect(container.querySelectorAll('.ant-card-multiple-words').length).toBe(0);
});
it('badge should support float number', () => {
const { container } = render();
expect(container.querySelectorAll('.ant-badge-multiple-words')[0].textContent).toEqual('3.5');
const { container: anotherContainer, unmount } = render();
expect(anotherContainer.querySelectorAll('.ant-badge-multiple-words')[0].textContent).toEqual(
'3.5',
);
expect(() => unmount()).not.toThrow();
});
it('badge dot not showing count == 0', () => {
const { container } = render();
expect(container.querySelectorAll('.ant-badge-dot').length).toBe(0);
});
it('should have an overridden title attribute', () => {
const { container } = render();
expect((container.querySelector('.ant-scroll-number')! as HTMLElement).title).toEqual(
'Custom title',
);
});
// https://github.com/ant-design/ant-design/issues/10626
it('should be composable with Tooltip', () => {
const ref = React.createRef>();
const { container } = render(
,
);
act(() => {
fireEvent.mouseEnter(container.querySelector('.ant-badge')!);
jest.runAllTimers();
});
expect(container.querySelector('.ant-tooltip-open')).toBeTruthy();
});
it('should render when count is changed', () => {
const { asFragment, rerender } = render();
function updateMatch(count: number) {
rerender();
act(() => {
jest.runAllTimers();
expect(asFragment().firstChild).toMatchSnapshot();
});
}
updateMatch(10);
updateMatch(11);
updateMatch(11);
updateMatch(111);
updateMatch(10);
updateMatch(9);
});
it('should be compatible with borderColor style', () => {
const { asFragment } = render(
,
);
expect(asFragment().firstChild).toMatchSnapshot();
});
// https://github.com/ant-design/ant-design/issues/13694
it('should support offset when count is a ReactNode', () => {
const { asFragment } = render(
} offset={[10, 20]}>
head
,
);
expect(asFragment().firstChild).toMatchSnapshot();
});
// https://github.com/ant-design/ant-design/issues/15349
it('should color style works on Badge', () => {
const { container } = render(
,
);
expect((container.querySelector('.ant-badge-status-text')! as HTMLElement).style.color).toEqual(
'red',
);
});
// https://github.com/ant-design/ant-design/issues/15799
it('render correct with negative number', () => {
const { asFragment } = render(
,
);
expect(asFragment().firstChild).toMatchSnapshot();
});
// https://github.com/ant-design/ant-design/issues/21331
// https://github.com/ant-design/ant-design/issues/31590
it('render Badge status/color when contains children', () => {
const { container, asFragment } = render(
,
);
expect(asFragment().firstChild).toMatchSnapshot();
expect(container.querySelectorAll('.ant-scroll-number-only-unit')[0].textContent).toBe('5');
expect(container.querySelectorAll('.ant-scroll-number-only-unit')[1].textContent).toBe('5');
expect(container.querySelectorAll('.ant-scroll-number-only-unit')[2].textContent).toBe('5');
});
it('Badge should work when status/color is empty string', () => {
const { container } = render(
<>
>,
);
expect(container.querySelectorAll('.ant-badge')).toHaveLength(2);
});
it('Badge should display count when color and count are both exist', () => {
const { container } = render(
<>
>,
);
expect(container.querySelectorAll('.ant-badge-count')).toHaveLength(1);
expect(container.querySelectorAll('[title="44"]')).toHaveLength(1);
expect(container.querySelectorAll('.ant-badge-status-dot')).toHaveLength(2);
});
it('Badge not render status-text when text is empty string', () => {
const { container } = render();
expect(container.querySelectorAll('.ant-badge > .ant-badge-status-text')).toHaveLength(0);
});
// https://github.com/ant-design/ant-design/issues/38965
it('should display custom color and number is 0', () => {
const { container } = render(
<>
>,
);
expect(container).toMatchSnapshot();
expect(container.querySelectorAll('.ant-badge-count')).toHaveLength(4);
expect(container.querySelectorAll('[title="0"]')).toHaveLength(4);
});
it('should support classNames and styles', () => {
const { container } = render(
test
,
);
const element = container.querySelector('.ant-badge');
// classNames
expect(element).toHaveClass('test-root');
expect(element?.querySelector('sup')).toHaveClass('test-indicator');
// styles
expect(element).toHaveStyle({ backgroundColor: 'yellow' });
expect(element?.querySelector('sup')).toHaveStyle({ backgroundColor: 'blue' });
});
});