Stabilize ImageComparatorTests

aquality-automation · Aug 30, 2024 · eda97e6 · eda97e6
1 parent d28171d
commit eda97e6
Showing 1 changed file with 13 additions and 4 deletions.
diff --git a/...ty.Selenium.Core/tests/Aquality.Selenium.Core.Tests/Visualization/ImageComparatorTests.cs b/...ty.Selenium.Core/tests/Aquality.Selenium.Core.Tests/Visualization/ImageComparatorTests.cs
@@ -7,6 +7,9 @@
 using NUnit.Framework;
 using OpenQA.Selenium;
 using System;
+using Aquality.Selenium.Core.Utilities;
+using SkiaSharp;
+using System.Collections.Generic;
 
 namespace Aquality.Selenium.Core.Tests.Visualization
 {
@@ -66,10 +69,16 @@ public void Should_BePossibleTo_GetPercentageDifference_ForDifferentElements_Wit
         [Test]
         public void Should_BePossibleTo_GetPercentageDifference_ForSimilarElements()
         {
-            StartLoading();
-            var firstImage = LoadingLabel.GetElement().GetScreenshot().AsImage();
-            AqualityServices.ServiceProvider.GetRequiredService<IConditionalWait>().WaitFor(() => firstImage.Height < LoadingLabel.Visual.Size.Height);
-            var secondImage = LoadingLabel.GetElement().GetScreenshot().AsImage();
+            SKImage firstImage = null, secondImage = null;
+            AqualityServices.ServiceProvider.GetRequiredService<IActionRetrier>().DoWithRetry(() =>
+            {
+                AqualityServices.Application.Driver.Navigate().Refresh();
+                StartLoading();
+                firstImage = LoadingLabel.GetElement().GetScreenshot().AsImage();
+                AqualityServices.ServiceProvider.GetRequiredService<IConditionalWait>().WaitFor(() => firstImage.Height < LoadingLabel.Visual.Size.Height);
+                secondImage = LoadingLabel.GetElement().GetScreenshot().AsImage();
+            }, new List<Type> { typeof(WebDriverException)});
+
             Assert.Multiple(() =>
             {
                 Assert.That(ImageComparator.PercentageDifference(firstImage, secondImage, threshold: 0), Is.Not.EqualTo(0));