Enabled asserting partial string comparisons of generated source.

Author: Arlodotexe

CommunityToolkit.Tooling.SampleGen.Tests/Helpers/TestHelpers.cs

@@ -6,7 +6,7 @@ namespace CommunityToolkit.Tooling.SampleGen.Tests.Helpers;
 public static partial class TestHelpers
 {
     internal static SourceGeneratorRunResult RunSourceGenerator<TGenerator>(this string source, string assemblyName, string markdown = "") where TGenerator : class, IIncrementalGenerator, new() => RunSourceGenerator<TGenerator>(source.ToSyntaxTree(), assemblyName, markdown);
-    
+
     internal static SourceGeneratorRunResult RunSourceGenerator<TGenerator>(this SyntaxTree syntaxTree, string assemblyName, string markdown = "")
         where TGenerator : class, IIncrementalGenerator, new()
     {
@@ -42,17 +42,19 @@ internal static void AssertNoCompilationErrors(this Compilation outputCompilatio
         Assert.IsTrue(generatedCompilationDiagnostics.All(x => x.Severity != DiagnosticSeverity.Error), $"Expected no generated compilation errors. Got: \n{string.Join("\n", generatedCompilationDiagnostics.Where(x => x.Severity == DiagnosticSeverity.Error).Select(x => $"[{x.Id}: {x.GetMessage()}]"))}");
     }
 
-    internal static void AssertSourceGenerated(this Compilation compilation, string filename, string expectedContents)
+    internal static string GetFileContentsByName(this Compilation compilation, string filename)
     {
         var generatedTree = compilation.SyntaxTrees.SingleOrDefault(tree => Path.GetFileName(tree.FilePath) == filename);
-
         Assert.IsNotNull(generatedTree, $"No file named {filename} was generated");
-        Assert.AreEqual(expectedContents, generatedTree.ToString(), "Unexpected code generated");
+
+        return generatedTree.ToString();
+    }
+
+    internal static void AssertSourceGenerated(this Compilation compilation, string filename, string expectedContents)
+    {
     }
 
     internal static void AssertDiagnosticsAre(this SourceGeneratorRunResult result, params DiagnosticDescriptor[] expectedDiagnosticDescriptors) => AssertDiagnosticsAre(result.Diagnostics, expectedDiagnosticDescriptors);
 
     internal static void AssertNoCompilationErrors(this SourceGeneratorRunResult result) => AssertNoCompilationErrors(result.Compilation);
-
-    internal static void AssertSourceGenerated(this SourceGeneratorRunResult result, string filename, string expectedContents) => AssertSourceGenerated(result.Compilation, filename, expectedContents);
 }

CommunityToolkit.Tooling.SampleGen.Tests/ToolkitSampleGeneratedPaneTests.cs

@@ -80,17 +80,19 @@ public class UserControl { }
             .CreateCompilation("MyApp.Samples")
             .ToMetadataReference();
 
-        // Get all current referenced assemblies + our generated sample project.
-        var headReferences = TestHelpers.GetAllReferencedAssemblies().Concat(new[] { sampleProjectAssembly });
-        var headCompilation = string.Empty.ToSyntaxTree().CreateCompilation("MyApp.Head", headReferences);
+        // Create application head that references generated sample project
+        var headCompilation = string.Empty
+            .ToSyntaxTree()
+            .CreateCompilation("MyApp.Head", TestHelpers.GetAllReferencedAssemblies())
+            .AddReferences(sampleProjectAssembly);
 
         // Run source generator
         var result = headCompilation.RunSourceGenerator<ToolkitSampleMetadataGenerator>();
 
         result.AssertDiagnosticsAre();
         result.AssertNoCompilationErrors();
 
-        result.AssertSourceGenerated(filename: "ToolkitSampleRegistry.g.cs", expectedContents: """
+        Assert.AreEqual(result.Compilation.GetFileContentsByName("ToolkitSampleRegistry.g.cs"), """
         #nullable enable
         namespace CommunityToolkit.Tooling.SampleGen;
 
@@ -101,7 +103,7 @@ public static class ToolkitSampleRegistry
                 ["Sample"] = new CommunityToolkit.Tooling.SampleGen.Metadata.ToolkitSampleMetadata("Sample", "Test Sample", "", typeof(MyApp.Sample), () => new MyApp.Sample(), null, null, new CommunityToolkit.Tooling.SampleGen.Metadata.IGeneratedToolkitSampleOptionViewModel[] { new CommunityToolkit.Tooling.SampleGen.Metadata.ToolkitSampleNumericOptionMetadataViewModel(name: "TextSize", initial: 12, min: 8, max: 48, step: 2, showAsNumberBox: false, title: "FontSize") })
             };
         }
-        """);
+        """, "Unexpected code generated");
     }
 
     [TestMethod]
@@ -385,7 +387,7 @@ public class UserControl {{ }}
     [TestMethod]
     public void GeneratedPaneOption_ButtonAction()
     {
-        var syntaxTree = $@"
+        var sampleProjectAssembly = $@"
             using System.ComponentModel;
             using CommunityToolkit.Tooling.SampleGen;
             using CommunityToolkit.Tooling.SampleGen.Attributes;
@@ -405,19 +407,24 @@ private void RaiseNotification()
             namespace Windows.UI.Xaml.Controls
             {{
                 public class UserControl {{ }}
-            }}".ToSyntaxTree();
-
+            }}"
+            .ToSyntaxTree()
+            .CreateCompilation("MyApp.Samples")
+            .ToMetadataReference();
 
-        // Create compilation builder with custom assembly name
-        var compilation = syntaxTree.CreateCompilation("MyApp.Tests");
+        // Create application head that references generated sample project
+        var headCompilation = string.Empty
+            .ToSyntaxTree()
+            .CreateCompilation("MyApp.Head", TestHelpers.GetAllReferencedAssemblies())
+            .AddReferences(sampleProjectAssembly);
 
         // Run source generator
-        var result = compilation.RunSourceGenerator<ToolkitSampleMetadataGenerator>();
+        var result = headCompilation.RunSourceGenerator<ToolkitSampleMetadataGenerator>();
 
         result.AssertDiagnosticsAre();
         result.AssertNoCompilationErrors();
 
-        result.AssertSourceGenerated(filename: "ToolkitSampleRegistry.g.cs", expectedContents: """
+        Assert.AreEqual(result.Compilation.GetFileContentsByName("ToolkitSampleRegistry.g.cs"), """
         #nullable enable
         namespace CommunityToolkit.Tooling.SampleGen;
 
@@ -428,6 +435,6 @@ public static class ToolkitSampleRegistry
                 ["Sample"] = new CommunityToolkit.Tooling.SampleGen.Metadata.ToolkitSampleMetadata("Sample", "Test Sample", "", typeof(MyApp.Sample), () => new MyApp.Sample(), null, null, new CommunityToolkit.Tooling.SampleGen.Metadata.IGeneratedToolkitSampleOptionViewModel[] { new CommunityToolkit.Tooling.SampleGen.Metadata.ToolkitSampleNumericOptionMetadataViewModel(name: "TextSize", initial: 12, min: 8, max: 48, step: 2, showAsNumberBox: false, title: "FontSize") })
             };
         }
-        """);
+        """, "Unexpected code generated");
     }
 }