From 75aeedfa004e47506b4b8948926e405de942ed03 Mon Sep 17 00:00:00 2001
From: sunkaixuan <610262374@qq.com>
Date: Tue, 23 Aug 2022 23:29:23 +0800
Subject: [PATCH] Add SqlServer Unit test

---
 .../SqlServerTest/SqlServerTest.csproj        |  1 +
 Src/Asp.Net/SqlServerTest/UnitTest/Main.cs    |  1 +
 .../UnitTest/UnitTestReturnPkList.cs          | 75 +++++++++++++++++++
 3 files changed, 77 insertions(+)
 create mode 100644 Src/Asp.Net/SqlServerTest/UnitTest/UnitTestReturnPkList.cs

diff --git a/Src/Asp.Net/SqlServerTest/SqlServerTest.csproj b/Src/Asp.Net/SqlServerTest/SqlServerTest.csproj
index fd67ef17e..3bed4f05e 100644
--- a/Src/Asp.Net/SqlServerTest/SqlServerTest.csproj
+++ b/Src/Asp.Net/SqlServerTest/SqlServerTest.csproj
@@ -106,6 +106,7 @@
     <Compile Include="UnitTest\Models\ImsBillStk.cs" />
     <Compile Include="UnitTest\UnitSub.cs" />
     <Compile Include="UnitTest\UnitTestConfigQuery.cs" />
+    <Compile Include="UnitTest\UnitTestReturnPkList.cs" />
     <Compile Include="UnitTest\UnitUpdateNavN3.cs" />
     <Compile Include="UnitTest\UnitOneToOne12.cs" />
     <Compile Include="UnitTest\Models\poetry_video_comment.cs" />
diff --git a/Src/Asp.Net/SqlServerTest/UnitTest/Main.cs b/Src/Asp.Net/SqlServerTest/UnitTest/Main.cs
index 783aaa23f..31a9b9c84 100644
--- a/Src/Asp.Net/SqlServerTest/UnitTest/Main.cs
+++ b/Src/Asp.Net/SqlServerTest/UnitTest/Main.cs
@@ -31,6 +31,7 @@ namespace OrmTest
         }
         public static void Init()
         {
+            UnitTestReturnPkList.Init();
             UnitCustom12312.Init();
             UnitEnum22.Init();
             UCustom025.Init();
diff --git a/Src/Asp.Net/SqlServerTest/UnitTest/UnitTestReturnPkList.cs b/Src/Asp.Net/SqlServerTest/UnitTest/UnitTestReturnPkList.cs
new file mode 100644
index 000000000..cb21ed2de
--- /dev/null
+++ b/Src/Asp.Net/SqlServerTest/UnitTest/UnitTestReturnPkList.cs
@@ -0,0 +1,75 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace OrmTest
+{
+    public class UnitTestReturnPkList
+    {
+        public static void Init()
+        {
+            TestIdentity();
+            TestLong();
+            TestGuid();
+        }
+
+        private static void TestGuid()
+        {
+            var db = NewUnitTest.Db;
+            db.CodeFirst.InitTables<UnitGuid123>();
+            var ids = db.Insertable(new UnitGuid123() { Name = "a" }).ExecuteReturnPkList<Guid>();
+            if (ids.Count() != 1) { throw new Exception("unit error"); }
+            if (db.Queryable<UnitGuid123>().In(ids).Count() !=1) { throw new Exception("unit error"); }
+            var ids2 = db.Insertable(new List<UnitGuid123>() { new UnitGuid123() { Name = "c" }, new UnitGuid123() { Name = "c" } }).ExecuteReturnPkList<Guid>();
+            if (ids2.Count() != 2) { throw new Exception("unit error"); }
+            if (db.Queryable<UnitGuid123>().In(ids2).Count() != 2) { throw new Exception("unit error"); }
+        }
+
+        private static void TestLong()
+        {
+            var db = NewUnitTest.Db;
+            db.CodeFirst.InitTables<UnitLong1231>();
+            var ids = db.Insertable(new UnitLong1231() { Name = "a" }).ExecuteReturnPkList<long>();
+            if (ids.Count(z => z > 0) != 1) { throw new Exception("unit error"); }
+            var ids2 = db.Insertable(new List<UnitLong1231>() { new UnitLong1231() { Name = "c" }, new UnitLong1231() { Name = "c" } }).ExecuteReturnPkList<long>();
+            if (ids2.Count(z => z > 0) != 2) { throw new Exception("unit error"); }
+            if(db.Queryable<UnitLong1231>().In(ids2).Count()!=2) { throw new Exception("unit error"); }
+        }
+
+        private static void TestIdentity()
+        {
+            var db = NewUnitTest.Db;
+            db.CodeFirst.InitTables<UnitIdentity01>();
+            var ids = db.Insertable(new UnitIdentity01() { Name = "a" }).ExecuteReturnPkList<int>();
+            if (ids.Count(z => z > 0) != 1) { throw new Exception("unit error"); }
+            var ids2 = db.Insertable(new List<UnitIdentity01>() { new UnitIdentity01() { Name = "c" }, new UnitIdentity01() { Name = "c" } }).ExecuteReturnPkList<int>();
+            if (ids2.Count(z => z > 0) != 2) { throw new Exception("unit error"); }
+            if (db.Queryable<UnitIdentity01>().In(ids2).Count() != 2) { throw new Exception("unit error"); }
+        }
+
+        public class UnitGuid123
+        {
+            [SqlSugar.SugarColumn(IsPrimaryKey = true)]
+            public Guid id { get; set; }
+
+            public string Name { get; set; }
+        }
+        public class UnitLong1231
+        {
+            [SqlSugar.SugarColumn(IsPrimaryKey = true)]
+            public long id { get; set; }
+
+            public string Name { get; set; }
+        }
+
+        public class UnitIdentity01 
+        {
+            [SqlSugar.SugarColumn(IsIdentity =true,IsPrimaryKey =true)]
+            public int id { get; set; }
+
+            public string Name { get; set; }
+        }
+    }
+}