diff --git a/src/button/__tests__/index.test.tsx b/src/button/__tests__/index.test.tsx index 6fb4df277..7e1378a2c 100644 --- a/src/button/__tests__/index.test.tsx +++ b/src/button/__tests__/index.test.tsx @@ -33,6 +33,11 @@ describe('Button', () => { expect(container.firstChild).toHaveClass('custom-class'); }); + it('applies size className', () => { + const { container } = render(); + expect(container.firstChild).toHaveClass('ant-btn-sm'); + }); + it('passes other props to AntdButton', () => { const { getByText } = render(); expect(getByText('Primary').parentNode).toHaveClass('ant-btn-primary'); diff --git a/src/button/index.tsx b/src/button/index.tsx index 9fb9dc386..b18ac868f 100644 --- a/src/button/index.tsx +++ b/src/button/index.tsx @@ -14,7 +14,7 @@ export default function Button({ ...rest }: ButtonProps) { return ( - + {icon && {icon}} {children && ( {children}