fix: keep active scroll source during drag

Author: zombieJ

src/Table.tsx

@@ -369,9 +369,12 @@ const Table = <RecordType extends DefaultRecordType>(
   const scrollBodyContainerRef = React.useRef<HTMLDivElement>(null);
   const [hoverEle, setHoverEle] = React.useState<ScrollSource | null>(null);
   const onHoverScrollSource = React.useCallback(
-    (source: ScrollSource) => () => {
-      setHoverEle(source);
-    },
+    (source: ScrollSource): React.MouseEventHandler<HTMLElement> =>
+      event => {
+        if (event.buttons === 0) {
+          setHoverEle(source);
+        }
+      },
     [],
   );
   const onLeaveTable = React.useCallback(() => {
@@ -519,7 +522,7 @@ const Table = <RecordType extends DefaultRecordType>(
         sticky: undefined,
       });
 
-      if (source && hoverEle && scrollSource && hoverEle !== scrollSource) {
+      if (source === 'body' && hoverEle && hoverEle !== 'body') {
         return;
       }
 

tests/Virtual.spec.tsx

@@ -263,6 +263,41 @@ describe('Table.Virtual', () => {
     expect(setScrollLeft).not.toHaveBeenCalled();
   });
 
+  it('should not change hovered scroll source while dragging', () => {
+    const { container } = getTable();
+
+    resize(container.querySelector('.rc-table')!);
+
+    fireEvent.mouseEnter(container.querySelector('.rc-table-header')!);
+    fireEvent.mouseEnter(container.querySelector('.rc-table-tbody-virtual')!, {
+      buttons: 1,
+    });
+
+    setScrollLeft.mockClear();
+
+    fireEvent.wheel(container.querySelector('.rc-table-tbody-virtual-holder')!, {
+      deltaX: 10,
+    });
+
+    expect(setScrollLeft).not.toHaveBeenCalled();
+  });
+
+  it('should process direct header scroll when body is hovered', () => {
+    const { container } = getTable();
+
+    resize(container.querySelector('.rc-table')!);
+
+    fireEvent.mouseEnter(container.querySelector('.rc-table-tbody-virtual')!);
+
+    setScrollLeft.mockClear();
+
+    fireEvent.wheel(container.querySelector('.rc-table-header')!, {
+      deltaX: 10,
+    });
+
+    expect(setScrollLeft).toHaveBeenCalled();
+  });
+
   it('should reset hovered scroll source when leaving table', () => {
     const { container } = getTable();
     const table = container.querySelector('.rc-table') as HTMLElement;