fix: synchronize code that reads/writes to the internal render tree i…

…n TestRenderer/Renderer

CHANGELOG.md

@@ -10,7 +10,6 @@ All notable changes to **bUnit** will be documented in this file. The project ad
 ### Changed 
 
 - Changed test renderer such that updates to rendered components markup happen in the same synchronization context as the test framework is using (if any), if any, to avoid memory race conditions. By [@egil](https://github.com/egil).
-- Changed default "WaitFor" timeout to 10 seconds. By [@egil](https://github.com/egil).
 
 ## [1.18.4] - 2023-02-26
 

src/bunit.core/Rendering/TestRenderer.cs

@@ -8,6 +8,7 @@ namespace Bunit.Rendering;
 /// </summary>
 public class TestRenderer : Renderer, ITestRenderer
 {
+	private readonly object renderTreeUpdateLock = new();
 	private readonly SynchronizationContext? usersSyncContext = SynchronizationContext.Current;
 	private readonly Dictionary<int, IRenderedFragmentBase> renderedComponents = new();
 	private readonly List<RootComponent> rootComponents = new();
@@ -80,33 +81,39 @@ public Task DispatchEventAsync(
 		if (fieldInfo is null)
 			throw new ArgumentNullException(nameof(fieldInfo));
 
-		var result = Dispatcher.InvokeAsync(() =>
+		// Calling base.DispatchEventAsync updates the render tree
+		// if the event contains associated data.
+		lock (renderTreeUpdateLock)
 		{
-			ResetUnhandledException();
-
-			try
-			{
-				return base.DispatchEventAsync(eventHandlerId, fieldInfo, eventArgs);
-			}
-			catch (ArgumentException ex) when (string.Equals(ex.Message, $"There is no event handler associated with this event. EventId: '{eventHandlerId}'. (Parameter 'eventHandlerId')", StringComparison.Ordinal))
+			var result = Dispatcher.InvokeAsync(() =>
 			{
-				if (ignoreUnknownEventHandlers)
+				ResetUnhandledException();
+
+				try
 				{
-					return Task.CompletedTask;
+					return base.DispatchEventAsync(eventHandlerId, fieldInfo, eventArgs);
 				}
+				catch (ArgumentException ex) when (string.Equals(ex.Message, $"There is no event handler associated with this event. EventId: '{eventHandlerId}'. (Parameter 'eventHandlerId')", StringComparison.Ordinal))
+				{
+					if (ignoreUnknownEventHandlers)
+					{
+						return Task.CompletedTask;
+					}
+
+					var betterExceptionMsg = new UnknownEventHandlerIdException(eventHandlerId, fieldInfo, ex);
+					return Task.FromException(betterExceptionMsg);
+				}
+			});
 
-				var betterExceptionMsg = new UnknownEventHandlerIdException(eventHandlerId, fieldInfo, ex);
-				return Task.FromException(betterExceptionMsg);
+			if (result.IsFaulted && result.Exception is not null)
+			{
+				HandleException(result.Exception);
 			}
-		});
 
-		if (result.IsFaulted && result.Exception is not null)
-		{
-			HandleException(result.Exception);
-		}
+			AssertNoUnhandledExceptions();
 
-		AssertNoUnhandledExceptions();
-		return result;
+			return result;
+		}
 	}
 
 	/// <inheritdoc/>
@@ -150,6 +157,19 @@ public void DisposeComponents()
 		AssertNoUnhandledExceptions();
 	}
 
+	/// <inheritdoc/>
+	protected override void ProcessPendingRender()
+	{
+		// Blocks updates to the renderers internal render tree
+		// while the render tree is being read elsewhere.
+		// base.ProcessPendingRender calls UpdateDisplayAsync,
+		// so there is no need to lock in that method.
+		lock (renderTreeUpdateLock)
+		{
+			base.ProcessPendingRender();
+		}
+	}
+
 	/// <inheritdoc/>
 	protected override Task UpdateDisplayAsync(in RenderBatch renderBatch)
 	{
@@ -281,7 +301,12 @@ private IReadOnlyList<IRenderedComponentBase<TComponent>> FindComponents<TCompon
 		var result = new List<IRenderedComponentBase<TComponent>>();
 		var framesCollection = new RenderTreeFrameDictionary();
 
-		FindComponentsInRenderTree(parentComponent.ComponentId);
+		// Blocks the renderer from changing the render tree
+		// while this method searches through it.
+		lock (renderTreeUpdateLock)
+		{
+			FindComponentsInRenderTree(parentComponent.ComponentId);
+		}
 
 		return result;
 

tests/bunit.web.tests/Extensions/WaitForHelpers/RenderedFragmentWaitForElementsHelperExtensions.Async.Test.cs

@@ -5,7 +5,7 @@ namespace Bunit.Extensions.WaitForHelpers;
 
 public class RenderedFragmentWaitForElementsHelperExtensionsAsyncTest : TestContext
 {
-    private readonly static TimeSpan WaitForTestTimeout = TimeSpan.FromMilliseconds(5);
+    private readonly static TimeSpan WaitForTestTimeout = TimeSpan.FromMilliseconds(100);
 
     public RenderedFragmentWaitForElementsHelperExtensionsAsyncTest(ITestOutputHelper testOutput)
 	{