Added ability to take over parameter values

Author: linkdotnet

src/bunit.generators/Web.Stubs/StubGenerator.cs

@@ -128,7 +128,7 @@ private static bool GenerateStubComponent(StubClassInfo classInfo, SourceProduct
 		var sourceBuilder = new StringBuilder();
 
 		sourceBuilder.AppendLine($"namespace {classInfo.TargetTypeNamespace};");
-		sourceBuilder.AppendLine($"public partial class {classInfo.StubClassName} : Microsoft.AspNetCore.Components.ComponentBase");
+		sourceBuilder.AppendLine($"internal partial class {classInfo.StubClassName} : Microsoft.AspNetCore.Components.ComponentBase");
 		sourceBuilder.Append("{");
 
 		foreach (var member in targetTypeSymbol
@@ -147,13 +147,9 @@ private static bool GenerateStubComponent(StubClassInfo classInfo, SourceProduct
 			var propertyType = member.Type.ToDisplayString();
 			var propertyName = member.Name;
 
-			var isParameterAttribute = member.GetAttributes().Any(attr =>
-				attr.AttributeClass?.ToDisplayString() == "Microsoft.AspNetCore.Components.ParameterAttribute");
-			var attributeLine = isParameterAttribute
-				? "\t[global::Microsoft.AspNetCore.Components.Parameter]"
-				: "\t[global::Microsoft.AspNetCore.Components.CascadingParameter]";
-
+			var attributeLine = GetAttributeLine(member);
 			sourceBuilder.AppendLine(attributeLine);
+
 			sourceBuilder.AppendLine($"\tpublic {propertyType} {propertyName} {{ get; set; }}");
 		}
 
@@ -165,6 +161,39 @@ private static bool GenerateStubComponent(StubClassInfo classInfo, SourceProduct
 		}
 
 		return hasSomethingToStub;
+
+		string GetAttributeLine(IPropertySymbol member)
+		{
+			var attribute = member.GetAttributes().First(attr =>
+				attr.AttributeClass?.ToDisplayString() == "Microsoft.AspNetCore.Components.ParameterAttribute" ||
+				attr.AttributeClass?.ToDisplayString() == "Microsoft.AspNetCore.Components.CascadingParameterAttribute");
+
+			var attributeLine = new StringBuilder("\t[");
+			if (attribute.AttributeClass?.ToDisplayString() == "Microsoft.AspNetCore.Components.ParameterAttribute")
+			{
+				attributeLine.Append("global::Microsoft.AspNetCore.Components.Parameter");
+				var captureUnmatchedValuesArg = attribute.NamedArguments
+					.FirstOrDefault(arg => arg.Key == "CaptureUnmatchedValues").Value;
+				if (captureUnmatchedValuesArg.Value is bool captureUnmatchedValues)
+				{
+					var captureString = captureUnmatchedValues ? "true" : "false";
+					attributeLine.Append($"(CaptureUnmatchedValues = {captureString})");
+				}
+			}
+			else if (attribute.AttributeClass?.ToDisplayString() ==
+			         "Microsoft.AspNetCore.Components.CascadingParameterAttribute")
+			{
+				attributeLine.Append("global::Microsoft.AspNetCore.Components.CascadingParameter");
+				var nameArg = attribute.NamedArguments.FirstOrDefault(arg => arg.Key == "Name").Value;
+				if (!nameArg.IsNull)
+				{
+					attributeLine.Append($"(Name = \"{nameArg.Value}\")");
+				}
+			}
+
+			attributeLine.Append("]");
+			return attributeLine.ToString();
+		}
 	}
 
 	private static void GenerateInterceptorCode(StubClassInfo stubbedComponentGroup, IEnumerable<StubClassInfo> stubClassGrouped, SourceProductionContext context)

tests/bunit.generators.tests/Web.Stub/Components/CounterComponent.cs

@@ -0,0 +1,10 @@
+using Microsoft.AspNetCore.Components;
+
+namespace Bunit.Web.Stub.Components;
+
+public class CounterComponent : ComponentBase
+{
+	[Parameter] public int Count { get; set; }
+	[CascadingParameter(Name = "Cascading")] public int CascadingCount { get; set; }
+	[Parameter(CaptureUnmatchedValues = true)] public Dictionary<string, object> UnmatchedValues { get; set; }
+}

…ators.tests/Web.Stub/CounterComponent.cs …s/Web.Stub/Components/ParentComponent.cstests/bunit.generators.tests/Web.Stub/CounterComponent.cs renamed to tests/bunit.generators.tests/Web.Stub/Components/ParentComponent.cs tests/bunit.generators.tests/Web.Stub/CounterComponent.cs renamed to tests/bunit.generators.tests/Web.Stub/Components/ParentComponent.cs

@@ -1,7 +1,7 @@
 using Microsoft.AspNetCore.Components;
 using Microsoft.AspNetCore.Components.Rendering;
 
-namespace Bunit.Web.Stub;
+namespace Bunit.Web.Stub.Components;
 
 public class ParentComponent : ComponentBase
 {
@@ -12,10 +12,3 @@ protected override void BuildRenderTree(RenderTreeBuilder builder)
 		builder.CloseComponent();
 	}
 }
-
-public class CounterComponent : ComponentBase
-{
-	[Parameter] public int Count { get; set; }
-	[CascadingParameter] public int CascadingCount { get; set; }
-	[Parameter] public EventCallback IncrementCount { get; set; }
-}

tests/bunit.generators.tests/Web.Stub/StubTests.cs

@@ -1,3 +1,8 @@
+using System.Reflection;
+using Bunit.Web.Stub.Components;
+using Microsoft.AspNetCore.Components;
+using Microsoft.Extensions.DependencyInjection;
+
 namespace Bunit.Web.Stub;
 
 public class StubTests : TestContext
@@ -14,12 +19,17 @@ public void Stubbed_component_has_same_parameters()
 	}
 
 	[Fact]
-	public void Stubbed_component_can_be_used_in_multiple_tests()
+	public void Stubbed_component_has_same_parameter_values()
 	{
-		ComponentFactories.AddGeneratedStub<CounterComponent>();
-
-		var cut = RenderComponent<ParentComponent>();
+		var type = typeof(CounterComponentStub);
+		var cascadingParameterValue = type.GetProperty(nameof(CounterComponentStub.CascadingCount))
+			?.GetCustomAttribute<CascadingParameterAttribute>()
+			?.Name;
+		var captureUnmatchedValues = type.GetProperty(nameof(CounterComponentStub.UnmatchedValues))
+			?.GetCustomAttribute<ParameterAttribute>()
+			?.CaptureUnmatchedValues;
 
-		Assert.True(cut.HasComponent<CounterComponentStub>());
+		Assert.Equal("Cascading", cascadingParameterValue);
+		Assert.True(captureUnmatchedValues);
 	}
 }