test: add test case

Author: zombieJ

src/Step.tsx

@@ -139,7 +139,7 @@ export default function Step(props: StepProps) {
     iconNode = iconRender(iconNode, {
       ...renderInfo,
       components: {
-        icon: StepIcon,
+        Icon: StepIcon,
       },
     }) as React.ReactElement;
   }

src/StepIcon.tsx

@@ -6,7 +6,7 @@ import pickAttrs from '@rc-component/util/lib/pickAttrs';
 export type StepIconProps = React.HTMLAttributes<HTMLDivElement>;
 
 const StepIcon = React.forwardRef<HTMLDivElement, StepIconProps>((props, ref) => {
-  const { className, style, ...restProps } = props;
+  const { className, style, children, ...restProps } = props;
   const { prefixCls, classNames, styles } = React.useContext(StepsContext);
 
   const itemCls = `${prefixCls}-item`;
@@ -20,7 +20,9 @@ const StepIcon = React.forwardRef<HTMLDivElement, StepIconProps>((props, ref) =>
         ...styles.itemIcon,
         ...style,
       }}
-    />
+    >
+      {children}
+    </div>
   );
 });
 

src/Steps.tsx

@@ -73,7 +73,7 @@ export interface StepsProps {
     originNode: React.ReactElement,
     info: RenderInfo & {
       components: {
-        icon: typeof StepIcon;
+        Icon: typeof StepIcon;
       };
     },
   ) => React.ReactNode;

tests/index.test.tsx

@@ -344,4 +344,23 @@ describe('Steps', () => {
     expect(container.querySelector('.rc-steps-item')).toBeTruthy();
     expect(container.querySelector('.rc-steps-item > .bamboo')).toBeTruthy();
   });
+
+  it('iconRender', () => {
+    const { container } = render(
+      <Steps
+        items={[
+          {
+            title: 'test',
+          },
+        ]}
+        iconRender={(_, { components: { Icon } }) => {
+          return <Icon className="bamboo">little</Icon>;
+        }}
+      />,
+    );
+
+    const iconEle = container.querySelector('.rc-steps-item-icon')!;
+    expect(iconEle).toHaveClass('bamboo');
+    expect(iconEle.textContent).toBe('little');
+  });
 });