mirror of
https://github.com/ant-design/ant-design.git
synced 2024-12-15 17:19:11 +08:00
854cdb4352
* type: optimize CompoundedComponent type * chore: fix * type: fix type * chore: fix * type: fix type * type: add ts-ignore * fix: fix * chore: revert
255 lines
7.6 KiB
TypeScript
255 lines
7.6 KiB
TypeScript
import React from 'react';
|
|
import { fireEvent, render } from '@testing-library/react';
|
|
import { act } from 'react-dom/test-utils';
|
|
|
|
import type { GetRef } from '../../_util/type';
|
|
import mountTest from '../../../tests/shared/mountTest';
|
|
import rtlTest from '../../../tests/shared/rtlTest';
|
|
import Tooltip from '../../tooltip';
|
|
import Badge from '../index';
|
|
|
|
describe('Badge', () => {
|
|
mountTest(Badge);
|
|
rtlTest(Badge);
|
|
rtlTest(() => (
|
|
<Badge count={5} offset={[10, 10]}>
|
|
<a href="#" className="head-example">
|
|
head
|
|
</a>
|
|
</Badge>
|
|
));
|
|
|
|
beforeEach(() => {
|
|
jest.useFakeTimers();
|
|
});
|
|
|
|
afterEach(() => {
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
it('no strict warning', () => {
|
|
const errSpy = jest.spyOn(console, 'error').mockImplementation(() => {});
|
|
const { rerender } = render(
|
|
<Badge dot>
|
|
<span />
|
|
</Badge>,
|
|
);
|
|
|
|
rerender(
|
|
<Badge>
|
|
<span />
|
|
</Badge>,
|
|
);
|
|
|
|
expect(errSpy).not.toHaveBeenCalled();
|
|
errSpy.mockRestore();
|
|
});
|
|
|
|
it('badge dot not scaling count > 9', () => {
|
|
const { container } = render(<Badge count={10} dot />);
|
|
expect(container.querySelectorAll('.ant-card-multiple-words').length).toBe(0);
|
|
});
|
|
|
|
it('badge should support float number', () => {
|
|
const { container } = render(<Badge count={3.5} />);
|
|
expect(container.querySelectorAll('.ant-badge-multiple-words')[0].textContent).toEqual('3.5');
|
|
|
|
const { container: anotherContainer, unmount } = render(<Badge count="3.5" />);
|
|
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(<Badge count={0} dot />);
|
|
expect(container.querySelectorAll('.ant-badge-dot').length).toBe(0);
|
|
});
|
|
|
|
it('should have an overridden title attribute', () => {
|
|
const { container } = render(<Badge count={10} title="Custom title" />);
|
|
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<GetRef<typeof Tooltip>>();
|
|
const { container } = render(
|
|
<Tooltip title="Fix the error" ref={ref}>
|
|
<Badge status="error" />
|
|
</Tooltip>,
|
|
);
|
|
|
|
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(<Badge count={9} />);
|
|
|
|
function updateMatch(count: number) {
|
|
rerender(<Badge count={count} />);
|
|
|
|
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(
|
|
<Badge
|
|
count={4}
|
|
style={{ backgroundColor: '#fff', color: '#999', borderColor: '#d9d9d9' }}
|
|
/>,
|
|
);
|
|
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(
|
|
<Badge count={<span className="custom" style={{ color: '#f5222d' }} />} offset={[10, 20]}>
|
|
<a href="#" className="head-example">
|
|
head
|
|
</a>
|
|
</Badge>,
|
|
);
|
|
expect(asFragment().firstChild).toMatchSnapshot();
|
|
});
|
|
|
|
// https://github.com/ant-design/ant-design/issues/15349
|
|
it('should color style works on Badge', () => {
|
|
const { container } = render(
|
|
<Badge style={{ color: 'red' }} status="success" text="Success" />,
|
|
);
|
|
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(
|
|
<div>
|
|
<Badge count="-10" />
|
|
<Badge count={-10} />
|
|
</div>,
|
|
);
|
|
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(
|
|
<div>
|
|
<Badge count={5} status="success">
|
|
<a />
|
|
</Badge>
|
|
<Badge count={5} color="blue">
|
|
<a />
|
|
</Badge>
|
|
<Badge count={5} color="#08c">
|
|
<a />
|
|
</Badge>
|
|
</div>,
|
|
);
|
|
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(
|
|
<>
|
|
<Badge color="" text="text" />
|
|
<Badge status={'' as any} text="text" />
|
|
</>,
|
|
);
|
|
|
|
expect(container.querySelectorAll('.ant-badge')).toHaveLength(2);
|
|
});
|
|
|
|
it('Badge should display count when color and count are both exist', () => {
|
|
const { container } = render(
|
|
<>
|
|
<Badge className="badge1" text="badge" color="pink" count={44} />
|
|
<Badge className="badge2" text="badge" color="pink" count={0} />
|
|
<Badge className="badge3" text="badge" color="pink" />
|
|
</>,
|
|
);
|
|
|
|
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(<Badge status="default" text={undefined} />);
|
|
|
|
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(
|
|
<>
|
|
<Badge count={0} showZero color="#ff0" />
|
|
<Badge count={0} showZero color="blue" />
|
|
<Badge count={0} showZero />
|
|
<Badge count={0} showZero color="green">
|
|
<div />
|
|
</Badge>
|
|
</>,
|
|
);
|
|
|
|
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(
|
|
<Badge
|
|
count={10}
|
|
classNames={{
|
|
root: 'test-root',
|
|
indicator: 'test-indicator',
|
|
}}
|
|
styles={{
|
|
root: { backgroundColor: 'yellow' },
|
|
indicator: { backgroundColor: 'blue' },
|
|
}}
|
|
>
|
|
test
|
|
</Badge>,
|
|
);
|
|
|
|
const element = container.querySelector<HTMLSpanElement>('.ant-badge');
|
|
|
|
// classNames
|
|
expect(element).toHaveClass('test-root');
|
|
expect(element?.querySelector<HTMLElement>('sup')).toHaveClass('test-indicator');
|
|
|
|
// styles
|
|
expect(element).toHaveStyle({ backgroundColor: 'yellow' });
|
|
expect(element?.querySelector<HTMLElement>('sup')).toHaveStyle({ backgroundColor: 'blue' });
|
|
});
|
|
});
|