Update setup media picker helper to handle single and multiple items

WordPress · May 5, 2023 · 92ef384 · 92ef384
1 parent 8379426
commit 92ef384
Showing 1 changed file with 21 additions and 10 deletions.
diff --git a/test/native/integration-test-helpers/setup-media-picker.js b/test/native/integration-test-helpers/setup-media-picker.js
@@ -20,9 +20,11 @@ import { requestMediaPicker } from '@wordpress/react-native-bridge';
  */
 export const setupMediaPicker = () => {
 	let mediaPickerCallback;
+	let multipleItems;
 	requestMediaPicker.mockImplementation(
 		( source, filter, multiple, callback ) => {
 			mediaPickerCallback = callback;
+			multipleItems = multiple;
 		}
 	);
 	return {
@@ -34,16 +36,25 @@ export const setupMediaPicker = () => {
 				mediaPickerCallback
 			),
 		mediaPickerCallback: async ( ...mediaItems ) =>
-			act( async () =>
+			act( async () => {
+				const items = mediaItems.map(
+					( {
+						localId,
+						localUrl,
+						type = 'image',
+						url,
+						id,
+						metadata,
+					} ) => ( {
+						type,
+						url: url ?? localUrl,
+						id: id ?? localId,
+						metadata,
+					} )
+				);
 				mediaPickerCallback(
-					mediaItems.map(
-						( { localId, localUrl, type = 'image' } ) => ( {
-							type,
-							url: localUrl,
-							id: localId,
-						} )
-					)
-				)
-			),
+					items.length === 1 && ! multipleItems ? items[ 0 ] : items
+				);
+			} ),
 	};
 };