ReadEntity_Test.cs 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322
  1. using System.Data;
  2. using Microsoft.VisualStudio.TestTools.UnitTesting;
  3. namespace Vitorm.MsTest.CustomTest
  4. {
  5. [TestClass]
  6. public class ReadEntity_Test
  7. {
  8. [TestMethod]
  9. public void ReadValue()
  10. {
  11. using var dbContext = DataSource.CreateDbContext();
  12. var userQuery = dbContext.Query<User>();
  13. // ReadValue
  14. {
  15. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  16. var ids = reader.ReadValue<int>().ToList();
  17. Assert.AreEqual("1,2,3,4", String.Join(",", ids));
  18. }
  19. // ReadValue
  20. {
  21. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  22. var names = reader.ReadValue<string>(1).ToList();
  23. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", names));
  24. }
  25. // ReadValue
  26. {
  27. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  28. var ids = reader.ReadValue<int?>("UserFatherId").ToList();
  29. Assert.AreEqual("4,4,5,", String.Join(",", ids));
  30. }
  31. }
  32. [TestMethod]
  33. public void ReadTuple()
  34. {
  35. using var dbContext = DataSource.CreateDbContext();
  36. var userQuery = dbContext.Query<User>();
  37. // ReadTuple
  38. {
  39. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  40. List<(int id, string name)> userList = reader.ReadTuple<int, string>().ToList();
  41. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  42. }
  43. // ReadTuple by indexes
  44. {
  45. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  46. List<(int id, int? fatherId)> userList = reader.ReadTuple<int, int?>([0, 3]).ToList();
  47. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  48. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  49. }
  50. // ReadTuple by columnNames
  51. {
  52. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  53. List<(int id, int? fatherId)> userList = reader.ReadTuple<int, int?>(["userId", "userFatherId"]).ToList();
  54. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  55. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  56. }
  57. }
  58. [TestMethod]
  59. public void ReadTuple3()
  60. {
  61. using var dbContext = DataSource.CreateDbContext();
  62. var userQuery = dbContext.Query<User>();
  63. // ReadTuple
  64. {
  65. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  66. List<(int id, string name, DateTime? birth)> userList = reader.ReadTuple<int, string, DateTime?>().ToList();
  67. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  68. }
  69. // ReadTuple by indexes
  70. {
  71. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  72. List<(int id, int? fatherId, int? motherId)> userList = reader.ReadTuple<int, int?, int?>([0, 3, 4]).ToList();
  73. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  74. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  75. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  76. }
  77. // ReadTuple by columnNames
  78. {
  79. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  80. List<(int id, int? fatherId, int? motherId)> userList = reader.ReadTuple<int, int?, int?>(["userId", "userFatherId", "UserMotherId"]).ToList();
  81. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  82. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  83. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  84. }
  85. }
  86. [TestMethod]
  87. public void ReadTuple4()
  88. {
  89. using var dbContext = DataSource.CreateDbContext();
  90. var userQuery = dbContext.Query<User>();
  91. // ReadTuple
  92. {
  93. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  94. List<(int id, string name, DateTime? birth, int? fatherId)> userList = reader.ReadTuple<int, string, DateTime?, int?>().ToList();
  95. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  96. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  97. }
  98. // ReadTuple by indexes
  99. {
  100. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  101. List<(int id, int? fatherId, int? motherId, string name)> userList = reader.ReadTuple<int, int?, int?, string>([0, 3, 4, 1]).ToList();
  102. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  103. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  104. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  105. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  106. }
  107. // ReadTuple by columnNames
  108. {
  109. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  110. List<(int id, int? fatherId, int? motherId, string name)> userList = reader.ReadTuple<int, int?, int?, string>(["userId", "userFatherId", "UserMotherId", "userName"]).ToList();
  111. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  112. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  113. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  114. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  115. }
  116. }
  117. [TestMethod]
  118. public void ReadEntity()
  119. {
  120. using var dbContext = DataSource.CreateDbContext();
  121. var userQuery = dbContext.Query<User>();
  122. // ReadEntity
  123. {
  124. using var reader = dbContext.ExecuteReader("select * from `User` where userId<=4 order by userId");
  125. var userList = reader.ReadEntity<User>().ToList();
  126. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  127. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  128. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  129. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  130. }
  131. }
  132. [TestMethod]
  133. public void ReadEntity2()
  134. {
  135. using var dbContext = DataSource.CreateDbContext();
  136. var userQuery = dbContext.Query<User>();
  137. // ReadEntity : try read all columns for each entities
  138. {
  139. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  140. var userList = reader.ReadEntity<User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1) => (u0.id, u0.name, u1.fatherId, u1.motherId)).ToList();
  141. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  142. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  143. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  144. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  145. }
  146. // ReadEntity splitOn column not exist
  147. {
  148. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  149. var userList = reader.ReadEntity<User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1) => (u0.id, u0.name, u1.fatherId, u1.motherId), "dummyField").ToList();
  150. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  151. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  152. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  153. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  154. }
  155. // ReadEntity splitOn
  156. {
  157. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  158. var userList = reader.ReadEntity<User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1) => (u0.id, u0.name, u1.fatherId, u1.motherId), "userFatherId").ToList();
  159. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  160. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  161. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  162. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  163. }
  164. // ReadEntity splitIndex
  165. {
  166. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  167. var userList = reader.ReadEntity<User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1) => (u0.id, u0.name, u1.fatherId, u1.motherId), 3).ToList();
  168. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  169. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  170. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  171. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  172. }
  173. }
  174. [TestMethod]
  175. public void ReadEntity3()
  176. {
  177. using var dbContext = DataSource.CreateDbContext();
  178. var userQuery = dbContext.Query<User>();
  179. // ReadEntity : try read all columns for each entities
  180. {
  181. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  182. var userList = reader.ReadEntity<User, User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1, u2) => (u0.id, u0.name, u1.fatherId, u2.motherId)).ToList();
  183. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  184. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  185. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  186. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  187. }
  188. // ReadEntity splitOn column not exist
  189. {
  190. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  191. var userList = reader.ReadEntity<User, User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1, u2) => (u0.id, u0.name, u1.fatherId, u2.motherId), "dummyField", "dummyField").ToList();
  192. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  193. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  194. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  195. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  196. }
  197. // ReadEntity splitOns
  198. {
  199. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  200. var userList = reader.ReadEntity<User, User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1, u2) => (u0.id, u0.name, u1.fatherId, u2.motherId), "userFatherId", "userMotherId").ToList();
  201. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  202. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  203. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  204. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  205. }
  206. // ReadEntity splitOns
  207. {
  208. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth, userFatherId,userId, userMotherId,userId from `User` where userId<=4 order by userId");
  209. var userList = reader.ReadEntity<User, User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1, u2) => (u0.id, u0.name, u1.fatherId, u2.motherId), "userFatherId", "userMotherId").ToList();
  210. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  211. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  212. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  213. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  214. }
  215. // ReadEntity splitIndex
  216. {
  217. using var reader = dbContext.ExecuteReader("select userId,userName,userBirth,userFatherId,userMotherId from `User` where userId<=4 order by userId");
  218. var userList = reader.ReadEntity<User, User, User, (int id, string name, int? fatherId, int? motherId)>((u0, u1, u2) => (u0.id, u0.name, u1.fatherId, u2.motherId), 3, 4).ToList();
  219. Assert.AreEqual("1,2,3,4", String.Join(",", userList.Select(u => u.id)));
  220. Assert.AreEqual("u146,u246,u356,u400", String.Join(",", userList.Select(u => u.name)));
  221. Assert.AreEqual("4,4,5,", String.Join(",", userList.Select(u => u.fatherId)));
  222. Assert.AreEqual("6,6,6,", String.Join(",", userList.Select(u => u.motherId)));
  223. }
  224. }
  225. }
  226. }