Perf tweak

koenbeuk · Jun 14, 2021 · 389eec3 · 389eec3
1 parent 9e2cf04
commit 389eec3
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/ScenarioTests.Generator/ScenarioTestGenerator.cs b/src/ScenarioTests.Generator/ScenarioTestGenerator.cs
@@ -24,7 +24,7 @@ public void Initialize(GeneratorInitializationContext context) =>
         public void Execute(GeneratorExecutionContext context)
         {
             // retrieve the populated receiver 
-            if (context.SyntaxReceiver is not SyntaxReceiver receiver)
+            if (context.SyntaxReceiver is not SyntaxReceiver receiver || receiver.ScenarioCandidates is null)
             {
                 return;
             }

diff --git a/src/ScenarioTests.Generator/SyntaxReceiver.cs b/src/ScenarioTests.Generator/SyntaxReceiver.cs
@@ -7,7 +7,7 @@ namespace ScenarioTests.Generator
 {
     public class SyntaxReceiver : ISyntaxReceiver
     {
-        public List<MethodDeclarationSyntax> ScenarioCandidates { get; } = new List<MethodDeclarationSyntax>();
+        public List<MethodDeclarationSyntax> ScenarioCandidates { get; private set; }
 
         public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
         {
@@ -20,6 +20,11 @@ public void OnVisitSyntaxNode(SyntaxNode syntaxNode)
 
                 if (hasScenarioAttribute)
                 {
+                    if (ScenarioCandidates is null)
+                    {
+                        ScenarioCandidates = new();
+                    }
+
                     ScenarioCandidates.Add(methodDeclarationSyntax);
                 }
             }