From 20bf6dcde542145c0a02e5eceb2d4bdec38fe2bd Mon Sep 17 00:00:00 2001
From: remax21 <deronzier.remi@gmail.com>
Date: Tue, 6 May 2025 21:56:09 +0200
Subject: [PATCH] feat(await images): :sparkles: manage images from shape
 decoration

---
 lib/src/helpers/await_images.dart | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/lib/src/helpers/await_images.dart b/lib/src/helpers/await_images.dart
index 11d5b7e..13eeaeb 100644
--- a/lib/src/helpers/await_images.dart
+++ b/lib/src/helpers/await_images.dart
@@ -30,12 +30,14 @@ extension AwaitImages on WidgetTester {
       for (final element in find.byType(DecoratedBox).evaluate().toList()) {
         final widget = element.widget as DecoratedBox;
         final decoration = widget.decoration;
-        if (decoration is BoxDecoration) {
-          final image = decoration.image?.image;
-          if (image != null) {
-            await precacheImage(image, element);
-            await pump();
-          }
+        final image = switch (decoration) {
+          BoxDecoration() => decoration.image?.image,
+          ShapeDecoration() => decoration.image?.image,
+          _ => null,
+        };
+        if (image != null) {
+          await precacheImage(image, element);
+          await pump();
         }
       }
     });