refactor(ImageUploader): get complete upload data

src/components/image-uploader/image-uploader.tsx

@@ -1,22 +1,22 @@
-import React, { forwardRef, useRef, useState, useImperativeHandle } from 'react'
+import { useIsomorphicLayoutEffect, useSize, useUnmount } from 'ahooks'
+import { AddOutline, CloseOutline } from 'antd-mobile-icons'
 import type {
-  ReactNode,
-  InputHTMLAttributes,
   CSSProperties,
+  InputHTMLAttributes,
   ReactElement,
+  ReactNode,
 } from 'react'
-import { AddOutline, CloseOutline } from 'antd-mobile-icons'
-import { mergeProps } from '../../utils/with-default-props'
-import ImageViewer, { ImageViewerShowHandler } from '../image-viewer'
-import PreviewItem from './preview-item'
-import { usePropsValue } from '../../utils/use-props-value'
-import { useIsomorphicLayoutEffect, useUnmount, useSize } from 'ahooks'
-import Space from '../space'
-import { NativeProps, withNativeProps } from '../../utils/native-props'
+import React, { forwardRef, useImperativeHandle, useRef, useState } from 'react'
 import { measureCSSLength } from '../../utils/measure-css-length'
+import { NativeProps, withNativeProps } from '../../utils/native-props'
+import { usePropsValue } from '../../utils/use-props-value'
+import { mergeProps } from '../../utils/with-default-props'
 import { useConfig } from '../config-provider'
-import type { ImageProps } from '../image'
 import Grid, { GridProps } from '../grid'
+import type { ImageProps } from '../image'
+import ImageViewer, { ImageViewerShowHandler } from '../image-viewer'
+import Space from '../space'
+import PreviewItem from './preview-item'
 
 export type TaskStatus = 'pending' | 'fail' | 'success'
 
@@ -225,10 +225,10 @@ export const ImageUploader = forwardRef<ImageUploaderRef, ImageUploaderProps>(
                 return task
               })
             })
-            throw e
+            console.error(e)
           }
         })
-      ).catch(error => console.error(error))
+      )
       setValue(prev => prev.concat(newVal))
     }
 

src/components/image-uploader/tests/image-uploader.test.tsx

@@ -1,14 +1,14 @@
 import React, { createRef, forwardRef, useState } from 'react'
 import {
+  act,
+  cleanup,
+  fireEvent,
   render,
+  screen,
+  sleep,
   testA11y,
-  fireEvent,
-  waitFor,
   userEvent,
-  sleep,
-  screen,
-  cleanup,
-  act,
+  waitFor,
   waitForElementToBeRemoved,
 } from 'testing'
 import ImageUploader, { ImageUploadItem, ImageUploaderRef } from '..'
@@ -65,19 +65,24 @@ describe('ImageUploader', () => {
   })
 
   const App = forwardRef<ImageUploaderRef, any>((props, ref) => {
+    const { onChange: propsOnChange, ...restProps } = props
     const [fileList, setFileList] = useState<ImageUploadItem[]>([
       {
         url: demoSrc,
       },
     ])
+    const onChange = (newFileList: ImageUploadItem[]) => {
+      setFileList(newFileList)
+      propsOnChange?.(newFileList)
+    }
 
     return (
       <ImageUploader
         ref={ref}
         value={fileList}
-        onChange={setFileList}
+        onChange={onChange}
         upload={mockUpload}
-        {...props}
+        {...restProps}
       />
     )
   })
@@ -389,4 +394,36 @@ describe('ImageUploader', () => {
     expect(ref.current).toBeDefined()
     expect(ref.current?.nativeElement).toBeDefined()
   })
+
+  test('get all upload url', async () => {
+    const fn = jest.fn()
+    const mockUpload = mockUploadWithFailure(2)
+
+    render(<App multiple upload={mockUpload} onChange={fn} />)
+
+    mockInputFile([
+      new File(['one'], 'one.png', { type: 'image/png' }),
+      new File(['two'], 'two.png', { type: 'image/png' }),
+      new File(['three'], 'three.png', { type: 'image/png' }),
+    ])
+
+    await act(async () => {
+      jest.runAllTimers()
+    })
+
+    expect(fn.mock.lastCall[0].length).toBe(4)
+
+    function mockUploadWithFailure(failOnCount: number) {
+      let count = 0
+      return async () => {
+        count++
+        if (count === failOnCount) {
+          throw new Error('Fail to upload')
+        }
+        return {
+          url: 'count: ' + count,
+        }
+      }
+    }
+  })
 })