test: add menu focus test (#35252)

ant-design · Apr 26, 2022 · 42525ff · 42525ff
1 parent 8a36abc
commit 42525ff
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/components/menu/__tests__/index.test.js b/components/menu/__tests__/index.test.js
@@ -940,14 +940,16 @@ describe('Menu', () => {
 
   it('should support ref', async () => {
     const ref = React.createRef();
-    const wrapper = mount(
+    const { container } = render(
       <Menu ref={ref}>
         <SubMenu key="sub1" title="Navigation One">
           <Menu.Item key="1">Option 1</Menu.Item>
         </SubMenu>
       </Menu>,
     );
-    expect(ref.current?.menu?.list).toBe(wrapper.find('ul').first().getDOMNode());
+    expect(ref.current?.menu?.list).toBe(container.querySelector('ul'));
+    ref.current?.focus();
+    expect(document.activeElement).toBe(container.querySelector('ul'));
   });
 
   it('expandIcon', () => {