Update UseStaticGenericMapperStaticMethodFromAnotherAssembly

riok · Jul 26, 2024 · b5ee4f9 · b5ee4f9
1 parent 1b2d702
commit b5ee4f9
Showing 1 changed file with 26 additions and 7 deletions.
diff --git a/test/Riok.Mapperly.Tests/Mapping/UseStaticMapperTest.cs b/test/Riok.Mapperly.Tests/Mapping/UseStaticMapperTest.cs
@@ -363,8 +363,8 @@ public partial class Mapper
     }
 
     [Theory]
-    [InlineData(true)]
-    [InlineData(false)]
+    [InlineData(true)] // Usually when your IDE runs the source generator (references are other syntax trees)
+    [InlineData(false)] // When compiler produces final assembly (references are compiled assemblies)
     public void UseStaticGenericMapperStaticMethodFromAnotherAssembly(bool asCompilationReference)
     {
         var testDependencySource = TestSourceBuilder.SyntaxTree(
@@ -416,20 +416,39 @@ public class Target
             }
             """
         );
-        TestHelper
+        var result = TestHelper
             .GenerateMapper(source, TestHelperOptions.AllowDiagnostics, additionalAssemblies: [testDependencyAssembly])
-            .Should()
-            .HaveDiagnostic(DiagnosticDescriptors.QueryableProjectionMappingCannotInline)
-            .HaveMethodBody(
+            .Should();
+
+        if (asCompilationReference)
+        {
+            result.HaveMethodBody(
                 "ProjectToTarget",
                 """
                 #nullable disable
                         return System.Linq.Queryable.Select(source, x => new global::Mapper.Target()
                         {
-                            DateTime = global::Riok.Mapperly.TestDependency.Mapper.DateTimeMapper.MapToDateTimeOffset(x.DateTime),
+                            DateTime = new global::System.DateTimeOffset(x.DateTime, global::System.TimeSpan.Zero),
                         });
                 #nullable enable
                 """
             );
+        }
+        else
+        {
+            result
+                .HaveDiagnostic(DiagnosticDescriptors.QueryableProjectionMappingCannotInline)
+                .HaveMethodBody(
+                    "ProjectToTarget",
+                    """
+                    #nullable disable
+                            return System.Linq.Queryable.Select(source, x => new global::Mapper.Target()
+                            {
+                                DateTime = global::Riok.Mapperly.TestDependency.Mapper.DateTimeMapper.MapToDateTimeOffset(x.DateTime),
+                            });
+                    #nullable enable
+                    """
+                );
+        }
     }
 }