Merge pull request #79 from amis92/feature/document-transform-tests

DocumentTransform unit tests

Author: AArnott

src/CodeGeneration.Roslyn.Tests.Generators/EmptyPartialAttribute.cs

@@ -0,0 +1,15 @@
+// Copyright (c) Andrew Arnott. All rights reserved.
+// Licensed under the MIT license. See LICENSE.txt file in the project root for full license information.
+
+namespace CodeGeneration.Roslyn.Tests.Generators
+{
+    using System;
+    using System.Diagnostics;
+    
+    [AttributeUsage(AttributeTargets.All, Inherited = false)]
+    [CodeGenerationAttribute(typeof(EmptyPartialGenerator))]
+    [Conditional("CodeGeneration")]
+    public class EmptyPartialAttribute : Attribute
+    {
+    }
+}

src/CodeGeneration.Roslyn.Tests.Generators/EmptyPartialGenerator.cs

@@ -0,0 +1,48 @@
+// Copyright (c) Andrew Arnott. All rights reserved.
+// Licensed under the MIT license. See LICENSE.txt file in the project root for full license information.
+
+namespace CodeGeneration.Roslyn.Tests.Generators
+{
+    using System;
+    using System.Threading;
+    using System.Threading.Tasks;
+    using Microsoft.CodeAnalysis;
+    using Microsoft.CodeAnalysis.CSharp;
+    using Microsoft.CodeAnalysis.CSharp.Syntax;
+
+    public class EmptyPartialGenerator : ICodeGenerator
+    {
+
+        public EmptyPartialGenerator(AttributeData attributeData)
+        {
+        }
+
+        public Task<SyntaxList<MemberDeclarationSyntax>> GenerateAsync(TransformationContext context, IProgress<Diagnostic> progress, CancellationToken cancellationToken)
+        {
+            if (!(context.ProcessingNode is TypeDeclarationSyntax typeDeclaration))
+            {
+                return Task.FromResult(SyntaxFactory.List<MemberDeclarationSyntax>());
+            }
+
+            var partial = typeDeclaration is StructDeclarationSyntax structDeclaration
+                ? StructPartial(structDeclaration)
+                : ClassPartial((ClassDeclarationSyntax) typeDeclaration);
+            var results = SyntaxFactory.SingletonList(partial);
+            return Task.FromResult(results);
+        }
+
+        private static MemberDeclarationSyntax ClassPartial(ClassDeclarationSyntax declaration)
+        {
+            return SyntaxFactory.ClassDeclaration(declaration.Identifier)
+                .WithTypeParameterList(declaration.TypeParameterList)
+                .WithModifiers(SyntaxTokenList.Create(SyntaxFactory.Token(SyntaxKind.PartialKeyword)));
+        }
+
+        private static MemberDeclarationSyntax StructPartial(StructDeclarationSyntax declaration)
+        {
+            return SyntaxFactory.StructDeclaration(declaration.Identifier)
+                .WithTypeParameterList(declaration.TypeParameterList)
+                .WithModifiers(SyntaxTokenList.Create(SyntaxFactory.Token(SyntaxKind.PartialKeyword)));
+        }
+    }
+}

src/CodeGeneration.Roslyn.Tests/CodeGeneration.Roslyn.Tests.csproj

@@ -14,6 +14,7 @@
   </ItemGroup>
 
   <ItemGroup>
+    <PackageReference Include="Microsoft.CodeAnalysis.CSharp.Workspaces" Version="2.2.0" />
     <PackageReference Include="Microsoft.CSharp" Version="4.4.1" />
     <PackageReference Include="Xunit" Version="2.3.1" />
     <PackageReference Include="Xunit.runner.visualstudio" Version="2.3.1" />