From c708b1fe134000102c48449aaa4c32832bbbb2f0 Mon Sep 17 00:00:00 2001
From: LuckyFBB <976060700@qq.com>
Date: Wed, 13 Aug 2025 20:48:14 +0800
Subject: [PATCH] fix(button): add size prop to AntdButton

#597
---
 src/button/__tests__/index.test.tsx | 5 +++++
 src/button/index.tsx                | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

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(<Button size="small">Test</Button>);
+        expect(container.firstChild).toHaveClass('ant-btn-sm');
+    });
+
     it('passes other props to AntdButton', () => {
         const { getByText } = render(<Button type="primary">Primary</Button>);
         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 (
-        <AntdButton className={classNames('dtc-button', className)} {...rest}>
+        <AntdButton className={classNames('dtc-button', className)} size={size} {...rest}>
             {icon && <span className={`dtc-button__icon dtc-button__icon--${size}`}>{icon}</span>}
             {children && (
                 <span className={`dtc-button__text dtc-button__text--${size}`}>{children}</span>