|
@@ -0,0 +1,470 @@
|
|
|
+
|
|
|
+using Microsoft.VisualStudio.TestTools.UnitTesting;
|
|
|
+using Vit.Extensions.Linq_Extensions;
|
|
|
+using System.Data;
|
|
|
+using static Vit.Core.Util.XmlComment.MethodComment;
|
|
|
+
|
|
|
+namespace Vit.Orm.Sqlite.MsTest
|
|
|
+{
|
|
|
+
|
|
|
+ [TestClass]
|
|
|
+ public class Query_Test
|
|
|
+ {
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Get()
|
|
|
+ {
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var user = dbContext.Get<User>(3);
|
|
|
+ Assert.AreEqual(3, user?.id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var user = dbContext.DbSet<User>().Get(5);
|
|
|
+ Assert.AreEqual(5, user?.id);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_PlainQuery()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.ToList();
|
|
|
+ Assert.AreEqual(6, userList.Count);
|
|
|
+ Assert.AreEqual(1, userList.First().id);
|
|
|
+ Assert.AreEqual(6, userList.Last().id);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.Select(u => u.id).ToList();
|
|
|
+ Assert.AreEqual(6, userList.Count);
|
|
|
+ Assert.AreEqual(1, userList.First());
|
|
|
+ Assert.AreEqual(6, userList.Last());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Where()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.Where(u => u.id > 2).Where(m => m.id < 4).ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var userList = userQuery.Where(u => u.id + 1 == 4).Where(m => m.fatherId == 5).ToList();
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var userList = userQuery.Where(u => 4 == u.id + 1).Where(m => m.fatherId == 5).ToList();
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.Where(u => u.birth == new DateTime(2021, 01, 01, 03, 00, 00)).ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var userList = userQuery.Where(u => u.birth == DateTime.Parse("2021-01-01 01:00:00").AddHours(2)).ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Select()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.Select(u => u).Where(user => user.id > 2).Where(user => user.id < 4).Select(u => u).ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = (from user in userQuery
|
|
|
+ select new
|
|
|
+ {
|
|
|
+ uniqueId1 = user.id + "_" + user.fatherId + "_" + user.motherId,
|
|
|
+ uniqueId2 = $"{user.id}_{user.fatherId}_{user.motherId}"
|
|
|
+ }).ToList();
|
|
|
+
|
|
|
+ Assert.AreEqual(6, userList.Count);
|
|
|
+ Assert.AreEqual("1_4_6", userList.First().uniqueId1);
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Count()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var count = (from user in userQuery
|
|
|
+ from father in userQuery.Where(father => user.fatherId == father.id).DefaultIfEmpty()
|
|
|
+ where user.id > 2 && father == null
|
|
|
+ select new
|
|
|
+ {
|
|
|
+ father
|
|
|
+ }).Count();
|
|
|
+
|
|
|
+ Assert.AreEqual(3, count);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_AllFeatures()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ #region SelectMany().Where().OrderBy().Skip().Take().ToExecuteString()
|
|
|
+ /*
|
|
|
+ users.SelectMany(
|
|
|
+ user => users.Where(father => user.fatherId == father.id).DefaultIfEmpty()
|
|
|
+ , (user, father) => new {user = user, father = father}
|
|
|
+ ).Where(row => row.user.id > 2)
|
|
|
+ .Select(row => new {row.user })
|
|
|
+ .OrderBy(user=>user.id)
|
|
|
+ .Skip(1).Take(2);
|
|
|
+ */
|
|
|
+ {
|
|
|
+ var query = (from user in userQuery
|
|
|
+ from father in userQuery.Where(father => user.fatherId == father.id).DefaultIfEmpty()
|
|
|
+ where user.id > 2
|
|
|
+ orderby father.id, user.id descending
|
|
|
+ select new
|
|
|
+ {
|
|
|
+ user
|
|
|
+ })
|
|
|
+ .Skip(1).Take(2);
|
|
|
+
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var list = query.ToList();
|
|
|
+
|
|
|
+ Assert.AreEqual(2, list.Count);
|
|
|
+ Assert.AreEqual(5, list[0].user.id);
|
|
|
+ Assert.AreEqual(4, list[1].user.id);
|
|
|
+ }
|
|
|
+ #endregion
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_FirstOrDefault()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.FirstOrDefault();
|
|
|
+ Assert.AreEqual(1, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.FirstOrDefault(user => user.id == 3);
|
|
|
+ Assert.AreEqual(3, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.FirstOrDefault(user => user.id == 13);
|
|
|
+ Assert.AreEqual(null, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.OrderByDescending(m => m.id).FirstOrDefault();
|
|
|
+ Assert.AreEqual(6, user?.id);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_First()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.First();
|
|
|
+ Assert.AreEqual(1, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.First(user => user.id == 3);
|
|
|
+ Assert.AreEqual(3, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var user = userQuery.First(user => user.id == 13);
|
|
|
+ Assert.Fail("IQueryalbe.First should throw Exception");
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_LastOrDefault()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.LastOrDefault();
|
|
|
+ Assert.AreEqual(6, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.LastOrDefault(user => user.id == 3);
|
|
|
+ Assert.AreEqual(3, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.LastOrDefault(user => user.id == 13);
|
|
|
+ Assert.AreEqual(null, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.OrderByDescending(m => m.id).LastOrDefault();
|
|
|
+ Assert.AreEqual(1, user?.id);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Last()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.Last();
|
|
|
+ Assert.AreEqual(6, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var user = userQuery.Last(user => user.id == 3);
|
|
|
+ Assert.AreEqual(3, user?.id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ try
|
|
|
+ {
|
|
|
+ var user = userQuery.Last(user => user.id == 13);
|
|
|
+ Assert.Fail("IQueryalbe.First should throw Exception");
|
|
|
+ }
|
|
|
+ catch (Exception ex)
|
|
|
+ {
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+ // Enumerable.ToArray
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Enumerable_ToArray()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.ToArray();
|
|
|
+ Assert.AreEqual(6, userList.Length);
|
|
|
+ Assert.AreEqual(1, userList.First().id);
|
|
|
+ Assert.AreEqual(6, userList.Last().id);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.Select(u => u.id).ToArray();
|
|
|
+ Assert.AreEqual(6, userList.Length);
|
|
|
+ Assert.AreEqual(1, userList.First());
|
|
|
+ Assert.AreEqual(6, userList.Last());
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ // Enumerable.Contains
|
|
|
+ // Queryable.Contains
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Contains()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var userList = userQuery.Where(u => new[] { 3, 5 }.Contains(u.id)).ToList();
|
|
|
+ Assert.AreEqual(2, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ Assert.AreEqual(5, userList.Last().id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var ids = new[] { 3, 5 }.AsEnumerable();
|
|
|
+ var userList = userQuery.Where(u => ids.Contains(u.id)).ToList();
|
|
|
+ Assert.AreEqual(2, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ Assert.AreEqual(5, userList.Last().id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var ids = new[] { 3, 5 }.AsQueryable();
|
|
|
+ var userList = userQuery.Where(u => ids.Contains(u.id)).ToList();
|
|
|
+ Assert.AreEqual(2, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ Assert.AreEqual(5, userList.Last().id);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_StringMethods()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ userQuery.ExecuteUpdate(row => new User
|
|
|
+ {
|
|
|
+ name = "u|" + row.id + "|" + (row.fatherId.ToString() ?? "") + "|" + (row.motherId.ToString() ?? "")
|
|
|
+ });
|
|
|
+
|
|
|
+ // StartsWith
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => u.name.StartsWith("u|3|5"));
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ Assert.AreEqual("u|3|5|6", userList.First().name);
|
|
|
+ }
|
|
|
+ // EndsWith
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => u.name.EndsWith("3|5|6"));
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ Assert.AreEqual("u|3|5|6", userList.First().name);
|
|
|
+ }
|
|
|
+ // Contains
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => u.name.Contains("|3|5|"));
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.First().id);
|
|
|
+ Assert.AreEqual("u|3|5|6", userList.First().name);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_DbFunction()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+
|
|
|
+ // select * from `User` as t0 where IIF(`t0`.`fatherId` is not null,true, false)
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => DbFunction.Call<bool>("IIF", u.fatherId != null, true, false));
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(3, userList.Count);
|
|
|
+ Assert.AreEqual(3, userList.Last().id);
|
|
|
+ }
|
|
|
+
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => u.birth == DbFunction.Call<DateTime?>("datetime", "2021-01-01 00:00:00", "+2 hours"));
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(1, userList.Count);
|
|
|
+ Assert.AreEqual(2, userList.First().id);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => u.birth == DbFunction.Call<DateTime>("datetime", "2021-01-01 00:00:00", "+" + u.id + " hours"));
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(6, userList.Count);
|
|
|
+ Assert.AreEqual(1, userList.First().id);
|
|
|
+ }
|
|
|
+
|
|
|
+ // coalesce(parameter1,parameter2, …)
|
|
|
+ {
|
|
|
+ var query = userQuery.Where(u => DbFunction.Call<int?>("coalesce", u.fatherId, u.motherId) != null);
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+ Assert.AreEqual(3, userList.Count);
|
|
|
+ Assert.AreEqual(1, userList.First().id);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ [TestMethod]
|
|
|
+ public void Test_Distinct()
|
|
|
+ {
|
|
|
+ using var dbContext = DataSource.CreateFormatedDbContext(System.Reflection.MethodBase.GetCurrentMethod()?.Name ?? "_");
|
|
|
+ var userQuery = dbContext.Query<User>();
|
|
|
+
|
|
|
+ {
|
|
|
+ var query = userQuery.Select(u => new { u.fatherId }).Distinct();
|
|
|
+
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+
|
|
|
+ Assert.AreEqual(3, userList.Count);
|
|
|
+ Assert.AreEqual(4, userList.First().fatherId);
|
|
|
+ Assert.AreEqual(null, userList.Last().fatherId);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var query = userQuery.Select(u => u.fatherId).Distinct();
|
|
|
+
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var fatherId = query.FirstOrDefault();
|
|
|
+ Assert.AreEqual(4, fatherId);
|
|
|
+ }
|
|
|
+ {
|
|
|
+ var query = userQuery.Distinct();
|
|
|
+
|
|
|
+ var sql = query.ToExecuteString();
|
|
|
+ var userList = query.ToList();
|
|
|
+
|
|
|
+ Assert.AreEqual(6, userList.Count);
|
|
|
+ }
|
|
|
+
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+
|
|
|
+ }
|
|
|
+}
|