fix tests

loresoft · Aug 9, 2023 · 1900db6 · 1900db6
1 parent 3a2f1f6
commit 1900db6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/src/FluentCommand.Generators/DataReaderFactoryWriter.cs b/src/FluentCommand.Generators/DataReaderFactoryWriter.cs
@@ -8,7 +8,7 @@ public static class DataReaderFactoryWriter
     private static int _counter = 0;
 #endif
 
-    public static string Generate(EntityClass entityClass, bool skipVersion = false)
+    public static string Generate(EntityClass entityClass)
     {
         if (entityClass == null)
             throw new ArgumentNullException(nameof(entityClass));
@@ -38,12 +38,12 @@ public static string Generate(EntityClass entityClass, bool skipVersion = false)
             .AppendLine("/// Extension methods for FluentCommand")
             .AppendLine("/// </summary>");
 
-        //codeBuilder
-        //    .Append("[global::System.CodeDom.Compiler.GeneratedCode(\"")
-        //    .Append(ThisAssembly.Project.AssemblyName)
-        //    .Append("\", \"")
-        //    .Append(skipVersion ? "1.0.0.0" : ThisAssembly.Info.Version)
-        //    .AppendLine("\")]");
+        codeBuilder
+            .Append("[global::System.CodeDom.Compiler.GeneratedCode(\"")
+            .Append("FluentCommand.Generators")
+            .Append("\", \"")
+            .Append("1.0.0.0")
+            .AppendLine("\")]");
 
         codeBuilder
             .AppendLine("[global::System.Diagnostics.DebuggerNonUserCodeAttribute]")

diff --git a/test/FluentCommand.Generators.Tests/DataReaderFactoryWriterTests.cs b/test/FluentCommand.Generators.Tests/DataReaderFactoryWriterTests.cs
@@ -24,7 +24,7 @@ public async Task Generate()
             }.ToImmutableArray()
         );
 
-        var source = DataReaderFactoryWriter.Generate(entityClass, true);
+        var source = DataReaderFactoryWriter.Generate(entityClass);
 
         await Verifier
             .Verify(source)