feat: add test for image-picker

arco-design · Sep 20, 2023 · ee262e1 · ee262e1
1 parent fa911bb
commit ee262e1
Showing 1 changed file with 19 additions and 6 deletions.
diff --git a/packages/arcodesign/components/image-picker/__test__/index.spec.js b/packages/arcodesign/components/image-picker/__test__/index.spec.js
@@ -1,8 +1,9 @@
 import React from 'react';
-import { render, waitFor, waitForElementToBeRemoved } from '@testing-library/react';
+import { act, fireEvent, render, waitFor } from '@testing-library/react';
 import userEvent from '@testing-library/user-event';
 import demoTest from '../../../tests/demoTest';
 import mountTest from '../../../tests/mountTest';
+import { createStartTouchEventObject } from '../../../tests/helpers/mockEvent';
 import { defaultContext } from '../../context-provider';
 import ImagePicker from '..';
 import '@testing-library/jest-dom';
@@ -23,13 +24,15 @@ const mockImgFile = new File([IMG_DATA], 'img.png', { type: 'image/png' });
 
 describe('ImagePicker', () => {
     beforeEach(() => {
+        jest.useFakeTimers();
         jest.spyOn(global, 'FileReader').mockImplementation(function () {
             this.readAsDataURL = jest.fn();
             this.onload = jest.fn();
             this.onerror = jest.fn();
         });
     });
     afterEach(() => {
+        jest.useRealTimers();
         jest.restoreAllMocks();
     });
     it('ImagePicker renders correctly', () => {
@@ -161,6 +164,20 @@ describe('ImagePicker', () => {
         userEvent.click(container.querySelector(`.${imagePrefix}`));
         expect(handleClick).toBeCalledTimes(1);
     });
+    it('press image correctly', () => {
+        const handleLongPress = jest.fn();
+        const { container } = render(
+            <ImagePicker images={[{ url: IMG_URL }]} onLongPress={handleLongPress} />,
+        );
+        act(() => {
+            fireEvent.touchStart(
+                container.querySelector(`.${prefix}-image-container`),
+                createStartTouchEventObject({ x: 0, y: 0 }),
+            );
+            jest.advanceTimersByTime(1000);
+        });
+        expect(handleLongPress).toBeCalledTimes(1);
+    });
     it('selectAdapter', async () => {
         const handleMaxSizeExceed = jest.fn();
         const handleLimitExceed = jest.fn();
@@ -188,11 +205,7 @@ describe('ImagePicker', () => {
                             url: 'http://sf1-cdn-tos.toutiaostatic.com/obj/arco-mobile/_static_/large_image_3.jpg',
                         },
                     ];
-                    setTimeout(() => {
-                        resolve({
-                            files,
-                        });
-                    }, 1000);
+                    resolve({ files });
                 });
             };
             return (