Add support for collection initializers

Author: aromaa

SourceGenerator/SourceGenerator.csproj

@@ -11,7 +11,7 @@
 
   <PropertyGroup>
     <Description>Source Generator for adding missing methods and classes inside tests to allow compilation. Assists on creating code exercises.</Description>
-    <Version>1.1.2</Version>
+    <Version>1.1.3</Version>
   </PropertyGroup>
 
   <ItemGroup>

SourceGenerator/TestGenerator.cs

@@ -131,16 +131,28 @@ private void ProcessStatement(SourceGeneratorContext context, ref Compilation co
 
         private void ProcessObjectCreation(SourceGeneratorContext context, ref Compilation compilation, NamespaceDeclarationSyntax @namespace, ClassDeclarationSyntax @class, ObjectCreationExpressionSyntax objectCreation)
         {
+            if (!(objectCreation.Initializer is null))
+            {
+                foreach (ExpressionSyntax initializerExpression in objectCreation.Initializer.Expressions)
+                {
+                    if (initializerExpression is ObjectCreationExpressionSyntax initializerObjectCreation)
+                    {
+                        this.GenerateEmptyClass(context, ref compilation, @namespace, initializerObjectCreation.Type.ToString());
+                        this.ProcessObjectCreation(context, ref compilation, @namespace, @class, initializerObjectCreation);
+                    }
+                }
+            }
+
             //Ignore constructors without arguments
-            if (objectCreation.ArgumentList?.Arguments.Count <= 0)
+            if (objectCreation.ArgumentList is null || objectCreation.ArgumentList.Arguments.Count <= 0)
             {
                 return;
             }
 
             SemanticModel methodModel = compilation.GetSemanticModel(objectCreation.SyntaxTree);
             SymbolInfo targetSymbol = methodModel.GetSymbolInfo(objectCreation.Type);
 
-            this.ProcessArguments(context, ref compilation, @namespace, @class, objectCreation.ArgumentList!.Arguments);
+            this.ProcessArguments(context, ref compilation, @namespace, @class, objectCreation.ArgumentList.Arguments);
 
             //This is used for generating constructors only (for now, at least), type should exist already
             if (targetSymbol.Symbol is null)
@@ -390,6 +402,10 @@ private void ProcessInvocation(SourceGeneratorContext context, ref Compilation c
                     break;
                 default:
                 {
+                    if (targetIdentifier.ToString() == "Test")
+                    {
+                            return default;
+                    }
                     //Ehh... Lets assume its a missing static class? Ehheheh...
                     ITypeSymbol symbol = this.GenerateEmptyClass(context, ref compilation, @namespace, targetIdentifier.ToString());
 

Tests/Instance/InstanceMethodTests.cs

@@ -113,5 +113,17 @@ public void GenerateMissingInstantiationArguments()
                 MissingClass3 missingClass = new MissingClass3(MissingClass3Static.MissingArgument);
             });
         }
+
+        [Fact]
+        public void GenerateCollectionInitializerObjects()
+        {
+            Assert.Throws<NotImplementedException>(() =>
+            {
+                List<object> list = new List<object>
+                {
+                    new MissingCollectionInitializerClass("First")
+                };
+            });
+        }
     }
 }