PaletteEdit: Fix flaky test (WordPress#61791)

Co-authored-by: tyxla <tyxla@git.wordpress.org> Co-authored-by: cbravobernal <cbravobernal@git.wordpress.org>
patil-vipul · Jun 17, 2024 · d5097c8 · d5097c8
1 parent 6f2d635
commit d5097c8
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/packages/components/src/palette-edit/test/index.tsx b/packages/components/src/palette-edit/test/index.tsx
@@ -170,11 +170,13 @@ describe( 'PaletteEdit', () => {
 			} )
 		);
 
-		expect(
-			screen.getByRole( 'button', {
-				name: 'Remove all colors',
-			} )
-		).toBeVisible();
+		await waitFor( () => {
+			expect(
+				screen.getByRole( 'button', {
+					name: 'Remove all colors',
+				} )
+			).toBeVisible();
+		} );
 	} );
 
 	it( 'shows a reset option when the `canReset` prop is enabled', async () => {