Stabilize browser context warmup bootstrap

Author: KSemenenko

tests/PrompterOne.Web.UITests/Infrastructure/StandaloneAppFixture.Warmup.cs

@@ -50,18 +50,28 @@ await page.EvaluateAsync(
         await page.EvaluateAsync(BrowserTestLibrarySeedData.CreateInitializationScript());
     }
 
-    private static async Task WarmUpContextPageIfNeededAsync(IPage page, string baseAddress)
+    private static async Task WarmUpContextPageIfNeededAsync(
+        IPage page,
+        string baseAddress,
+        bool warmAllRuntimeRoutes = false)
     {
         if (!TestEnvironment.IsCiEnvironment)
         {
             return;
         }
 
         var browserErrors = BrowserErrorCollector.Attach(page);
+        var warmupRoutes = warmAllRuntimeRoutes
+            ? RuntimeWarmupRoutes
+            : [RuntimeWarmupRoutes[0]];
 
         try
         {
-            await WarmUpRouteAsync(page, BrowserTestConstants.Routes.Library, UiTestIds.Library.Page);
+            foreach (var (route, pageTestId) in warmupRoutes)
+            {
+                await WarmUpRouteAsync(page, route, pageTestId);
+            }
+
             await PrimeIsolatedBrowserStorageAsync(page, baseAddress);
             await browserErrors.AssertNoCriticalUiErrorsAsync();
         }

tests/PrompterOne.Web.UITests/Infrastructure/StandaloneAppFixture.cs

@@ -8,6 +8,7 @@ namespace PrompterOne.Web.UITests;
 public sealed partial class StandaloneAppFixture : IAsyncInitializer, IAsyncDisposable
 {
     private const int MinimumPageCount = 1;
+    private const int ContextBootstrapAttemptCount = 3;
     private const int ServerStartupTimeoutSeconds = 60;
     private const int ServerProbeDelayMilliseconds = 500;
     private readonly ConcurrentDictionary<IBrowserContext, byte> _contexts = [];
@@ -72,7 +73,7 @@ public async Task<IReadOnlyList<IPage>> NewSharedPagesAsync(
 
     private async Task<IPage> CreateAdditionalPageAsync()
     {
-        while (true)
+        for (var attempt = 1; ; attempt++)
         {
             var context = await CreateTrackedContextAsync();
 
@@ -84,7 +85,11 @@ private async Task<IPage> CreateAdditionalPageAsync()
                 await WarmUpContextPageIfNeededAsync(page, BaseAddress);
                 return page;
             }
-            catch (PlaywrightException exception) when (IsBrowserClosedException(exception))
+            catch (PlaywrightException exception) when (attempt < ContextBootstrapAttemptCount && IsBrowserClosedException(exception))
+            {
+                await DisposeContextAsync(context);
+            }
+            catch (InvalidOperationException exception) when (attempt < ContextBootstrapAttemptCount && IsContextWarmupFailure(exception))
             {
                 await DisposeContextAsync(context);
             }
@@ -93,7 +98,7 @@ private async Task<IPage> CreateAdditionalPageAsync()
 
     private async Task<IPage> CreateSharedPageAsync(string contextKey)
     {
-        while (true)
+        for (var attempt = 1; ; attempt++)
         {
             var (context, isNewSharedContext) = await GetOrCreateSharedContextAsync(contextKey);
 
@@ -105,7 +110,7 @@ private async Task<IPage> CreateSharedPageAsync(string contextKey)
                 if (isNewSharedContext)
                 {
                     await PrimeIsolatedBrowserStorageAsync(page);
-                    await WarmUpContextPageIfNeededAsync(page, BaseAddress);
+                    await WarmUpContextPageIfNeededAsync(page, BaseAddress, warmAllRuntimeRoutes: true);
                 }
                 else
                 {
@@ -114,7 +119,11 @@ private async Task<IPage> CreateSharedPageAsync(string contextKey)
 
                 return page;
             }
-            catch (PlaywrightException exception) when (IsBrowserClosedException(exception))
+            catch (PlaywrightException exception) when (attempt < ContextBootstrapAttemptCount && IsBrowserClosedException(exception))
+            {
+                RemoveSharedContext(context);
+            }
+            catch (InvalidOperationException exception) when (attempt < ContextBootstrapAttemptCount && IsContextWarmupFailure(exception))
             {
                 RemoveSharedContext(context);
             }
@@ -229,6 +238,9 @@ private static bool IsBrowserClosedException(PlaywrightException exception) =>
         exception.Message.Contains("Target page, context or browser has been closed", StringComparison.Ordinal)
         || exception.Message.Contains("Process exited", StringComparison.Ordinal);
 
+    private static bool IsContextWarmupFailure(InvalidOperationException exception) =>
+        exception.Message.StartsWith("Browser context warmup failed.", StringComparison.Ordinal);
+
     private Task PrimeIsolatedBrowserStorageAsync(IPage page) =>
         PrimeIsolatedBrowserStorageAsync(page, BaseAddress);