From d68147c6266832ee888f0fa8868cf519b2023b8f Mon Sep 17 00:00:00 2001
From: zombiej <smith3816@gmail.com>
Date: Fri, 28 May 2021 15:56:18 +0800
Subject: [PATCH] fix: layout mode of className

---
 src/SubMenu/index.tsx |  5 ++++-
 tests/SubMenu.spec.js | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 36 insertions(+), 1 deletion(-)

diff --git a/src/SubMenu/index.tsx b/src/SubMenu/index.tsx
index 8064d2ee..6871d733 100644
--- a/src/SubMenu/index.tsx
+++ b/src/SubMenu/index.tsx
@@ -300,7 +300,10 @@ const InternalSubMenu = (props: SubMenuProps) => {
         popupClassName={popupClassName}
         popupOffset={popupOffset}
         popup={
-          <MenuContextProvider mode={triggerMode}>
+          <MenuContextProvider
+            // Special handle of horizontal mode
+            mode={triggerMode === 'horizontal' ? 'vertical' : triggerMode}
+          >
             <SubMenuList id={popupId} ref={popupRef}>
               {children}
             </SubMenuList>
diff --git a/tests/SubMenu.spec.js b/tests/SubMenu.spec.js
index c7de9afd..7c17b949 100644
--- a/tests/SubMenu.spec.js
+++ b/tests/SubMenu.spec.js
@@ -407,5 +407,37 @@ describe('SubMenu', () => {
     wrapper.find('div.rc-menu-submenu-title').last().simulate('click');
     expect(onOpenChange).toHaveBeenCalledWith(['light']);
   });
+
+  it('popup className should correct', () => {
+    jest.useFakeTimers();
+
+    const wrapper = mount(
+      <Menu mode="horizontal" openKeys={['light']} disabledOverflow>
+        <SubMenu key="light">
+          <SubMenu key="bamboo" />
+        </SubMenu>
+      </Menu>,
+    );
+
+    act(() => {
+      jest.runAllTimers();
+      wrapper.update();
+    });
+
+    expect(
+      wrapper
+        .find('li.rc-menu-submenu')
+        .first()
+        .hasClass('rc-menu-submenu-horizontal'),
+    ).toBeTruthy();
+    expect(
+      wrapper
+        .find('li.rc-menu-submenu')
+        .last()
+        .hasClass('rc-menu-submenu-vertical'),
+    ).toBeTruthy();
+
+    jest.useRealTimers();
+  });
 });
 /* eslint-enable */