sysProductMemberModel_test.go 34 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071
  1. package productmember
  2. import (
  3. "context"
  4. "errors"
  5. "math/rand"
  6. "testing"
  7. "time"
  8. "perms-system-server/internal/testutil"
  9. "github.com/go-sql-driver/mysql"
  10. "github.com/stretchr/testify/assert"
  11. "github.com/stretchr/testify/require"
  12. "github.com/zeromicro/go-zero/core/stores/sqlx"
  13. )
  14. func randProductMemberUserId() int64 {
  15. return int64(900000 + rand.Intn(100000))
  16. }
  17. // TC-0310: 正常插入
  18. func TestSysProductMemberModel_CRUD(t *testing.T) {
  19. ctx := context.Background()
  20. conn := testutil.GetTestSqlConn()
  21. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  22. pc := "t_pm_" + testutil.UniqueId()
  23. userId := randProductMemberUserId()
  24. ts := time.Now().Unix()
  25. data := &SysProductMember{
  26. ProductCode: pc,
  27. UserId: userId,
  28. MemberType: "MEMBER",
  29. Status: 1,
  30. CreateTime: ts,
  31. UpdateTime: ts,
  32. }
  33. res, err := m.Insert(ctx, data)
  34. if err != nil {
  35. t.Fatalf("Insert: %v", err)
  36. }
  37. id, err := res.LastInsertId()
  38. if err != nil {
  39. t.Fatalf("LastInsertId: %v", err)
  40. }
  41. defer testutil.CleanTable(ctx, conn, "sys_product_member", id)
  42. got, err := m.FindOne(ctx, id)
  43. if err != nil {
  44. t.Fatalf("FindOne: %v", err)
  45. }
  46. if got.ProductCode != pc || got.UserId != userId {
  47. t.Fatalf("FindOne mismatch: %+v", got)
  48. }
  49. byPair, err := m.FindOneByProductCodeUserId(ctx, pc, userId)
  50. if err != nil {
  51. t.Fatalf("FindOneByProductCodeUserId: %v", err)
  52. }
  53. if byPair.Id != id {
  54. t.Fatalf("FindOneByProductCodeUserId id want %d got %d", id, byPair.Id)
  55. }
  56. newTs := ts + 1
  57. got.Status = 2
  58. got.UpdateTime = newTs
  59. if err := m.Update(ctx, got); err != nil {
  60. t.Fatalf("Update: %v", err)
  61. }
  62. updated, err := m.FindOne(ctx, id)
  63. if err != nil {
  64. t.Fatalf("FindOne after update: %v", err)
  65. }
  66. if updated.Status != 2 || updated.UpdateTime != newTs {
  67. t.Fatalf("after Update: %+v", updated)
  68. }
  69. if err := m.Delete(ctx, id); err != nil {
  70. t.Fatalf("Delete: %v", err)
  71. }
  72. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  73. t.Fatalf("after Delete want ErrNotFound got %v", err)
  74. }
  75. }
  76. // TC-0475: 正常分页
  77. func TestSysProductMemberModel_FindListByProductCode(t *testing.T) {
  78. ctx := context.Background()
  79. conn := testutil.GetTestSqlConn()
  80. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  81. pc := "t_pm_page_" + testutil.UniqueId()
  82. ts := time.Now().Unix()
  83. var ids []int64
  84. for i := 0; i < 5; i++ {
  85. res, err := m.Insert(ctx, &SysProductMember{
  86. ProductCode: pc,
  87. UserId: randProductMemberUserId(),
  88. MemberType: "MEMBER",
  89. Status: 1,
  90. CreateTime: ts,
  91. UpdateTime: ts,
  92. })
  93. if err != nil {
  94. t.Fatalf("Insert: %v", err)
  95. }
  96. id, _ := res.LastInsertId()
  97. ids = append(ids, id)
  98. }
  99. defer func() {
  100. for _, id := range ids {
  101. testutil.CleanTable(ctx, conn, "sys_product_member", id)
  102. }
  103. }()
  104. list, total, err := m.FindListByProductCode(ctx, pc, 1, 2)
  105. if err != nil {
  106. t.Fatalf("page1: %v", err)
  107. }
  108. if total != 5 || len(list) != 2 {
  109. t.Fatalf("page1 total=%d len=%d", total, len(list))
  110. }
  111. list2, total2, err := m.FindListByProductCode(ctx, pc, 2, 2)
  112. if err != nil {
  113. t.Fatalf("page2: %v", err)
  114. }
  115. if total2 != 5 || len(list2) != 2 {
  116. t.Fatalf("page2 total=%d len=%d", total2, len(list2))
  117. }
  118. list3, total3, err := m.FindListByProductCode(ctx, pc, 3, 2)
  119. if err != nil {
  120. t.Fatalf("page3: %v", err)
  121. }
  122. if total3 != 5 || len(list3) != 1 {
  123. t.Fatalf("page3 total=%d len=%d", total3, len(list3))
  124. }
  125. }
  126. // TC-0477: 正常批量
  127. func TestSysProductMemberModel_FindMapByProductCodeUserIds(t *testing.T) {
  128. ctx := context.Background()
  129. conn := testutil.GetTestSqlConn()
  130. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  131. pc := "t_pm_map_" + testutil.UniqueId()
  132. u1, u2 := randProductMemberUserId(), randProductMemberUserId()
  133. uMissing := randProductMemberUserId()
  134. ts := time.Now().Unix()
  135. res1, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: u1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  136. if err != nil {
  137. t.Fatalf("Insert: %v", err)
  138. }
  139. id1, _ := res1.LastInsertId()
  140. res2, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: u2, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  141. if err != nil {
  142. t.Fatalf("Insert: %v", err)
  143. }
  144. id2, _ := res2.LastInsertId()
  145. defer func() {
  146. testutil.CleanTable(ctx, conn, "sys_product_member", id1, id2)
  147. }()
  148. empty, err := m.FindMapByProductCodeUserIds(ctx, pc, nil)
  149. if err != nil {
  150. t.Fatalf("empty nil: %v", err)
  151. }
  152. if len(empty) != 0 {
  153. t.Fatalf("empty nil want len 0 got %d", len(empty))
  154. }
  155. empty2, err := m.FindMapByProductCodeUserIds(ctx, pc, []int64{})
  156. if err != nil {
  157. t.Fatalf("empty slice: %v", err)
  158. }
  159. if len(empty2) != 0 {
  160. t.Fatalf("empty slice want len 0 got %d", len(empty2))
  161. }
  162. m1, err := m.FindMapByProductCodeUserIds(ctx, pc, []int64{u1, u2})
  163. if err != nil {
  164. t.Fatalf("normal: %v", err)
  165. }
  166. if len(m1) != 2 || m1[u1] == nil || m1[u2] == nil {
  167. t.Fatalf("normal map: %+v", m1)
  168. }
  169. m2, err := m.FindMapByProductCodeUserIds(ctx, pc, []int64{u1, uMissing})
  170. if err != nil {
  171. t.Fatalf("partial: %v", err)
  172. }
  173. if len(m2) != 1 || m2[u1] == nil || m2[uMissing] != nil {
  174. t.Fatalf("partial map: %+v", m2)
  175. }
  176. }
  177. // TC-0336: 多条记录(3条)
  178. func TestSysProductMemberModel_BatchInsert(t *testing.T) {
  179. ctx := context.Background()
  180. conn := testutil.GetTestSqlConn()
  181. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  182. pc := "t_pm_bi_" + testutil.UniqueId()
  183. u1, u2 := randProductMemberUserId(), randProductMemberUserId()
  184. ts := time.Now().Unix()
  185. list := []*SysProductMember{
  186. {Id: 930000001, ProductCode: pc, UserId: u1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts},
  187. {Id: 930000002, ProductCode: pc, UserId: u2, MemberType: "ADMIN", Status: 1, CreateTime: ts, UpdateTime: ts},
  188. }
  189. if err := m.BatchInsert(ctx, list); err != nil {
  190. t.Fatalf("BatchInsert: %v", err)
  191. }
  192. var rows []struct {
  193. Id int64 `db:"id"`
  194. }
  195. q := "SELECT `id` FROM `sys_product_member` WHERE `productCode` = ? ORDER BY `id`"
  196. if err := conn.QueryRowsCtx(ctx, &rows, q, pc); err != nil {
  197. t.Fatalf("query: %v", err)
  198. }
  199. defer func() {
  200. for _, r := range rows {
  201. testutil.CleanTable(ctx, conn, "sys_product_member", r.Id)
  202. }
  203. }()
  204. if len(rows) != 2 {
  205. t.Fatalf("want 2 rows got %d", len(rows))
  206. }
  207. }
  208. // TC-0312: 唯一索引冲突
  209. func TestSysProductMemberModel_DuplicateConstraint(t *testing.T) {
  210. ctx := context.Background()
  211. conn := testutil.GetTestSqlConn()
  212. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  213. pc := "t_pm_dup_" + testutil.UniqueId()
  214. userId := randProductMemberUserId()
  215. ts := time.Now().Unix()
  216. res, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  217. if err != nil {
  218. t.Fatalf("Insert: %v", err)
  219. }
  220. id, _ := res.LastInsertId()
  221. defer testutil.CleanTable(ctx, conn, "sys_product_member", id)
  222. _, err = m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: userId, MemberType: "ADMIN", Status: 1, CreateTime: ts, UpdateTime: ts})
  223. if err == nil {
  224. t.Fatal("second Insert want error")
  225. }
  226. var me *mysql.MySQLError
  227. if !errors.As(err, &me) || me.Number != 1062 {
  228. t.Fatalf("want duplicate key 1062, got %v", err)
  229. }
  230. }
  231. // TC-0319: 记录不存在
  232. func TestSysProductMemberModel_FindOne_NotFound(t *testing.T) {
  233. conn := testutil.GetTestSqlConn()
  234. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  235. _, err := m.FindOne(context.Background(), 999999999999)
  236. if err != ErrNotFound {
  237. t.Fatalf("want ErrNotFound got %v", err)
  238. }
  239. }
  240. // TC-0392: FindOneByProductCodeUserId
  241. func TestSysProductMemberModel_FindOneByProductCodeUserId_NotFound(t *testing.T) {
  242. conn := testutil.GetTestSqlConn()
  243. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  244. _, err := m.FindOneByProductCodeUserId(context.Background(), "notexist_"+testutil.UniqueId(), 999999999)
  245. if err != ErrNotFound {
  246. t.Fatalf("want ErrNotFound got %v", err)
  247. }
  248. }
  249. // TC-0476: 空结果
  250. func TestSysProductMemberModel_FindListByProductCode_Empty(t *testing.T) {
  251. conn := testutil.GetTestSqlConn()
  252. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  253. list, total, err := m.FindListByProductCode(context.Background(), "empty_"+testutil.UniqueId(), 1, 10)
  254. if err != nil {
  255. t.Fatalf("err: %v", err)
  256. }
  257. if total != 0 || len(list) != 0 {
  258. t.Fatalf("want empty got total=%d len=%d", total, len(list))
  259. }
  260. }
  261. // TC-0334: 空列表
  262. func TestSysProductMemberModel_BatchInsert_Empty(t *testing.T) {
  263. conn := testutil.GetTestSqlConn()
  264. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  265. if err := m.BatchInsert(context.Background(), nil); err != nil {
  266. t.Fatalf("nil: %v", err)
  267. }
  268. if err := m.BatchInsert(context.Background(), []*SysProductMember{}); err != nil {
  269. t.Fatalf("empty: %v", err)
  270. }
  271. }
  272. // TC-0353: 空ids
  273. func TestSysProductMemberModel_BatchDelete_Empty(t *testing.T) {
  274. conn := testutil.GetTestSqlConn()
  275. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  276. if err := m.BatchDelete(context.Background(), nil); err != nil {
  277. t.Fatalf("nil: %v", err)
  278. }
  279. if err := m.BatchDelete(context.Background(), []int64{}); err != nil {
  280. t.Fatalf("empty: %v", err)
  281. }
  282. }
  283. // TC-0314: 事务内插入
  284. func TestSysProductMemberModel_InsertWithTx_Normal(t *testing.T) {
  285. ctx := context.Background()
  286. conn := testutil.GetTestSqlConn()
  287. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  288. pc := "t_pm_itx_" + testutil.UniqueId()
  289. userId := randProductMemberUserId()
  290. ts := time.Now().Unix()
  291. var insertedId int64
  292. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  293. res, err := m.InsertWithTx(c, session, &SysProductMember{
  294. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  295. })
  296. if err != nil {
  297. return err
  298. }
  299. insertedId, _ = res.LastInsertId()
  300. return nil
  301. })
  302. if err != nil {
  303. t.Fatalf("TransactCtx: %v", err)
  304. }
  305. defer testutil.CleanTable(ctx, conn, "sys_product_member", insertedId)
  306. got, err := m.FindOne(ctx, insertedId)
  307. if err != nil {
  308. t.Fatalf("FindOne: %v", err)
  309. }
  310. if got.ProductCode != pc || got.UserId != userId {
  311. t.Fatalf("mismatch: %+v", got)
  312. }
  313. }
  314. // TC-0316: 事务回滚后无数据
  315. func TestSysProductMemberModel_InsertWithTx_Rollback(t *testing.T) {
  316. ctx := context.Background()
  317. conn := testutil.GetTestSqlConn()
  318. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  319. pc := "t_pm_irb_" + testutil.UniqueId()
  320. userId := randProductMemberUserId()
  321. ts := time.Now().Unix()
  322. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  323. _, err := m.InsertWithTx(c, session, &SysProductMember{
  324. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  325. })
  326. if err != nil {
  327. return err
  328. }
  329. return errors.New("rollback")
  330. })
  331. if err == nil || err.Error() != "rollback" {
  332. t.Fatalf("want rollback error got %v", err)
  333. }
  334. _, err = m.FindOneByProductCodeUserId(ctx, pc, userId)
  335. if err != ErrNotFound {
  336. t.Fatalf("after rollback want ErrNotFound got %v", err)
  337. }
  338. }
  339. // TC-0326: 记录不存在
  340. func TestSysProductMemberModel_Update_NotFound(t *testing.T) {
  341. conn := testutil.GetTestSqlConn()
  342. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  343. ts := time.Now().Unix()
  344. err := m.Update(context.Background(), &SysProductMember{
  345. Id: 999999999, ProductCode: "nope", UserId: 1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  346. })
  347. if err != ErrNotFound {
  348. t.Fatalf("want ErrNotFound got %v", err)
  349. }
  350. }
  351. // TC-0327: 事务内更新
  352. func TestSysProductMemberModel_UpdateWithTx(t *testing.T) {
  353. ctx := context.Background()
  354. conn := testutil.GetTestSqlConn()
  355. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  356. pc := "t_pm_utx_" + testutil.UniqueId()
  357. userId := randProductMemberUserId()
  358. ts := time.Now().Unix()
  359. res, err := m.Insert(ctx, &SysProductMember{
  360. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  361. })
  362. if err != nil {
  363. t.Fatalf("Insert: %v", err)
  364. }
  365. id, _ := res.LastInsertId()
  366. defer testutil.CleanTable(ctx, conn, "sys_product_member", id)
  367. newTs := ts + 100
  368. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  369. return m.UpdateWithTx(c, session, &SysProductMember{
  370. Id: id, ProductCode: pc, UserId: userId, MemberType: "ADMIN", Status: 2, CreateTime: ts, UpdateTime: newTs,
  371. })
  372. })
  373. if err != nil {
  374. t.Fatalf("UpdateWithTx: %v", err)
  375. }
  376. got, err := m.FindOne(ctx, id)
  377. if err != nil {
  378. t.Fatalf("FindOne: %v", err)
  379. }
  380. if got.MemberType != "ADMIN" || got.Status != 2 || got.UpdateTime != newTs {
  381. t.Fatalf("mismatch: %+v", got)
  382. }
  383. }
  384. // TC-0329: 记录不存在
  385. func TestSysProductMemberModel_Delete_NotFound(t *testing.T) {
  386. conn := testutil.GetTestSqlConn()
  387. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  388. err := m.Delete(context.Background(), 999999999)
  389. if err != ErrNotFound {
  390. t.Fatalf("want ErrNotFound got %v", err)
  391. }
  392. }
  393. // TC-0330: 事务内删除
  394. func TestSysProductMemberModel_DeleteWithTx(t *testing.T) {
  395. ctx := context.Background()
  396. conn := testutil.GetTestSqlConn()
  397. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  398. pc := "t_pm_dtx_" + testutil.UniqueId()
  399. userId := randProductMemberUserId()
  400. ts := time.Now().Unix()
  401. res, err := m.Insert(ctx, &SysProductMember{
  402. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  403. })
  404. if err != nil {
  405. t.Fatalf("Insert: %v", err)
  406. }
  407. id, _ := res.LastInsertId()
  408. defer testutil.CleanTable(ctx, conn, "sys_product_member", id)
  409. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  410. return m.DeleteWithTx(c, session, id)
  411. })
  412. if err != nil {
  413. t.Fatalf("DeleteWithTx: %v", err)
  414. }
  415. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  416. t.Fatalf("after DeleteWithTx want ErrNotFound got %v", err)
  417. }
  418. }
  419. // TC-0331: 正常事务
  420. func TestSysProductMemberModel_TransactCtx_CommitAndRollback(t *testing.T) {
  421. ctx := context.Background()
  422. conn := testutil.GetTestSqlConn()
  423. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  424. pc := "t_pm_txc_" + testutil.UniqueId()
  425. userId := randProductMemberUserId()
  426. ts := time.Now().Unix()
  427. var insertedId int64
  428. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  429. res, err := m.InsertWithTx(c, session, &SysProductMember{
  430. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  431. })
  432. if err != nil {
  433. return err
  434. }
  435. insertedId, _ = res.LastInsertId()
  436. return nil
  437. })
  438. if err != nil {
  439. t.Fatalf("commit: %v", err)
  440. }
  441. defer testutil.CleanTable(ctx, conn, "sys_product_member", insertedId)
  442. got, err := m.FindOne(ctx, insertedId)
  443. if err != nil {
  444. t.Fatalf("FindOne after commit: %v", err)
  445. }
  446. if got.ProductCode != pc {
  447. t.Fatalf("productCode mismatch: %s", got.ProductCode)
  448. }
  449. pc2 := "t_pm_txr_" + testutil.UniqueId()
  450. userId2 := randProductMemberUserId()
  451. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  452. _, err := m.InsertWithTx(c, session, &SysProductMember{
  453. ProductCode: pc2, UserId: userId2, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  454. })
  455. if err != nil {
  456. return err
  457. }
  458. return errors.New("rollback")
  459. })
  460. if err == nil || err.Error() != "rollback" {
  461. t.Fatalf("want rollback error got %v", err)
  462. }
  463. _, err = m.FindOneByProductCodeUserId(ctx, pc2, userId2)
  464. if err != ErrNotFound {
  465. t.Fatalf("after rollback want ErrNotFound got %v", err)
  466. }
  467. }
  468. // TC-0333: 获取表名
  469. func TestSysProductMemberModel_TableName(t *testing.T) {
  470. conn := testutil.GetTestSqlConn()
  471. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  472. if m.TableName() != "`sys_product_member`" {
  473. t.Fatalf("want `sys_product_member` got %s", m.TableName())
  474. }
  475. }
  476. // TC-0335: 单条记录
  477. func TestSysProductMemberModel_BatchInsert_Single(t *testing.T) {
  478. ctx := context.Background()
  479. conn := testutil.GetTestSqlConn()
  480. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  481. pc := "t_pm_bis_" + testutil.UniqueId()
  482. userId := randProductMemberUserId()
  483. ts := time.Now().Unix()
  484. if err := m.BatchInsert(ctx, []*SysProductMember{
  485. {ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts},
  486. }); err != nil {
  487. t.Fatalf("BatchInsert: %v", err)
  488. }
  489. got, err := m.FindOneByProductCodeUserId(ctx, pc, userId)
  490. if err != nil {
  491. t.Fatalf("FindOneByProductCodeUserId: %v", err)
  492. }
  493. defer testutil.CleanTable(ctx, conn, "sys_product_member", got.Id)
  494. if got.MemberType != "MEMBER" {
  495. t.Fatalf("mismatch: %+v", got)
  496. }
  497. }
  498. // TC-0338: 唯一索引冲突
  499. func TestSysProductMemberModel_BatchInsert_UniqueConflict(t *testing.T) {
  500. ctx := context.Background()
  501. conn := testutil.GetTestSqlConn()
  502. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  503. pc := "t_pm_bic_" + testutil.UniqueId()
  504. userId := randProductMemberUserId()
  505. ts := time.Now().Unix()
  506. err := m.BatchInsert(ctx, []*SysProductMember{
  507. {ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts},
  508. {ProductCode: pc, UserId: userId, MemberType: "ADMIN", Status: 1, CreateTime: ts, UpdateTime: ts},
  509. })
  510. if err == nil {
  511. t.Fatal("want error for duplicate")
  512. }
  513. var me *mysql.MySQLError
  514. if !errors.As(err, &me) || me.Number != 1062 {
  515. t.Fatalf("want duplicate key 1062, got %v", err)
  516. }
  517. }
  518. // TC-0343: 空列表
  519. func TestSysProductMemberModel_BatchUpdate_Empty(t *testing.T) {
  520. conn := testutil.GetTestSqlConn()
  521. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  522. if err := m.BatchUpdate(context.Background(), nil); err != nil {
  523. t.Fatalf("nil: %v", err)
  524. }
  525. if err := m.BatchUpdate(context.Background(), []*SysProductMember{}); err != nil {
  526. t.Fatalf("empty: %v", err)
  527. }
  528. }
  529. // TC-0345: 多条记录(3条)
  530. func TestSysProductMemberModel_BatchUpdate_Multi(t *testing.T) {
  531. ctx := context.Background()
  532. conn := testutil.GetTestSqlConn()
  533. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  534. pc := "t_pm_bum_" + testutil.UniqueId()
  535. u1, u2 := randProductMemberUserId(), randProductMemberUserId()
  536. ts := time.Now().Unix()
  537. res1, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: u1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  538. if err != nil {
  539. t.Fatalf("Insert1: %v", err)
  540. }
  541. id1, _ := res1.LastInsertId()
  542. res2, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: u2, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  543. if err != nil {
  544. t.Fatalf("Insert2: %v", err)
  545. }
  546. id2, _ := res2.LastInsertId()
  547. defer testutil.CleanTable(ctx, conn, "sys_product_member", id1, id2)
  548. newTs := ts + 100
  549. err = m.BatchUpdate(ctx, []*SysProductMember{
  550. {Id: id1, ProductCode: pc, UserId: u1, MemberType: "ADMIN", Status: 2, CreateTime: ts, UpdateTime: newTs},
  551. {Id: id2, ProductCode: pc, UserId: u2, MemberType: "ADMIN", Status: 2, CreateTime: ts, UpdateTime: newTs},
  552. })
  553. if err != nil {
  554. t.Fatalf("BatchUpdate: %v", err)
  555. }
  556. got1, err := m.FindOne(ctx, id1)
  557. if err != nil {
  558. t.Fatalf("FindOne1: %v", err)
  559. }
  560. if got1.MemberType != "ADMIN" || got1.Status != 2 || got1.UpdateTime != newTs {
  561. t.Fatalf("got1 mismatch: %+v", got1)
  562. }
  563. got2, err := m.FindOne(ctx, id2)
  564. if err != nil {
  565. t.Fatalf("FindOne2: %v", err)
  566. }
  567. if got2.MemberType != "ADMIN" || got2.Status != 2 || got2.UpdateTime != newTs {
  568. t.Fatalf("got2 mismatch: %+v", got2)
  569. }
  570. }
  571. // TC-0355: 多个id(3个)
  572. func TestSysProductMemberModel_BatchDelete_Multi(t *testing.T) {
  573. ctx := context.Background()
  574. conn := testutil.GetTestSqlConn()
  575. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  576. pc := "t_pm_bdm_" + testutil.UniqueId()
  577. ts := time.Now().Unix()
  578. var ids []int64
  579. for i := 0; i < 3; i++ {
  580. res, err := m.Insert(ctx, &SysProductMember{
  581. ProductCode: pc, UserId: randProductMemberUserId(), MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  582. })
  583. if err != nil {
  584. t.Fatalf("Insert: %v", err)
  585. }
  586. id, _ := res.LastInsertId()
  587. ids = append(ids, id)
  588. }
  589. defer func() {
  590. for _, id := range ids {
  591. testutil.CleanTable(ctx, conn, "sys_product_member", id)
  592. }
  593. }()
  594. if err := m.BatchDelete(ctx, ids); err != nil {
  595. t.Fatalf("BatchDelete: %v", err)
  596. }
  597. for _, id := range ids {
  598. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  599. t.Fatalf("id %d should be deleted: %v", id, err)
  600. }
  601. }
  602. }
  603. // TC-0354: 单个id
  604. func TestSysProductMemberModel_BatchDelete_Single(t *testing.T) {
  605. ctx := context.Background()
  606. conn := testutil.GetTestSqlConn()
  607. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  608. pc := "t_pm_bds_" + testutil.UniqueId()
  609. userId := randProductMemberUserId()
  610. ts := time.Now().Unix()
  611. res, err := m.Insert(ctx, &SysProductMember{
  612. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  613. })
  614. if err != nil {
  615. t.Fatalf("Insert: %v", err)
  616. }
  617. id, _ := res.LastInsertId()
  618. defer testutil.CleanTable(ctx, conn, "sys_product_member", id)
  619. if err := m.BatchDelete(ctx, []int64{id}); err != nil {
  620. t.Fatalf("BatchDelete: %v", err)
  621. }
  622. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  623. t.Fatalf("want ErrNotFound got %v", err)
  624. }
  625. }
  626. // TC-0356: 包含不存在id
  627. func TestSysProductMemberModel_BatchDelete_ContainsNonExist(t *testing.T) {
  628. ctx := context.Background()
  629. conn := testutil.GetTestSqlConn()
  630. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  631. pc := "t_pm_bdn_" + testutil.UniqueId()
  632. userId := randProductMemberUserId()
  633. ts := time.Now().Unix()
  634. res, err := m.Insert(ctx, &SysProductMember{
  635. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  636. })
  637. if err != nil {
  638. t.Fatalf("Insert: %v", err)
  639. }
  640. id, _ := res.LastInsertId()
  641. defer testutil.CleanTable(ctx, conn, "sys_product_member", id)
  642. if err := m.BatchDelete(ctx, []int64{id, 999999999}); err != nil {
  643. t.Fatalf("BatchDelete: %v", err)
  644. }
  645. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  646. t.Fatalf("want ErrNotFound got %v", err)
  647. }
  648. }
  649. // TC-0341: 正常多条
  650. func TestSysProductMemberModel_BatchInsertWithTx_Normal(t *testing.T) {
  651. ctx := context.Background()
  652. conn := testutil.GetTestSqlConn()
  653. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  654. pc := "t_pm_bitn_" + testutil.UniqueId()
  655. u1, u2 := randProductMemberUserId(), randProductMemberUserId()
  656. ts := time.Now().Unix()
  657. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  658. return m.BatchInsertWithTx(c, session, []*SysProductMember{
  659. {ProductCode: pc, UserId: u1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts},
  660. {ProductCode: pc, UserId: u2, MemberType: "ADMIN", Status: 1, CreateTime: ts, UpdateTime: ts},
  661. })
  662. })
  663. if err != nil {
  664. t.Fatalf("BatchInsertWithTx: %v", err)
  665. }
  666. got1, err := m.FindOneByProductCodeUserId(ctx, pc, u1)
  667. if err != nil {
  668. t.Fatalf("FindOne u1: %v", err)
  669. }
  670. got2, err := m.FindOneByProductCodeUserId(ctx, pc, u2)
  671. if err != nil {
  672. t.Fatalf("FindOne u2: %v", err)
  673. }
  674. defer testutil.CleanTable(ctx, conn, "sys_product_member", got1.Id, got2.Id)
  675. }
  676. // TC-0340: 空列表
  677. func TestSysProductMemberModel_BatchInsertWithTx_Empty(t *testing.T) {
  678. ctx := context.Background()
  679. conn := testutil.GetTestSqlConn()
  680. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  681. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  682. if err := m.BatchInsertWithTx(c, session, nil); err != nil {
  683. return err
  684. }
  685. return m.BatchInsertWithTx(c, session, []*SysProductMember{})
  686. })
  687. if err != nil {
  688. t.Fatalf("BatchInsertWithTx empty: %v", err)
  689. }
  690. }
  691. // TC-0342: 事务回滚
  692. func TestSysProductMemberModel_BatchInsertWithTx_Rollback(t *testing.T) {
  693. ctx := context.Background()
  694. conn := testutil.GetTestSqlConn()
  695. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  696. pc := "t_pm_bitr_" + testutil.UniqueId()
  697. userId := randProductMemberUserId()
  698. ts := time.Now().Unix()
  699. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  700. if err := m.BatchInsertWithTx(c, session, []*SysProductMember{
  701. {ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts},
  702. }); err != nil {
  703. return err
  704. }
  705. return errors.New("rollback")
  706. })
  707. if err == nil || err.Error() != "rollback" {
  708. t.Fatalf("want rollback error got %v", err)
  709. }
  710. _, err = m.FindOneByProductCodeUserId(ctx, pc, userId)
  711. if err != ErrNotFound {
  712. t.Fatalf("after rollback want ErrNotFound got %v", err)
  713. }
  714. }
  715. // TC-0349: 正常多条
  716. func TestSysProductMemberModel_BatchUpdateWithTx_Normal(t *testing.T) {
  717. ctx := context.Background()
  718. conn := testutil.GetTestSqlConn()
  719. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  720. pc := "t_pm_butn_" + testutil.UniqueId()
  721. u1, u2 := randProductMemberUserId(), randProductMemberUserId()
  722. ts := time.Now().Unix()
  723. res1, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: u1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  724. if err != nil {
  725. t.Fatalf("Insert1: %v", err)
  726. }
  727. id1, _ := res1.LastInsertId()
  728. res2, err := m.Insert(ctx, &SysProductMember{ProductCode: pc, UserId: u2, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts})
  729. if err != nil {
  730. t.Fatalf("Insert2: %v", err)
  731. }
  732. id2, _ := res2.LastInsertId()
  733. defer testutil.CleanTable(ctx, conn, "sys_product_member", id1, id2)
  734. newTs := ts + 200
  735. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  736. return m.BatchUpdateWithTx(c, session, []*SysProductMember{
  737. {Id: id1, ProductCode: pc, UserId: u1, MemberType: "ADMIN", Status: 2, CreateTime: ts, UpdateTime: newTs},
  738. {Id: id2, ProductCode: pc, UserId: u2, MemberType: "ADMIN", Status: 2, CreateTime: ts, UpdateTime: newTs},
  739. })
  740. })
  741. if err != nil {
  742. t.Fatalf("BatchUpdateWithTx: %v", err)
  743. }
  744. got1, err := m.FindOne(ctx, id1)
  745. if err != nil {
  746. t.Fatalf("FindOne1: %v", err)
  747. }
  748. if got1.MemberType != "ADMIN" || got1.Status != 2 || got1.UpdateTime != newTs {
  749. t.Fatalf("got1 mismatch: %+v", got1)
  750. }
  751. }
  752. // TC-0348: 空列表
  753. func TestSysProductMemberModel_BatchUpdateWithTx_Empty(t *testing.T) {
  754. ctx := context.Background()
  755. conn := testutil.GetTestSqlConn()
  756. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  757. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  758. if err := m.BatchUpdateWithTx(c, session, nil); err != nil {
  759. return err
  760. }
  761. return m.BatchUpdateWithTx(c, session, []*SysProductMember{})
  762. })
  763. if err != nil {
  764. t.Fatalf("BatchUpdateWithTx empty: %v", err)
  765. }
  766. }
  767. // TC-0358: 正常多条
  768. func TestSysProductMemberModel_BatchDeleteWithTx_Normal(t *testing.T) {
  769. ctx := context.Background()
  770. conn := testutil.GetTestSqlConn()
  771. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  772. pc := "t_pm_bdtn_" + testutil.UniqueId()
  773. ts := time.Now().Unix()
  774. var ids []int64
  775. for i := 0; i < 2; i++ {
  776. res, err := m.Insert(ctx, &SysProductMember{
  777. ProductCode: pc, UserId: randProductMemberUserId(), MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  778. })
  779. if err != nil {
  780. t.Fatalf("Insert: %v", err)
  781. }
  782. id, _ := res.LastInsertId()
  783. ids = append(ids, id)
  784. }
  785. defer func() {
  786. for _, id := range ids {
  787. testutil.CleanTable(ctx, conn, "sys_product_member", id)
  788. }
  789. }()
  790. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  791. return m.BatchDeleteWithTx(c, session, ids)
  792. })
  793. if err != nil {
  794. t.Fatalf("BatchDeleteWithTx: %v", err)
  795. }
  796. for _, id := range ids {
  797. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  798. t.Fatalf("id %d should be deleted: %v", id, err)
  799. }
  800. }
  801. }
  802. // TC-0357: 空ids
  803. func TestSysProductMemberModel_BatchDeleteWithTx_Empty(t *testing.T) {
  804. ctx := context.Background()
  805. conn := testutil.GetTestSqlConn()
  806. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  807. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  808. if err := m.BatchDeleteWithTx(c, session, nil); err != nil {
  809. return err
  810. }
  811. return m.BatchDeleteWithTx(c, session, []int64{})
  812. })
  813. if err != nil {
  814. t.Fatalf("BatchDeleteWithTx empty: %v", err)
  815. }
  816. }
  817. // TC-0323: 事务内可见性
  818. func TestSysProductMemberModel_FindOneWithTx_InsertThenFind(t *testing.T) {
  819. ctx := context.Background()
  820. conn := testutil.GetTestSqlConn()
  821. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  822. pc := "t_pm_fotx_" + testutil.UniqueId()
  823. userId := randProductMemberUserId()
  824. ts := time.Now().Unix()
  825. var insertedId int64
  826. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  827. res, err := m.InsertWithTx(c, session, &SysProductMember{
  828. ProductCode: pc, UserId: userId, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  829. })
  830. if err != nil {
  831. return err
  832. }
  833. insertedId, err = res.LastInsertId()
  834. if err != nil {
  835. return err
  836. }
  837. got, err := m.FindOneWithTx(c, session, insertedId)
  838. if err != nil {
  839. return err
  840. }
  841. assert.Equal(t, pc, got.ProductCode)
  842. assert.Equal(t, userId, got.UserId)
  843. assert.Equal(t, "MEMBER", got.MemberType)
  844. return nil
  845. })
  846. require.NoError(t, err)
  847. defer testutil.CleanTable(ctx, conn, "sys_product_member", insertedId)
  848. }
  849. // TC-0322: 事务内记录不存在
  850. func TestSysProductMemberModel_FindOneWithTx_NotFound(t *testing.T) {
  851. ctx := context.Background()
  852. conn := testutil.GetTestSqlConn()
  853. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  854. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  855. _, err := m.FindOneWithTx(c, session, 999999999999)
  856. require.ErrorIs(t, err, ErrNotFound)
  857. return nil
  858. })
  859. require.NoError(t, err)
  860. }
  861. // TC-0393: FindOneByProductCodeUserIdWithTx
  862. func TestSysProductMemberModel_FindOneByProductCodeUserIdWithTx_InsertThenFind(t *testing.T) {
  863. ctx := context.Background()
  864. conn := testutil.GetTestSqlConn()
  865. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  866. pc := "t_pm_fbytx_" + testutil.UniqueId()
  867. userId := randProductMemberUserId()
  868. ts := time.Now().Unix()
  869. var insertedId int64
  870. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  871. res, err := m.InsertWithTx(c, session, &SysProductMember{
  872. ProductCode: pc, UserId: userId, MemberType: "ADMIN", Status: 1, CreateTime: ts, UpdateTime: ts,
  873. })
  874. if err != nil {
  875. return err
  876. }
  877. insertedId, err = res.LastInsertId()
  878. if err != nil {
  879. return err
  880. }
  881. got, err := m.FindOneByProductCodeUserIdWithTx(c, session, pc, userId)
  882. if err != nil {
  883. return err
  884. }
  885. assert.Equal(t, insertedId, got.Id)
  886. assert.Equal(t, pc, got.ProductCode)
  887. assert.Equal(t, userId, got.UserId)
  888. return nil
  889. })
  890. require.NoError(t, err)
  891. defer testutil.CleanTable(ctx, conn, "sys_product_member", insertedId)
  892. }
  893. // TC-0394: FindOneByProductCodeUserIdWithTx
  894. func TestSysProductMemberModel_FindOneByProductCodeUserIdWithTx_NotFound(t *testing.T) {
  895. ctx := context.Background()
  896. conn := testutil.GetTestSqlConn()
  897. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  898. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  899. _, err := m.FindOneByProductCodeUserIdWithTx(c, session, "notexist_"+testutil.UniqueId(), 999999999)
  900. require.ErrorIs(t, err, ErrNotFound)
  901. return nil
  902. })
  903. require.NoError(t, err)
  904. }
  905. // TC-0478: FindMapByProductCodeUserIds - 空userIds
  906. func TestSysProductMemberModel_FindMapByProductCodeUserIds_EmptyUserIds(t *testing.T) {
  907. conn := testutil.GetTestSqlConn()
  908. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  909. pc := "t_pm_empty_" + testutil.UniqueId()
  910. m1, err := m.FindMapByProductCodeUserIds(context.Background(), pc, nil)
  911. require.NoError(t, err)
  912. assert.Empty(t, m1)
  913. m2, err := m.FindMapByProductCodeUserIds(context.Background(), pc, []int64{})
  914. require.NoError(t, err)
  915. assert.Empty(t, m2)
  916. }
  917. // TC-0480: FindMapByProductCodeUserIds - map key正确
  918. func TestSysProductMemberModel_FindMapByProductCodeUserIds_MapKeysCorrect(t *testing.T) {
  919. ctx := context.Background()
  920. conn := testutil.GetTestSqlConn()
  921. m := NewSysProductMemberModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  922. pc := "t_pm_key_" + testutil.UniqueId()
  923. u1, u2 := randProductMemberUserId(), randProductMemberUserId()
  924. ts := time.Now().Unix()
  925. res1, err := m.Insert(ctx, &SysProductMember{
  926. ProductCode: pc, UserId: u1, MemberType: "MEMBER", Status: 1, CreateTime: ts, UpdateTime: ts,
  927. })
  928. require.NoError(t, err)
  929. id1, _ := res1.LastInsertId()
  930. res2, err := m.Insert(ctx, &SysProductMember{
  931. ProductCode: pc, UserId: u2, MemberType: "ADMIN", Status: 1, CreateTime: ts, UpdateTime: ts,
  932. })
  933. require.NoError(t, err)
  934. id2, _ := res2.LastInsertId()
  935. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_product_member", id1, id2) })
  936. result, err := m.FindMapByProductCodeUserIds(ctx, pc, []int64{u1, u2})
  937. require.NoError(t, err)
  938. require.Len(t, result, 2)
  939. v1, ok1 := result[u1]
  940. require.True(t, ok1, "map should contain key userId=%d", u1)
  941. assert.Equal(t, id1, v1.Id)
  942. assert.Equal(t, pc, v1.ProductCode)
  943. assert.Equal(t, u1, v1.UserId)
  944. assert.Equal(t, "MEMBER", v1.MemberType)
  945. v2, ok2 := result[u2]
  946. require.True(t, ok2, "map should contain key userId=%d", u2)
  947. assert.Equal(t, id2, v2.Id)
  948. assert.Equal(t, pc, v2.ProductCode)
  949. assert.Equal(t, u2, v2.UserId)
  950. assert.Equal(t, "ADMIN", v2.MemberType)
  951. }