Skip to content

Commit 18c2470

Browse files
committed
Update UsesVerifyGenerator.cs
1 parent 77977fe commit 18c2470

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/Verify.MSTest.SourceGenerator/UsesVerifyGenerator.cs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
6262
return null;
6363
}
6464

65-
cancel.ThrowIfCancellationRequested();
66-
67-
if (!symbol.HasAttributeOfType(TestClassAttributeName, includeDerived: true))
65+
if (HasTestClassAttribute(symbol))
6866
{
6967
return null;
7068
}
@@ -96,6 +94,8 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
9694
context.RegisterSourceOutput(toGenerate, Execute);
9795
}
9896

97+
static bool HasTestClassAttribute(INamedTypeSymbol symbol) => !symbol.HasAttributeOfType(TestClassAttributeName, includeDerived: true);
98+
9999
static bool IsSyntaxEligibleForGeneration(SyntaxNode node, Cancel _) =>
100100
node is ClassDeclarationSyntax;
101101

0 commit comments

Comments
 (0)