Throw for returned rejected promises

Author: DAreRodz

packages/e2e-tests/plugins/interactive-blocks/generator-scope/render.php

@@ -15,4 +15,5 @@
 	<button type="button" data-wp-on--click="callbacks.reject" data-testid="reject">Async reject</button>
 	<button type="button" data-wp-on--click="callbacks.capture" data-testid="capture">Async capture</button>
 	<button type="button" data-wp-on--click="callbacks.captureThrow" data-testid="captureThrow">Async captureThrow</button>
+	<button type="button" data-wp-on--click="callbacks.captureReturnReject" data-testid="captureReturnReject">Async captureReturnReject</button>
 </div>

packages/e2e-tests/plugins/interactive-blocks/generator-scope/view.js

@@ -39,5 +39,15 @@ const { callbacks } = store( 'test/generator-scope', {
 				getContext().result = err.toString();
 			}
 		},
+		*returnReject() {
+			return Promise.reject( new Error( '🔚' ) );
+		},
+		*captureReturnReject() {
+			try {
+				yield callbacks.returnReject();
+			} catch ( err ) {
+				getContext().result = err.toString();
+			}
+		},
 	},
 } );

packages/interactivity/src/utils.ts

@@ -163,7 +163,11 @@ export function withScope( func: ( ...args: unknown[] ) => unknown ) {
 					error = e;
 				}
 				if ( it.done ) {
-					break;
+					if ( error ) {
+						throw error;
+					} else {
+						break;
+					}
 				}
 			}
 

test/e2e/specs/interactivity/async-actions.spec.ts

@@ -48,4 +48,14 @@ test.describe( 'async actions', () => {
 		await page.getByTestId( 'captureThrow' ).click();
 		await expect( resultInput ).toHaveValue( 'Error: 🤯' );
 	} );
+
+	test( 'Promise generator callbacks should throw when rejected promises are returned', async ( {
+		page,
+	} ) => {
+		const resultInput = page.getByTestId( 'result' );
+		await expect( resultInput ).toHaveValue( '' );
+
+		await page.getByTestId( 'captureReturnReject' ).click();
+		await expect( resultInput ).toHaveValue( 'Error: 🔚' );
+	} );
 } );