test: add case

Wxh16144-forks · Jan 6, 2023 · 2991e93 · 2991e93
1 parent 0dc7815
commit 2991e93
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/components/menu/__tests__/index.test.tsx b/components/menu/__tests__/index.test.tsx
@@ -1047,4 +1047,34 @@ describe('Menu', () => {
     );
     errorSpy.mockRestore();
   });
+
+  it('expandIconClassName', () => {
+    const Demo = ({ expandIcon }: Pick<MenuProps, 'expandIcon'>) => (
+      <Menu
+        expandIcon={expandIcon}
+        inlineCollapsed
+        items={[
+          {
+            label: 'Option 1',
+            key: '1',
+            icon: '112',
+            children: [
+              {
+                label: 'Option 1-1',
+                key: '1-1',
+              },
+            ],
+          },
+        ]}
+      />
+    );
+
+    const { container, rerender } = render(
+      <Demo expandIcon={<span className="custom-expand-icon" />} />,
+    );
+    expect(container.querySelector('.custom-expand-icon')).toBeTruthy();
+
+    rerender(<Demo expandIcon={() => <span className="function-custom-expand-icon" />} />);
+    expect(container.querySelector('.function-custom-expand-icon')).toBeTruthy();
+  });
 });