feat: add tests

riok · Nov 19, 2023 · 9e5ca4f · 9e5ca4f
1 parent 73b0e8d
commit 9e5ca4f
Show file tree

Hide file tree

Showing 7 changed files with 140 additions and 86 deletions.
diff --git a/test/Riok.Mapperly.Tests/Mapping/DerivedExistingTargetTypeTest.cs b/test/Riok.Mapperly.Tests/Mapping/DerivedExistingTargetTypeTest.cs
@@ -100,20 +100,6 @@ public Task WithInterfaceSourceAndTargetNullableShouldWork()
         return TestHelper.VerifyGenerator(source);
     }
 
-    [Fact]
-    public Task WithObjectShouldWork()
-    {
-        var source = TestSourceBuilder.MapperWithBody(
-            """
-            [MapDerivedType<string, int>]
-            [MapDerivedType<int, string>]
-            [MapDerivedType<IEnumerable<string>, IEnumerable<int>>]
-            public partial Map(object src, B trg);
-            """
-        );
-        return TestHelper.VerifyGenerator(source);
-    }
-
     [Fact]
     public void NotAssignableTargetTypeShouldDiagnostic()
     {
@@ -137,50 +123,6 @@ public void NotAssignableTargetTypeShouldDiagnostic()
             .HaveAssertedAllDiagnostics();
     }
 
-    [Fact]
-    public Task WithEnumerableOfInterfaceShouldWork()
-    {
-        var source = TestSourceBuilder.MapperWithBodyAndTypes(
-            """
-            public partial IEnumerable<B> Map(IEnumerable<A> source);
-
-            [MapDerivedType<AImpl1, BImpl1>]
-            [MapDerivedType<AImpl2, BImpl2>]
-            private partial B Map(A src);
-            """,
-            "interface A { string BaseValue { get; set; } }",
-            "interface B { string BaseValue { get; set; }}",
-            "class AImpl1 : A { public string BaseValue { get; set; } public string Value1 { get; set; } }",
-            "class AImpl2 : A { public string BaseValue { get; set; } public string Value2 { get; set; } }",
-            "class BImpl1 : B { public string BaseValue { get; set; } public string Value1 { get; set; } }",
-            "class BImpl2 : B { public string BaseValue { get; set; } public string Value2 { get; set; } }"
-        );
-        return TestHelper.VerifyGenerator(source);
-    }
-
-    [Fact]
-    public Task WithInterfacePropertyShouldWork()
-    {
-        var source = TestSourceBuilder.MapperWithBodyAndTypes(
-            """
-            public partial B Map(A source);
-
-            [MapDerivedType<AImpl1, BImpl1>]
-            [MapDerivedType<AImpl2, BImpl2>]
-            private partial BIntf Map(AIntf src);
-            """,
-            "class A { public AIntf Value { get; set; } }",
-            "class B { public BIntf Value { get; set; } }",
-            "interface AIntf { string BaseValue { get; set; } }",
-            "interface BIntf { string BaseValue { get; set; }}",
-            "class AImpl1 : AIntf { public string BaseValue { get; set; } public string Value1 { get; set; } }",
-            "class AImpl2 : AIntf { public string BaseValue { get; set; } public string Value2 { get; set; } }",
-            "class BImpl1 : BIntf { public string BaseValue { get; set; } public string Value1 { get; set; } }",
-            "class BImpl2 : BIntf { public string BaseValue { get; set; } public string Value2 { get; set; } }"
-        );
-        return TestHelper.VerifyGenerator(source);
-    }
-
     [Fact]
     public Task WithBaseTypeConfigShouldWork()
     {
@@ -189,31 +131,7 @@ public Task WithBaseTypeConfigShouldWork()
             [MapDerivedType<ASubType1, BSubType1>]
             [MapDerivedType<ASubType2, BSubType2>]
             [MapProperty(nameof(A.BaseValueA), nameof(B.BaseValueB)]
-            public partial B Map(A src);
-            """,
-            "abstract class A { public string BaseValueA { get; set; } }",
-            "abstract class B { public string BaseValueB { get; set; } }",
-            "class ASubType1 : A { public string Value1 { get; set; } }",
-            "class ASubType2 : A { public string Value2 { get; set; } }",
-            "class BSubType1 : B { public string Value1 { get; set; } }",
-            "class BSubType2 : B { public string Value2 { get; set; } }"
-        );
-        return TestHelper.VerifyGenerator(source);
-    }
-
-    [Fact]
-    public Task WithBaseTypeConfigAndSeparateMethodShouldWork()
-    {
-        var source = TestSourceBuilder.MapperWithBodyAndTypes(
-            """
-            [MapDerivedType<ASubType1, BSubType1>]
-            [MapDerivedType<ASubType2, BSubType2>]
-            [MapProperty(nameof(A.BaseValueA), nameof(B.BaseValueB)]
-            public partial B Map(A src);
-
-            [MapperIgnoreSource(nameof(A.BaseValueA)]
-            [MapperIgnoreTarget(nameof(B.BaseValueB)]
-            public partial BSubType1 Map(ASubType1 src);
+            public partial void Map(A src, B trg);
             """,
             "abstract class A { public string BaseValueA { get; set; } }",
             "abstract class B { public string BaseValueB { get; set; } }",
@@ -231,7 +149,7 @@ public void NotAssignableSourceTypeShouldDiagnostic()
         var source = TestSourceBuilder.MapperWithBodyAndTypes(
             """
             [MapDerivedType<AImpl1, BImpl1>]
-            public partial B Map(A src);
+            public partial void Map(A src, B trg);
             """,
             "interface A {}",
             "interface B {}",
@@ -255,7 +173,7 @@ public void DuplicatedSourceTypeShouldDiagnostic()
             """
             [MapDerivedType<AImpl1, BImpl1>]
             [MapDerivedType<AImpl1, BImpl2>]
-            public partial B Map(A src);
+            public partial void Map(A src, B trg);
             """,
             "interface A {}",
             "interface B {}",
@@ -279,7 +197,7 @@ public void NotMappableShouldDiagnostic()
         var source = TestSourceBuilder.MapperWithBody(
             """
             [MapDerivedType<Version, int>]
-            public partial object Map(object src);
+            public partial void Map(object src, object trg);
             """
         );
         TestHelper

diff --git a/...TargetTypeTest.WithAbstractBaseClassAndNonGenericInterfaceShouldWork#Mapper.g.verified.cs b/...TargetTypeTest.WithAbstractBaseClassAndNonGenericInterfaceShouldWork#Mapper.g.verified.cs
@@ -0,0 +1,22 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    public partial void Map(global::A src, global::B trg)
+    {
+        switch (src, trg)
+        {
+            case (global::ASubType1 source, global::BSubType1 target):
+                target.Value1 = source.Value1;
+                target.BaseValue = source.BaseValue;
+                break;
+            case (global::ASubType2 source, global::BSubType2 target):
+                target.Value2 = source.Value2;
+                target.BaseValue = source.BaseValue;
+                break;
+            default:
+                throw new System.ArgumentException($"Cannot map {src.GetType()} to {trg.GetType()} as there is no known derived type mapping", nameof(src));
+        }
+    }
+}
diff --git a/...pshots/DerivedExistingTargetTypeTest.WithAbstractBaseClassShouldWork#Mapper.g.verified.cs b/...pshots/DerivedExistingTargetTypeTest.WithAbstractBaseClassShouldWork#Mapper.g.verified.cs
@@ -0,0 +1,22 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    public partial void Map(global::A src, global::B trg)
+    {
+        switch (src, trg)
+        {
+            case (global::ASubType1 source, global::BSubType1 target):
+                target.Value1 = source.Value1;
+                target.BaseValue = source.BaseValue;
+                break;
+            case (global::ASubType2 source, global::BSubType2 target):
+                target.Value1 = source.Value1;
+                target.BaseValue = source.BaseValue;
+                break;
+            default:
+                throw new System.ArgumentException($"Cannot map {src.GetType()} to {trg.GetType()} as there is no known derived type mapping", nameof(src));
+        }
+    }
+}
diff --git a/...snapshots/DerivedExistingTargetTypeTest.WithBaseTypeConfigShouldWork#Mapper.g.verified.cs b/...snapshots/DerivedExistingTargetTypeTest.WithBaseTypeConfigShouldWork#Mapper.g.verified.cs
@@ -0,0 +1,22 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    public partial void Map(global::A src, global::B trg)
+    {
+        switch (src, trg)
+        {
+            case (global::ASubType1 source, global::BSubType1 target):
+                target.Value1 = source.Value1;
+                target.BaseValueB = source.BaseValueA;
+                break;
+            case (global::ASubType2 source, global::BSubType2 target):
+                target.Value2 = source.Value2;
+                target.BaseValueB = source.BaseValueA;
+                break;
+            default:
+                throw new System.ArgumentException($"Cannot map {src.GetType()} to {trg.GetType()} as there is no known derived type mapping", nameof(src));
+        }
+    }
+}
diff --git a/...sts/_snapshots/DerivedExistingTargetTypeTest.WithInterfaceShouldWork#Mapper.g.verified.cs b/...sts/_snapshots/DerivedExistingTargetTypeTest.WithInterfaceShouldWork#Mapper.g.verified.cs
@@ -0,0 +1,22 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    public partial void Map(global::A src, global::B trg)
+    {
+        switch (src, trg)
+        {
+            case (global::AImpl1 source, global::BImpl1 target):
+                target.BaseValue = source.BaseValue;
+                target.Value1 = source.Value1;
+                break;
+            case (global::AImpl2 source, global::BImpl2 target):
+                target.BaseValue = source.BaseValue;
+                target.Value2 = source.Value2;
+                break;
+            default:
+                throw new System.ArgumentException($"Cannot map {src.GetType()} to {trg.GetType()} as there is no known derived type mapping", nameof(src));
+        }
+    }
+}
diff --git a/...xistingTargetTypeTest.WithInterfaceSourceAndTargetNullableShouldWork#Mapper.g.verified.cs b/...xistingTargetTypeTest.WithInterfaceSourceAndTargetNullableShouldWork#Mapper.g.verified.cs
@@ -0,0 +1,24 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    public partial void Map(global::A? src, global::B? trg)
+    {
+        if (src == null || trg == null)
+            return;
+        switch (src, trg)
+        {
+            case (global::AImpl1 source, global::BImpl1 target):
+                target.BaseValue = source.BaseValue;
+                target.Value1 = source.Value1;
+                break;
+            case (global::AImpl2 source, global::BImpl2 target):
+                target.BaseValue = source.BaseValue;
+                target.Value2 = source.Value2;
+                break;
+            default:
+                throw new System.ArgumentException($"Cannot map {src.GetType()} to {trg.GetType()} as there is no known derived type mapping", nameof(src));
+        }
+    }
+}
diff --git a/.../DerivedExistingTargetTypeTest.WithInterfaceSourceNullableShouldWork#Mapper.g.verified.cs b/.../DerivedExistingTargetTypeTest.WithInterfaceSourceNullableShouldWork#Mapper.g.verified.cs
@@ -0,0 +1,24 @@
+//HintName: Mapper.g.cs
+// <auto-generated />
+#nullable enable
+public partial class Mapper
+{
+    public partial void Map(global::A? src, global::B trg)
+    {
+        if (src == null)
+            return;
+        switch (src, trg)
+        {
+            case (global::AImpl1 source, global::BImpl1 target):
+                target.BaseValue = source.BaseValue;
+                target.Value1 = source.Value1;
+                break;
+            case (global::AImpl2 source, global::BImpl2 target):
+                target.BaseValue = source.BaseValue;
+                target.Value2 = source.Value2;
+                break;
+            default:
+                throw new System.ArgumentException($"Cannot map {src.GetType()} to {trg.GetType()} as there is no known derived type mapping", nameof(src));
+        }
+    }
+}