WIP _ PLEASE SQUASH ME!

Author: linkdotnet

src/bunit.core/Rendering/TestRenderer.cs

@@ -255,7 +255,29 @@ protected override IComponent ResolveComponentForRenderMode(Type componentType,
 			return renderModeAttribute.Mode;
 		}
 
-		var parentComponentState = GetComponentState(component).ParentComponentState;
+		var componentState = GetComponentState(component);
+		// Should this be LogicalComponentState instead?
+		if (componentState.ParentComponentState is not null)
+		{
+			var parentFrames = GetCurrentRenderTreeFrames(componentState.ParentComponentState.ComponentId);
+			var foundComponentStart = false;
+			for (var i = 0; i < parentFrames.Count; i++)
+			{
+				ref var frame = ref parentFrames.Array[i];
+
+				if (frame.FrameType == RenderTreeFrameType.Component)
+				{
+					foundComponentStart = frame.ComponentId == componentState.ComponentId;
+				}
+				else if (foundComponentStart
+				         && frame.FrameType == RenderTreeFrameType.ComponentRenderMode)
+				{
+					return frame.ComponentRenderMode;
+				}
+			}
+		}
+
+		var parentComponentState = componentState.ParentComponentState;
 		return parentComponentState is not null
 			? GetComponentRenderMode(parentComponentState.Component)
 			: null;

….core.tests/Rendering/RenderModeTests.cs …re.tests/Rendering/RenderModeTests.razortests/bunit.core.tests/Rendering/RenderModeTests.cs renamed to tests/bunit.core.tests/Rendering/RenderModeTests.razor tests/bunit.core.tests/Rendering/RenderModeTests.cs renamed to tests/bunit.core.tests/Rendering/RenderModeTests.razor

@@ -1,20 +1,22 @@
+@{
 #if NET9_0_OR_GREATER
-using Bunit.TestAssets.RenderModes;
+}
 
-namespace Bunit.Rendering;
+@using Bunit.TestAssets.RenderModes;
+@inherits TestContext
 
-public class RenderModeTests : TestContext
-{
+@code {
+#if NET9_0_OR_GREATER
 	[Fact(DisplayName = "TestRenderer provides RendererInfo")]
 	public void Test001()
 	{
 		Renderer.SetRendererInfo(new RendererInfo("Server", true));
 		var cut = RenderComponent<RendererInfoComponent>();
 
-		cut.MarkupMatches("""
+		cut.MarkupMatches(@"
 		                  <p>Is interactive: True</p>
 		                  <p>Rendermode: Server</p>
-		                  """);
+		                  ");
 	}
 
 	[Fact(DisplayName = "Renderer throws exception if RendererInfo is not specified")]
@@ -39,10 +41,10 @@ public void Test004()
 		var cut = RenderComponent<ComponentWithoutRenderMode>(
 			c => c.AddChildContent<ComponentWithWebAssemblyRenderMode>());
 
-		cut.MarkupMatches("""
+		cut.MarkupMatches(@"
 		                  <div>Parent assigned render mode: </div>
 		                  <div>Assigned render mode: InteractiveWebAssemblyRenderMode</div>
-		                  """);
+		                  ");
 	}
 
 	[Fact(DisplayName = "Parent and child render mode is specified")]
@@ -51,10 +53,10 @@ public void Test005()
 		var cut = RenderComponent<ComponentWithWebAssemblyRenderMode>(
 			c => c.AddChildContent<ComponentWithServerRenderMode>());
 
-		cut.MarkupMatches("""
+		cut.MarkupMatches(@"
 		                  <div>Parent assigned render mode: InteractiveWebAssemblyRenderMode</div>
 		                  <div>Assigned render mode: InteractiveServerRenderMode</div>
-		                  """);
+		                  ");
 	}
 
 	[Fact(DisplayName = "Parent and child render mode is not specified")]
@@ -63,11 +65,44 @@ public void Test006()
 		var cut = RenderComponent<ComponentWithoutRenderMode>(
 			c => c.AddChildContent<ComponentWithoutRenderMode>());
 
-		cut.MarkupMatches("""
+		cut.MarkupMatches(@"
 		                  <div>Parent assigned render mode: </div>
 		                  <div>Assigned render mode: </div>
-		                  """);
-		
+		                  ");
+	}
+	
+	[Fact(DisplayName = "Parent that sets render mode to children is specified on child")]
+	public void Test007()
+	{
+		var cut = RenderComponent<ComponentWithChildContent>(
+			p => p.AddChildContent(
+				@<ComponentThatPrintsAssignedRenderMode @rendermode="RenderMode.InteractiveServer"/>));
+
+		cut.MarkupMatches(@<p>Assigned Render Mode: InteractiveServerRenderMode</p>);
+	}
+
+	[Fact(DisplayName = "Assigned Render Mode is inherited all the way down the component hierarchy")]
+	public void Test008()
+	{
+		var cut = RenderComponent<ComponentWithChildContent>(
+			p => p.AddChildContent(
+				@<ComponentWithChildContent @rendermode="RenderMode.InteractiveServer">
+					<ComponentThatPrintsAssignedRenderMode/>
+				</ComponentWithChildContent>));
+
+		cut.MarkupMatches(@<p>Assigned Render Mode: InteractiveServerRenderMode</p>);
+	}
+
+	[Fact(DisplayName = "Having a component with section outlet and RenderMode is specifing for child component")]
+	public void Test009()
+	{
+		// See: https://learn.microsoft.com/en-us/aspnet/core/blazor/components/sections?view=aspnetcore-8.0#section-interaction-with-other-blazor-features
+		var cut = Render(@<SectionOutletComponent/>);
+
+		cut.MarkupMatches(@<p>Assigned Render Mode: InteractiveWebAssemblyRenderMode</p>);
 	}
+#endif
 }
-#endif
+@{
+#endif
+}

tests/bunit.testassets/RenderModes/ComponentThatPrintsAssignedRenderMode.razor

@@ -0,0 +1,9 @@
+@{
+#if NET9_0_OR_GREATER
+}
+
+<p>Assigned Render Mode: @AssignedRenderMode?.GetType().Name</p>
+
+@{
+#endif
+}

tests/bunit.testassets/RenderModes/ComponentWithChildContent.razor

@@ -0,0 +1,6 @@
+@ChildContent
+@code {
+
+	[Parameter] public RenderFragment ChildContent { get; set; } = default!;
+
+}	

tests/bunit.testassets/RenderModes/SectionOutletComponent.razor

@@ -0,0 +1,11 @@
+@{
+#if NET9_0_OR_GREATER
+}
+
+<Microsoft.AspNetCore.Components.Sections.SectionOutlet SectionId="1" @rendermode="RenderMode.InteractiveWebAssembly" />
+<Microsoft.AspNetCore.Components.Sections.SectionContent SectionId="1">
+	<ComponentThatPrintsAssignedRenderMode/>
+</Microsoft.AspNetCore.Components.Sections.SectionContent>
+@{
+#endif
+}