refactor: use early return and simplify call

Author: linkdotnet

src/bunit.core/Rendering/RootRenderTree.cs

@@ -91,16 +91,9 @@ public RenderFragment Wrap(RenderFragment target)
 	public int GetCountOf<TComponent>()
 		where TComponent : IComponent
 	{
-		var result = 0;
 		var countType = typeof(TComponent);
 
-		for (int i = 0; i < registrations.Count; i++)
-		{
-			if (countType == registrations[i].ComponentType)
-				result++;
-		}
-
-		return result;
+		return registrations.Count(t => countType == t.ComponentType);
 	}
 
 	private static RenderFragment<RenderFragment> CreateRenderFragmentBuilder<TComponent>(Action<ComponentParameterCollectionBuilder<TComponent>>? parameterBuilder)

src/bunit.core/Rendering/TestRenderer.cs

@@ -157,7 +157,7 @@ protected override Task UpdateDisplayAsync(in RenderBatch renderBatch)
 		logger.LogNewRenderBatchReceived();
 
 		RenderCount++;
-		
+
 		var renderEvent = new RenderEvent(renderBatch, new RenderTreeFrameDictionary());
 
 		// removes disposed components
@@ -299,19 +299,15 @@ void FindComponentsInRenderTree(int componentId)
 	IRenderedComponentBase<TComponent> GetOrCreateRenderedComponent<TComponent>(RenderTreeFrameDictionary framesCollection, int componentId, TComponent component)
 		where TComponent : IComponent
 	{
-		IRenderedComponentBase<TComponent> result;
-
 		if (renderedComponents.TryGetValue(componentId, out var renderedComponent))
 		{
-			result = (IRenderedComponentBase<TComponent>)renderedComponent;
-		}
-		else
-		{
-			LoadRenderTreeFrames(componentId, framesCollection);
-			result = activator.CreateRenderedComponent(componentId, component, framesCollection);
-			renderedComponents.Add(result.ComponentId, result);
+			return (IRenderedComponentBase<TComponent>)renderedComponent;
 		}
 
+		LoadRenderTreeFrames(componentId, framesCollection);
+		var result = activator.CreateRenderedComponent(componentId, component, framesCollection);
+		renderedComponents.Add(result.ComponentId, result);
+
 		return result;
 	}