Ensure image in stream test actually loads before checking color

dotnet · Dec 20, 2023 · c0478f4 · c0478f4
1 parent 388088b
commit c0478f4
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/Controls/tests/DeviceTests/Elements/Image/ImageTests.Android.cs b/src/Controls/tests/DeviceTests/Elements/Image/ImageTests.Android.cs
@@ -9,6 +9,7 @@
 using Microsoft.Maui.Graphics;
 using Microsoft.Maui.Handlers;
 using Xunit;
+using static Microsoft.Maui.DeviceTests.AssertHelpers;
 
 namespace Microsoft.Maui.DeviceTests
 {
@@ -58,8 +59,13 @@ public async Task ImageSetFromStreamRenders()
 			await InvokeOnMainThreadAsync(async () =>
 			{
 				var handler = CreateHandler<LayoutHandler>(layout);
-				await image.WaitUntilLoaded();
-				await handler.ToPlatform().AssertContainsColor(Colors.Red, MauiContext);
+				var rootView = handler.ToPlatform();
+
+				await rootView.AttachAndRun(async () =>
+				{
+					await image.WaitUntilLoaded();
+					await rootView.AssertContainsColor(Colors.Red, MauiContext);
+				});
 			});
 		}
 	}