Query_Select_Test.cs 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168
  1. using System.Data;
  2. using Microsoft.VisualStudio.TestTools.UnitTesting;
  3. namespace Vitorm.MsTest.CommonTest
  4. {
  5. [TestClass]
  6. public class Query_Select_Test
  7. {
  8. [TestMethod]
  9. public void Test_Select()
  10. {
  11. using var dbContext = DataSource.CreateDbContext();
  12. var userQuery = dbContext.Query<User>();
  13. //
  14. {
  15. var userList = userQuery.ToList();
  16. }
  17. // userQuery.Select(user => user)
  18. {
  19. var rows = userQuery.Select(user => user).ToList();
  20. Assert.AreEqual(6, rows.Count);
  21. }
  22. // userQuery.Select(user => user.id);
  23. {
  24. var rows = userQuery.Select(user => user.id).ToList();
  25. }
  26. {
  27. var rows = userQuery.Select(user => user.id + 100).ToList();
  28. }
  29. {
  30. var rows = userQuery.Select(user => 1000 + user.id + 100).ToList();
  31. }
  32. {
  33. var rows = userQuery.Select(user => (double)user.id).ToList();
  34. }
  35. {
  36. var rows = userQuery.Select(user => (double)user.id + 0.1).ToList();
  37. }
  38. {
  39. var rows = userQuery.Select(user => user.id + 0.1).ToList();
  40. }
  41. {
  42. var query =
  43. from user in userQuery
  44. where user.id > 1
  45. orderby user.id
  46. select new
  47. {
  48. id = user.id,
  49. user = user,
  50. ext = new
  51. {
  52. father = user,
  53. fid = user.fatherId
  54. },
  55. trueValue = true,
  56. };
  57. var rows = query.ToList();
  58. }
  59. // String.Format(format: "{0}_{1}_{2}", "0", "1", "2")
  60. {
  61. var query =
  62. from user in userQuery
  63. orderby user.id
  64. select new
  65. {
  66. uniqueId1 = user.id + "_" + user.fatherId + "_" + user.motherId,
  67. uniqueId2 = $"{user.id}_{user.fatherId}_{user.motherId}"
  68. };
  69. var userList = query.ToList();
  70. Assert.AreEqual(6, userList.Count);
  71. Assert.AreEqual("1_4_6", userList[0].uniqueId1);
  72. Assert.AreEqual("1_4_6", userList[0].uniqueId2);
  73. Assert.AreEqual("4__", userList[3].uniqueId1);
  74. Assert.AreEqual("4__", userList[3].uniqueId2);
  75. }
  76. }
  77. [TestMethod]
  78. public void Test_Select_ExistClass()
  79. {
  80. using var dbContext = DataSource.CreateDbContext();
  81. var userQuery = dbContext.Query<User>();
  82. {
  83. var query =
  84. from user in userQuery
  85. where user.id > 1
  86. orderby user.id
  87. select new User2(user.id)
  88. ;
  89. var userList = query.ToList();
  90. Assert.AreEqual(5, userList.Count);
  91. }
  92. {
  93. var query =
  94. from user in userQuery
  95. where user.id > 1
  96. orderby user.id
  97. select new User2(user.name)
  98. ;
  99. var userList = query.ToList();
  100. Assert.AreEqual(5, userList.Count);
  101. }
  102. {
  103. var query =
  104. from user in userQuery
  105. where user.id > 1
  106. orderby user.id
  107. select new User2(user.name)
  108. {
  109. id = user.id + 100,
  110. fatherId = user.fatherId,
  111. father = user,
  112. }
  113. ;
  114. var userList = query.ToList();
  115. Assert.AreEqual(5, userList.Count);
  116. }
  117. {
  118. var query =
  119. from user in userQuery
  120. where user.id > 1
  121. orderby user.id
  122. select new User2
  123. {
  124. id = user.id,
  125. fatherId = 12,
  126. };
  127. var userList = query.ToList();
  128. Assert.AreEqual(5, userList.Count);
  129. }
  130. }
  131. class User2 : User
  132. {
  133. public User2() { }
  134. public User2(int nid) { id = nid; }
  135. public User2(string name) { this.name = name; }
  136. public User father;
  137. }
  138. }
  139. }