Merge pull request alibaba#1659 from miracles1919/master

test(useDrag): add some tests

jest.config.js

@@ -10,4 +10,11 @@ module.exports = {
       tsconfig: 'tsconfig.json',
     },
   },
+  collectCoverageFrom: [
+    '<rootDir>/**/src/**/*.{js,jsx,ts,tsx}',
+    '!**/demo/**',
+    '!**/example/**',
+    '!**/es/**',
+    '!**/lib/**',
+  ],
 };

packages/hooks/src/useDrag/__tests__/index.test.ts

@@ -0,0 +1,62 @@
+import { renderHook } from '@testing-library/react-hooks';
+import useDrag, { Options } from '../index';
+import type { BasicTarget } from '../../utils/domTarget';
+
+const setup = <T>(data: T, target: BasicTarget, options?: Options) =>
+  renderHook(() => useDrag(data, target, options));
+
+const events = {};
+const mockTarget = {
+  addEventListener: jest.fn((event, callback) => {
+    events[event] = callback;
+  }),
+  removeEventListener: jest.fn((event) => {
+    Reflect.deleteProperty(events, event);
+  }),
+  setAttribute: jest.fn(),
+};
+
+describe('useDrag', () => {
+  it('should be defined', () => {
+    expect(useDrag).toBeDefined();
+  });
+
+  it('should add/remove listener on mount/unmount', () => {
+    const { unmount } = setup(1, mockTarget as any);
+    expect(mockTarget.addEventListener).toBeCalled();
+    expect(mockTarget.addEventListener.mock.calls[0][0]).toBe('dragstart');
+    expect(mockTarget.addEventListener.mock.calls[1][0]).toBe('dragend');
+    expect(mockTarget.setAttribute).toBeCalledWith('draggable', 'true');
+    unmount();
+    expect(mockTarget.removeEventListener).toBeCalled();
+  });
+
+  it('should triggle drag callback', () => {
+    const onDragStart = jest.fn();
+    const onDragEnd = jest.fn();
+    const mockEvent = {
+      dataTransfer: {
+        setData: jest.fn(),
+      },
+    };
+    setup(1, mockTarget as any, {
+      onDragStart,
+      onDragEnd,
+    });
+    events['dragstart'](mockEvent);
+    expect(onDragStart).toBeCalled();
+    expect(mockEvent.dataTransfer.setData).toBeCalledWith('custom', '1');
+    events['dragend'](mockEvent);
+    expect(onDragEnd).toBeCalled();
+  });
+
+  it(`should not work when target don't support addEventListener method`, () => {
+    Object.defineProperty(mockTarget, 'addEventListener', {
+      get() {
+        return false;
+      },
+    });
+    setup(1, mockTarget as any);
+    expect(mockTarget.setAttribute).not.toBeCalled();
+  });
+});