From fed96c10e97d81721c18a70569637620715a43ad Mon Sep 17 00:00:00 2001
From: sunkaixuan <610262374@qq.com>
Date: Sat, 10 Sep 2022 20:18:36 +0800
Subject: [PATCH] Add unit test

---
 .../SqlServerTest/UnitTest/UCustom20.cs       | 64 +++++++++++++++----
 1 file changed, 53 insertions(+), 11 deletions(-)

diff --git a/Src/Asp.Net/SqlServerTest/UnitTest/UCustom20.cs b/Src/Asp.Net/SqlServerTest/UnitTest/UCustom20.cs
index 993c846d0..cb63b9fa1 100644
--- a/Src/Asp.Net/SqlServerTest/UnitTest/UCustom20.cs
+++ b/Src/Asp.Net/SqlServerTest/UnitTest/UCustom20.cs
@@ -4,27 +4,29 @@ using System.Linq;
 using System.Text;
 using System.Threading.Tasks;
 
-namespace OrmTest 
+namespace OrmTest
 {
-    public  class UCustom20
+    public class UCustom20
     {
-        public static void Init() 
+        public static void Init()
         {
             var db = NewUnitTest.Db;
             db.DbMaintenance.TruncateTable<Order>();
             db.Insertable(new Order() { Id = 1, Name = "jack", CreateTime = DateTime.Now, CustomId = 1 }).ExecuteCommand();
-           var test1= db.Queryable<Order>()
-                .ToList(z => new {
-                    name1 = new { z.Id, z.Name , ZId=100 }
-                }).First();
+            var test1 = db.Queryable<Order>()
+                 .ToList(z => new
+                 {
+                     name1 = new { z.Id, z.Name, ZId = 100 }
+                 }).First();
 
-            if (test1.name1.Id != 1 || test1.name1.Name != "jack" || test1.name1.ZId != 100) 
+            if (test1.name1.Id != 1 || test1.name1.Name != "jack" || test1.name1.ZId != 100)
             {
                 throw new Exception("unit error");
             }
 
             var test2 = db.Queryable<Order>()
-           .ToList(z => new {
+           .ToList(z => new
+           {
                name1 = new { z.Id, z.Name, ZId = z.Id.ToString() }
            }).First();
 
@@ -38,11 +40,51 @@ namespace OrmTest
              .Select(i => new TestDTO
              {
                  SubOne = new TestSubDTO { NameOne = "a1", NameTwo = i.Id.ToString() },
-              //   SubTwo = new TestSubDTO { NameOne = i.Name, NameTwo = i.Name }
+                 //   SubTwo = new TestSubDTO { NameOne = i.Name, NameTwo = i.Name }
              })
              .First();
 
-            if (test3.SubOne.NameOne !="a1"|| test3.SubOne.NameTwo != "1")
+            if (test3.SubOne.NameOne != "a1" || test3.SubOne.NameTwo != "1")
+            {
+                throw new Exception("unit error");
+            }
+
+            var test4 = db.Queryable<Order>()
+            .Take(2)
+             .Select(i => new TestDTO
+             {
+                 SubOne = new TestSubDTO { NameOne = "a1", NameTwo = i.Name },
+                 //SubTwo = new TestSubDTO { NameOne = i.Name, NameTwo = i.Name }
+             })
+            .ToList().First();
+            if (test4.SubOne.NameOne != "a1" || test4.SubOne.NameTwo != "jack")
+            {
+                throw new Exception("unit error");
+            }
+
+            var test5 = db.Queryable<Order>()
+           .Take(2)
+            .Select(i => new TestDTO
+            {
+                SubOne = new TestSubDTO { NameOne = i.Name, NameTwo = i.Name },
+                //SubTwo = new TestSubDTO { NameOne = i.Name, NameTwo = i.Name }
+            })
+           .ToList().First();
+            if (test5.SubOne.NameOne != "jack" || test5.SubOne.NameTwo != "jack")
+            {
+                throw new Exception("unit error");
+            }
+
+            var test6 = db.Queryable<Order>()
+            .Take(2)
+             .Select(i => new TestDTO
+             {
+                 SubOne = new TestSubDTO { NameOne = i.Name+"1", NameTwo = i.Name+"2" },
+                 SubTwo = new TestSubDTO { NameOne = i.Name+"3", NameTwo = i.Name+"4" }
+             })
+            .ToList().First();
+            if (test6.SubOne.NameOne != "jack1" || test6.SubOne.NameTwo != "jack2"||
+                test6.SubTwo.NameOne != "jack3" || test6.SubTwo.NameTwo != "jack4")
             {
                 throw new Exception("unit error");
             }