feat: narrow predicate

Author: linkdotnet

src/bunit.generators/README.md

@@ -74,19 +74,3 @@ internal partial class ThidPartyStub { }
 
 Current limitations of this approach:
  * The stubbed type is not allowed to be nested inside the test class.
-
-## Developer notes
-
-### Tips for developing with the generator
-
-When changing the source generator, to see the effect, clearing the build cache may be necessary:
-
-```
-dotnet build-server shutdown
-```
-
-A good way to quickly see if the generate is producing output:
-
-```
-dotnet build-server shutdown && dotnet clean && dotnet test -p:TargetFramework=net8.0
-```

src/bunit.generators/Web.Stubs/AddStubMethodStubGenerator/AddStubGenerator.cs

@@ -19,7 +19,13 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
 	{
 		var classesToStub = context.SyntaxProvider
 			.CreateSyntaxProvider(
-				predicate: static (s, _) => s is InvocationExpressionSyntax,
+				predicate: static (s, _) => s is InvocationExpressionSyntax
+				{
+					Expression: MemberAccessExpressionSyntax
+					{
+						Name.Identifier.Text: "AddStub"
+					}
+				},
 				transform: static (ctx, _) => GetStubClassInfo(ctx))
 			.Where(static m => m is not null)
 			.Collect();

src/bunit.generators/Web.Stubs/AttributeStubGenerator/StubAttributeGenerator.cs

@@ -23,7 +23,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
 		var classesToStub = context.SyntaxProvider
 			.ForAttributeWithMetadataName(
 				AttributeFullQualifiedName,
-				predicate: static (s, _) => s is ClassDeclarationSyntax,
+				predicate: static (s, _) => s is ClassDeclarationSyntax c && c.AttributeLists.SelectMany(a => a.Attributes).Any(at => at.Name.ToString() == "Stub"),
 				transform: static (ctx, _) => GetStubClassInfo(ctx))
 			.Where(static m => m is not null);
 

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

@@ -1,5 +1,4 @@
 using Bunit.Web.Stub.Components;
-using Microsoft.AspNetCore.Components;
 
 namespace Bunit.Web.Stub;