Moved extension method into generator

Author: linkdotnet

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

@@ -17,6 +17,29 @@ public class StubGenerator : IIncrementalGenerator
 	/// <inheritdoc/>
 	public void Initialize(IncrementalGeneratorInitializationContext context)
 	{
+		context.RegisterPostInitializationOutput(
+			ctx => ctx.AddSource("AddGeneratedStub.g.cs",
+				@"namespace Bunit
+{
+    public static class ComponentFactoriesExtensions
+	{
+        /// <summary>
+		/// Marks a component as a stub component so that a stub gets generated for it. The stub has the same name as the component, but with the suffix ""Stub"" added.
+		/// </summary>
+		/// <typeparam name=""TComponent"">The type of component to generate a stub for.</typeparam>
+		/// <remarks>
+		/// When <c>ComponentFactories.AddGeneratedStub&lt;MyButton&gt;()</c> is called, a stub component is generated for the component
+		/// with the name <c>MyButtonStub</c>. The stub component is added to the <see cref=""ComponentFactoryCollection""/> and can be used.
+		/// It can also be retrieved via `cut.FindComponent&lt;MyButtonStub&gt;()`.
+		/// This call does the same as <c>ComponentFactories.Add&lt;MyButton, MyButtonStub&gt;()</c>.
+		/// </remarks>
+		public static ComponentFactoryCollection AddGeneratedStub<TComponent>(this ComponentFactoryCollection factories)
+			where TComponent : Microsoft.AspNetCore.Components.IComponent
+		{
+			return factories.AddGeneratedStubInterceptor<TComponent>();
+		}
+	}
+}"));
 		var classesToStub = context.SyntaxProvider
 			.CreateSyntaxProvider(
 				predicate: static (s, _) => s is InvocationExpressionSyntax,

src/bunit.web/Extensions/StubComponentFactoryCollectionExtensions.cs

@@ -174,7 +174,7 @@ public static ComponentFactoryCollection AddStub(
 	/// This call does the same as <c>ComponentFactories.Add&lt;MyButton, MyButtonStub&gt;()</c>.
 	/// </remarks>
 #pragma warning disable S2326 // Unused type parameters should be removed - this is just the shape for the interceptor
-	public static ComponentFactoryCollection AddGeneratedStub<TComponent>(this ComponentFactoryCollection factories)
+	public static ComponentFactoryCollection AddGeneratedStub2<TComponent>(this ComponentFactoryCollection factories)
 		where TComponent : IComponent
 #pragma warning restore S2326
 	{