feature: add version to generated source (#1186)

* feature: add version to generated source * Automatically linting code --------- Co-authored-by: Rocket Understudy <[email protected]>
RocketSurgeonsGuild · Apr 1, 2024 · 6f364ca · 6f364ca
1 parent 7bd3107
commit 6f364ca
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 13 deletions.
diff --git a/src/Testing.AutoFixtures/Attribute.cs b/src/Testing.AutoFixtures/Attribute.cs
@@ -2,21 +2,22 @@ namespace Rocket.Surgery.Extensions.Testing.AutoFixtures;
 
 internal static class Attribute
 {
-    public const string Source = @"#nullable enable
+    public static readonly string source = @$"#nullable enable
 using System;
 using System.Diagnostics;
 
 namespace Rocket.Surgery.Extensions.Testing.AutoFixtures
-{
+{{
+    {AutoFixtures.CodeGenerationAttribute}
     [AttributeUsage(AttributeTargets.Class)]
     [Conditional(""CODEGEN"")]
     internal class AutoFixtureAttribute : Attribute
-    {
-        public AutoFixtureAttribute() : this(null) {}
+    {{
+        public AutoFixtureAttribute() : this(null) {{}}
 
         public AutoFixtureAttribute(Type? type) => Type = type;
 
-        public Type? Type { get; }
-    }
-}";
+        public Type? Type {{ get; }}
+    }}
+}}";
 }
diff --git a/src/Testing.AutoFixtures/AutoFixtureGenerator.cs b/src/Testing.AutoFixtures/AutoFixtureGenerator.cs
@@ -26,7 +26,7 @@ public void Initialize(IncrementalGeneratorInitializationContext context)
         context.RegisterPostInitializationOutput(
             initializationContext =>
             {
-                initializationContext.AddSource("AutoFixtureAttribute.g.cs", Attribute.Source);
+                initializationContext.AddSource("AutoFixtureAttribute.g.cs", Attribute.source);
                 initializationContext.AddSource($"{nameof(AutoFixtureBase)}.g.cs", AutoFixtureBase.Source);
             }
         );
@@ -108,10 +108,12 @@ void generateFixtureBuilder(
                 usingDirectives.Add(substituteMetadata.ContainingNamespace.ToDisplayString());
             }
 
-            var usingDirectiveSyntax = usingDirectives
-                                      .OrderBy(usingDirective => usingDirective, NamespaceComparer.Default)
-                                      .Select(x => UsingDirective(ParseName(x)))
-                                      .ToArray();
+            var usingDirectiveSyntax =
+                usingDirectives
+                   .OrderBy(usingDirective => usingDirective, NamespaceComparer.Default)
+                   .Select(x => UsingDirective(ParseName(x)))
+                   .ToArray();
+
             var unit =
                 CompilationUnit()
                    .AddUsings(usingDirectiveSyntax)

diff --git a/src/Testing.AutoFixtures/AutoFixtures.cs b/src/Testing.AutoFixtures/AutoFixtures.cs
@@ -0,0 +1,9 @@
+using System.Diagnostics;
+
+namespace Rocket.Surgery.Extensions.Testing.AutoFixtures;
+
+internal static class AutoFixtures
+{
+    public static string Version { get; } = FileVersionInfo.GetVersionInfo(typeof(AutoFixtureGenerator).Assembly.Location).ProductVersion;
+    public static string CodeGenerationAttribute { get; } = $@"[System.CodeDom.Compiler.GeneratedCode(""AutoFixtures"", ""{Version}"")]";
+}
diff --git a/test/Testing.AutoFixtures.Tests/AutoFixtureGeneratorTests.cs b/test/Testing.AutoFixtures.Tests/AutoFixtureGeneratorTests.cs
@@ -21,7 +21,7 @@ public async Task GivenAutoFixture_WhenGenerate_ThenShouldGenerateAutoFixtureAtt
         var result = await generatorInstance.GenerateAsync();
 
         // Then
-        await Verify(result);
+        await Verify(result).ScrubLines(text => text.Contains("System.CodeDom.Compiler.GeneratedCode"));
     }
 
     [Fact]