fix: search with enter not trigger correct change params (#253)

* fix: search with enter should give correct value

* test: update test case

Author: zombieJ

src/OptionList/useKeyboard.ts

@@ -1,8 +1,9 @@
 import * as React from 'react';
 import type { RefOptionListProps } from 'rc-select/lib/OptionList';
+import { useBaseProps } from 'rc-select';
 import KeyCode from 'rc-util/lib/KeyCode';
 import type { DefaultOptionType, InternalFieldNames, SingleValueType } from '../Cascader';
-import { useBaseProps } from 'rc-select';
+import { SEARCH_MARK } from '../hooks/useSearchOptions';
 
 export default (
   ref: React.Ref<RefOptionListProps>,
@@ -151,7 +152,18 @@ export default (
         // >>> Select
         case KeyCode.ENTER: {
           if (validActiveValueCells.length) {
-            onKeyBoardSelect(validActiveValueCells, lastActiveOptions[lastActiveIndex]);
+            const option = lastActiveOptions[lastActiveIndex];
+
+            // Search option should revert back of origin options
+            const originOptions: DefaultOptionType[] = option?.[SEARCH_MARK] || [];
+            if (originOptions.length) {
+              onKeyBoardSelect(
+                originOptions.map(opt => opt[fieldNames.value]),
+                originOptions[originOptions.length - 1],
+              );
+            } else {
+              onKeyBoardSelect(validActiveValueCells, lastActiveOptions[lastActiveIndex]);
+            }
           }
           break;
         }

tests/keyboard.spec.js

@@ -7,9 +7,11 @@ import { addressOptions } from './demoOptions';
 describe('Cascader.Keyboard', () => {
   let wrapper;
   let selectedValue;
+  let selectedOptions;
   let menus;
-  const onChange = value => {
+  const onChange = (value, options) => {
     selectedValue = value;
+    selectedOptions = options;
   };
 
   beforeEach(() => {
@@ -18,6 +20,7 @@ describe('Cascader.Keyboard', () => {
 
   afterEach(() => {
     selectedValue = null;
+    selectedOptions = null;
     menus = null;
   });
 
@@ -73,6 +76,24 @@ describe('Cascader.Keyboard', () => {
     wrapper.find('input').simulate('keyDown', { which: KeyCode.ENTER });
     expect(wrapper.isOpen()).toBeFalsy();
     expect(selectedValue).toEqual(['zj', 'hangzhou', 'yuhang']);
+    expect(selectedOptions).toEqual([
+      addressOptions[1],
+      addressOptions[1].children[0],
+      addressOptions[1].children[0].children[0],
+    ]);
+  });
+
+  it('enter on search', () => {
+    wrapper.find('input').simulate('change', { target: { value: '余杭' } });
+    wrapper.find('input').simulate('keyDown', { which: KeyCode.DOWN });
+    wrapper.find('input').simulate('keyDown', { which: KeyCode.ENTER });
+
+    expect(selectedValue).toEqual(['zj', 'hangzhou', 'yuhang']);
+    expect(selectedOptions).toEqual([
+      addressOptions[1],
+      addressOptions[1].children[0],
+      addressOptions[1].children[0].children[0],
+    ]);
   });
 
   it('rtl', () => {