fix: restore load switcher after clearing search

Author: DuQi

src/OptionList.tsx

@@ -138,8 +138,21 @@ const OptionList: React.ForwardRefRenderFunction<ReviseRefOptionListProps> = (_,
     if (treeExpandedKeys) {
       return [...treeExpandedKeys];
     }
-    return searchValue ? searchExpandedKeys : expandedKeys;
-  }, [expandedKeys, searchExpandedKeys, treeExpandedKeys, searchValue]);
+    if (searchValue) {
+      return searchExpandedKeys;
+    }
+    if (searchExpandedKeys && loadData && !treeDefaultExpandAll) {
+      return expandedKeys || [];
+    }
+    return expandedKeys;
+  }, [
+    expandedKeys,
+    searchExpandedKeys,
+    treeExpandedKeys,
+    searchValue,
+    loadData,
+    treeDefaultExpandAll,
+  ]);
 
   const onInternalExpand = (keys: Key[]) => {
     setExpandedKeys(keys);
@@ -315,8 +328,7 @@ const OptionList: React.ForwardRefRenderFunction<ReviseRefOptionListProps> = (_,
     () => (searchValue ? false : true),
     // eslint-disable-next-line react-hooks/exhaustive-deps
     [searchValue, treeExpandedKeys || expandedKeys],
-    ([preSearchValue], [nextSearchValue, nextExcludeSearchExpandedKeys]) =>
-      preSearchValue !== nextSearchValue && !!(nextSearchValue || nextExcludeSearchExpandedKeys),
+    ([preSearchValue], [nextSearchValue]) => preSearchValue !== nextSearchValue,
   );
 
   const syncLoadData = hasLoadDataFn ? loadData : null;

tests/Select.loadData.spec.tsx

@@ -43,4 +43,28 @@ describe('TreeSelect.loadData', () => {
       ).toHaveLength(2 + i);
     }
   });
+
+  it('keeps load switcher after clearing search value', () => {
+    const loadData = jest.fn(() => Promise.resolve());
+    const { container } = render(
+      <TreeSelect
+        open
+        showSearch
+        treeDataSimpleMode
+        loadData={loadData}
+        treeData={[{ id: 1, pId: 0, value: '1', title: 'Parent' }]}
+      />,
+    );
+
+    const input = container.querySelector('input')!;
+
+    expect(container.querySelector('.rc-tree-select-tree-switcher_close')).toBeTruthy();
+
+    fireEvent.change(input, { target: { value: '1' } });
+    fireEvent.change(input, { target: { value: '' } });
+
+    expect(loadData).not.toHaveBeenCalled();
+    expect(container.querySelector('.rc-tree-select-tree-switcher_close')).toBeTruthy();
+    expect(container.querySelector('.rc-tree-select-tree-switcher-noop')).toBeFalsy();
+  });
 });