sysUserRoleModel_test.go 35 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165
  1. package userrole
  2. import (
  3. "context"
  4. "errors"
  5. "math/rand"
  6. "sort"
  7. "testing"
  8. "time"
  9. "perms-system-server/internal/testutil"
  10. "github.com/go-sql-driver/mysql"
  11. "github.com/stretchr/testify/assert"
  12. "github.com/stretchr/testify/require"
  13. "github.com/zeromicro/go-zero/core/stores/sqlx"
  14. )
  15. func randUserRoleId() int64 {
  16. return int64(900000 + rand.Intn(100000))
  17. }
  18. func sortedInt64Copy(a []int64) []int64 {
  19. b := append([]int64(nil), a...)
  20. sort.Slice(b, func(i, j int) bool { return b[i] < b[j] })
  21. return b
  22. }
  23. func int64SliceEqualIgnoreOrder(a, b []int64) bool {
  24. if len(a) != len(b) {
  25. return false
  26. }
  27. aa, bb := sortedInt64Copy(a), sortedInt64Copy(b)
  28. for i := range aa {
  29. if aa[i] != bb[i] {
  30. return false
  31. }
  32. }
  33. return true
  34. }
  35. func insertTestRole(t *testing.T, ctx context.Context, conn sqlx.SqlConn, productCode, name string) int64 {
  36. t.Helper()
  37. ts := time.Now().Unix()
  38. q := "INSERT INTO `sys_role` (`productCode`, `name`, `remark`, `status`, `permsLevel`, `createTime`, `updateTime`) VALUES (?, ?, '', 1, 0, ?, ?)"
  39. res, err := conn.ExecCtx(ctx, q, productCode, name, ts, ts)
  40. if err != nil {
  41. t.Fatalf("insert sys_role: %v", err)
  42. }
  43. id, err := res.LastInsertId()
  44. if err != nil {
  45. t.Fatalf("role LastInsertId: %v", err)
  46. }
  47. return id
  48. }
  49. // TC-0281: 正常插入
  50. func TestSysUserRoleModel_CRUD(t *testing.T) {
  51. ctx := context.Background()
  52. conn := testutil.GetTestSqlConn()
  53. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  54. userId := randUserRoleId()
  55. roleId := randUserRoleId()
  56. ts := time.Now().Unix()
  57. data := &SysUserRole{
  58. UserId: userId,
  59. RoleId: roleId,
  60. CreateTime: ts,
  61. UpdateTime: ts,
  62. }
  63. res, err := m.Insert(ctx, data)
  64. if err != nil {
  65. t.Fatalf("Insert: %v", err)
  66. }
  67. id, err := res.LastInsertId()
  68. if err != nil {
  69. t.Fatalf("LastInsertId: %v", err)
  70. }
  71. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  72. got, err := m.FindOne(ctx, id)
  73. if err != nil {
  74. t.Fatalf("FindOne: %v", err)
  75. }
  76. if got.UserId != userId || got.RoleId != roleId {
  77. t.Fatalf("FindOne mismatch: %+v", got)
  78. }
  79. byPair, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
  80. if err != nil {
  81. t.Fatalf("FindOneByUserIdRoleId: %v", err)
  82. }
  83. if byPair.Id != id {
  84. t.Fatalf("FindOneByUserIdRoleId id want %d got %d", id, byPair.Id)
  85. }
  86. newTs := ts + 1
  87. got.UpdateTime = newTs
  88. if err := m.Update(ctx, got); err != nil {
  89. t.Fatalf("Update: %v", err)
  90. }
  91. updated, err := m.FindOne(ctx, id)
  92. if err != nil {
  93. t.Fatalf("FindOne after update: %v", err)
  94. }
  95. if updated.UpdateTime != newTs {
  96. t.Fatalf("UpdateTime want %d got %d", newTs, updated.UpdateTime)
  97. }
  98. if err := m.Delete(ctx, id); err != nil {
  99. t.Fatalf("Delete: %v", err)
  100. }
  101. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  102. t.Fatalf("after Delete want ErrNotFound got %v", err)
  103. }
  104. }
  105. // TC-0442: 正常查询
  106. func TestSysUserRoleModel_FindRoleIdsByUserId(t *testing.T) {
  107. ctx := context.Background()
  108. conn := testutil.GetTestSqlConn()
  109. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  110. userId := randUserRoleId()
  111. r1, r2 := randUserRoleId(), randUserRoleId()
  112. ts := time.Now().Unix()
  113. var ids []int64
  114. for _, roleId := range []int64{r1, r2} {
  115. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  116. if err != nil {
  117. t.Fatalf("Insert: %v", err)
  118. }
  119. id, _ := res.LastInsertId()
  120. ids = append(ids, id)
  121. }
  122. defer func() {
  123. for _, id := range ids {
  124. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  125. }
  126. }()
  127. got, err := m.FindRoleIdsByUserId(ctx, userId)
  128. if err != nil {
  129. t.Fatalf("FindRoleIdsByUserId: %v", err)
  130. }
  131. if !int64SliceEqualIgnoreOrder(got, []int64{r1, r2}) {
  132. t.Fatalf("got %v want %v", got, []int64{r1, r2})
  133. }
  134. }
  135. // TC-0444: 正常查询
  136. func TestSysUserRoleModel_FindByUserId(t *testing.T) {
  137. ctx := context.Background()
  138. conn := testutil.GetTestSqlConn()
  139. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  140. userId := randUserRoleId()
  141. r1 := randUserRoleId()
  142. ts := time.Now().Unix()
  143. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  144. if err != nil {
  145. t.Fatalf("Insert: %v", err)
  146. }
  147. id, _ := res.LastInsertId()
  148. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  149. list, err := m.FindByUserId(ctx, userId)
  150. if err != nil {
  151. t.Fatalf("FindByUserId: %v", err)
  152. }
  153. if len(list) != 1 || list[0].RoleId != r1 {
  154. t.Fatalf("list %+v", list)
  155. }
  156. }
  157. // TC-0445: 正常删除
  158. func TestSysUserRoleModel_DeleteByUserId(t *testing.T) {
  159. ctx := context.Background()
  160. conn := testutil.GetTestSqlConn()
  161. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  162. u1, u2 := randUserRoleId(), randUserRoleId()
  163. roleId := randUserRoleId()
  164. ts := time.Now().Unix()
  165. res1, err := m.Insert(ctx, &SysUserRole{UserId: u1, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  166. if err != nil {
  167. t.Fatalf("Insert: %v", err)
  168. }
  169. id1, _ := res1.LastInsertId()
  170. res2, err := m.Insert(ctx, &SysUserRole{UserId: u2, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  171. if err != nil {
  172. t.Fatalf("Insert: %v", err)
  173. }
  174. id2, _ := res2.LastInsertId()
  175. defer func() {
  176. testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  177. }()
  178. if err := m.DeleteByUserId(ctx, u1); err != nil {
  179. t.Fatalf("DeleteByUserId: %v", err)
  180. }
  181. if _, err := m.FindOne(ctx, id1); err != ErrNotFound {
  182. t.Fatalf("u1 gone: %v", err)
  183. }
  184. if _, err := m.FindOne(ctx, id2); err != nil {
  185. t.Fatalf("u2 remain: %v", err)
  186. }
  187. }
  188. // TC-0447: 正常删除
  189. func TestSysUserRoleModel_DeleteByRoleIdTx(t *testing.T) {
  190. ctx := context.Background()
  191. conn := testutil.GetTestSqlConn()
  192. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  193. userId := randUserRoleId()
  194. roleId := randUserRoleId()
  195. ts := time.Now().Unix()
  196. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  197. if err != nil {
  198. t.Fatalf("Insert: %v", err)
  199. }
  200. id, _ := res.LastInsertId()
  201. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  202. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  203. return m.DeleteByRoleIdTx(c, session, roleId)
  204. })
  205. if err != nil {
  206. t.Fatalf("DeleteByRoleIdTx: %v", err)
  207. }
  208. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  209. t.Fatalf("after tx delete want ErrNotFound got %v", err)
  210. }
  211. }
  212. // TC-0448: 正常删除
  213. func TestSysUserRoleModel_DeleteByUserIdForProduct(t *testing.T) {
  214. ctx := context.Background()
  215. conn := testutil.GetTestSqlConn()
  216. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  217. pcA := "t_ur_a_" + testutil.UniqueId()
  218. pcB := "t_ur_b_" + testutil.UniqueId()
  219. nameA := "n_a_" + testutil.UniqueId()
  220. nameB := "n_b_" + testutil.UniqueId()
  221. roleA := insertTestRole(t, ctx, conn, pcA, nameA)
  222. roleB := insertTestRole(t, ctx, conn, pcB, nameB)
  223. defer func() {
  224. testutil.CleanTable(ctx, conn, "sys_role", roleA, roleB)
  225. }()
  226. userId := randUserRoleId()
  227. ts := time.Now().Unix()
  228. resA, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleA, CreateTime: ts, UpdateTime: ts})
  229. if err != nil {
  230. t.Fatalf("Insert A: %v", err)
  231. }
  232. idA, _ := resA.LastInsertId()
  233. resB, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleB, CreateTime: ts, UpdateTime: ts})
  234. if err != nil {
  235. t.Fatalf("Insert B: %v", err)
  236. }
  237. idB, _ := resB.LastInsertId()
  238. defer func() {
  239. testutil.CleanTable(ctx, conn, "sys_user_role", idA, idB)
  240. }()
  241. if err := m.DeleteByUserIdForProduct(ctx, userId, pcA); err != nil {
  242. t.Fatalf("DeleteByUserIdForProduct: %v", err)
  243. }
  244. if _, err := m.FindOne(ctx, idA); err != ErrNotFound {
  245. t.Fatalf("roleA link should delete: %v", err)
  246. }
  247. if _, err := m.FindOne(ctx, idB); err != nil {
  248. t.Fatalf("roleB link should remain: %v", err)
  249. }
  250. }
  251. // TC-0290: 记录不存在
  252. func TestSysUserRoleModel_FindOne_NotFound(t *testing.T) {
  253. conn := testutil.GetTestSqlConn()
  254. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  255. _, err := m.FindOne(context.Background(), 999999999999)
  256. if err != ErrNotFound {
  257. t.Fatalf("want ErrNotFound got %v", err)
  258. }
  259. }
  260. // TC-0359: FindOneByUserIdRoleId
  261. func TestSysUserRoleModel_FindOneByUserIdRoleId_NotFound(t *testing.T) {
  262. conn := testutil.GetTestSqlConn()
  263. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  264. _, err := m.FindOneByUserIdRoleId(context.Background(), 999999999, 999999999)
  265. if err != ErrNotFound {
  266. t.Fatalf("want ErrNotFound got %v", err)
  267. }
  268. }
  269. // TC-0443: 无绑定
  270. func TestSysUserRoleModel_FindRoleIdsByUserId_Empty(t *testing.T) {
  271. conn := testutil.GetTestSqlConn()
  272. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  273. got, err := m.FindRoleIdsByUserId(context.Background(), 999999999)
  274. if err != nil {
  275. t.Fatalf("err: %v", err)
  276. }
  277. if len(got) != 0 {
  278. t.Fatalf("want empty got %v", got)
  279. }
  280. }
  281. // TC-0446: 事务内删除
  282. func TestSysUserRoleModel_DeleteByUserIdTx(t *testing.T) {
  283. ctx := context.Background()
  284. conn := testutil.GetTestSqlConn()
  285. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  286. userId := randUserRoleId()
  287. roleId := randUserRoleId()
  288. ts := time.Now().Unix()
  289. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  290. if err != nil {
  291. t.Fatalf("Insert: %v", err)
  292. }
  293. id, _ := res.LastInsertId()
  294. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  295. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  296. return m.DeleteByUserIdTx(c, session, userId)
  297. })
  298. if err != nil {
  299. t.Fatalf("DeleteByUserIdTx: %v", err)
  300. }
  301. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  302. t.Fatalf("after tx delete want ErrNotFound got %v", err)
  303. }
  304. }
  305. // TC-0450: 事务内跨产品删除
  306. func TestSysUserRoleModel_DeleteByUserIdForProductTx(t *testing.T) {
  307. ctx := context.Background()
  308. conn := testutil.GetTestSqlConn()
  309. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  310. pcA := "t_urt_a_" + testutil.UniqueId()
  311. pcB := "t_urt_b_" + testutil.UniqueId()
  312. nameA := "n_ta_" + testutil.UniqueId()
  313. nameB := "n_tb_" + testutil.UniqueId()
  314. roleA := insertTestRole(t, ctx, conn, pcA, nameA)
  315. roleB := insertTestRole(t, ctx, conn, pcB, nameB)
  316. defer testutil.CleanTable(ctx, conn, "sys_role", roleA, roleB)
  317. userId := randUserRoleId()
  318. ts := time.Now().Unix()
  319. resA, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleA, CreateTime: ts, UpdateTime: ts})
  320. if err != nil {
  321. t.Fatalf("Insert A: %v", err)
  322. }
  323. idA, _ := resA.LastInsertId()
  324. resB, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleB, CreateTime: ts, UpdateTime: ts})
  325. if err != nil {
  326. t.Fatalf("Insert B: %v", err)
  327. }
  328. idB, _ := resB.LastInsertId()
  329. defer testutil.CleanTable(ctx, conn, "sys_user_role", idA, idB)
  330. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  331. return m.DeleteByUserIdForProductTx(c, session, userId, pcA)
  332. })
  333. if err != nil {
  334. t.Fatalf("DeleteByUserIdForProductTx: %v", err)
  335. }
  336. if _, err := m.FindOne(ctx, idA); err != ErrNotFound {
  337. t.Fatalf("pcA row should delete: %v", err)
  338. }
  339. if _, err := m.FindOne(ctx, idB); err != nil {
  340. t.Fatalf("pcB row should remain: %v", err)
  341. }
  342. }
  343. // TC-0305: 空列表
  344. func TestSysUserRoleModel_BatchInsert_Empty(t *testing.T) {
  345. conn := testutil.GetTestSqlConn()
  346. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  347. if err := m.BatchInsert(context.Background(), nil); err != nil {
  348. t.Fatalf("nil: %v", err)
  349. }
  350. if err := m.BatchInsert(context.Background(), []*SysUserRole{}); err != nil {
  351. t.Fatalf("empty: %v", err)
  352. }
  353. }
  354. // TC-0324: 空ids
  355. func TestSysUserRoleModel_BatchDelete_Empty(t *testing.T) {
  356. conn := testutil.GetTestSqlConn()
  357. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  358. if err := m.BatchDelete(context.Background(), nil); err != nil {
  359. t.Fatalf("nil: %v", err)
  360. }
  361. if err := m.BatchDelete(context.Background(), []int64{}); err != nil {
  362. t.Fatalf("empty: %v", err)
  363. }
  364. }
  365. // TC-0283: 唯一索引冲突
  366. func TestSysUserRoleModel_Insert_UniqueConflict(t *testing.T) {
  367. ctx := context.Background()
  368. conn := testutil.GetTestSqlConn()
  369. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  370. userId := randUserRoleId()
  371. roleId := randUserRoleId()
  372. ts := time.Now().Unix()
  373. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  374. if err != nil {
  375. t.Fatalf("Insert: %v", err)
  376. }
  377. id, _ := res.LastInsertId()
  378. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  379. _, err = m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  380. if err == nil {
  381. t.Fatal("second Insert want error")
  382. }
  383. var me *mysql.MySQLError
  384. if !errors.As(err, &me) || me.Number != 1062 {
  385. t.Fatalf("want duplicate key 1062, got %v", err)
  386. }
  387. }
  388. // TC-0285: 事务内插入
  389. func TestSysUserRoleModel_InsertWithTx_Normal(t *testing.T) {
  390. ctx := context.Background()
  391. conn := testutil.GetTestSqlConn()
  392. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  393. userId := randUserRoleId()
  394. roleId := randUserRoleId()
  395. ts := time.Now().Unix()
  396. var insertedId int64
  397. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  398. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  399. if err != nil {
  400. return err
  401. }
  402. insertedId, _ = res.LastInsertId()
  403. return nil
  404. })
  405. if err != nil {
  406. t.Fatalf("TransactCtx: %v", err)
  407. }
  408. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  409. got, err := m.FindOne(ctx, insertedId)
  410. if err != nil {
  411. t.Fatalf("FindOne: %v", err)
  412. }
  413. if got.UserId != userId || got.RoleId != roleId {
  414. t.Fatalf("mismatch: %+v", got)
  415. }
  416. }
  417. // TC-0287: 事务回滚后无数据
  418. func TestSysUserRoleModel_InsertWithTx_Rollback(t *testing.T) {
  419. ctx := context.Background()
  420. conn := testutil.GetTestSqlConn()
  421. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  422. userId := randUserRoleId()
  423. roleId := randUserRoleId()
  424. ts := time.Now().Unix()
  425. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  426. _, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  427. if err != nil {
  428. return err
  429. }
  430. return errors.New("rollback")
  431. })
  432. if err == nil || err.Error() != "rollback" {
  433. t.Fatalf("want rollback error got %v", err)
  434. }
  435. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  436. if err != ErrNotFound {
  437. t.Fatalf("after rollback want ErrNotFound got %v", err)
  438. }
  439. }
  440. // TC-0297: 记录不存在
  441. func TestSysUserRoleModel_Update_NotFound(t *testing.T) {
  442. conn := testutil.GetTestSqlConn()
  443. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  444. ts := time.Now().Unix()
  445. err := m.Update(context.Background(), &SysUserRole{
  446. Id: 999999999, UserId: 1, RoleId: 1, CreateTime: ts, UpdateTime: ts,
  447. })
  448. if err != ErrNotFound {
  449. t.Fatalf("want ErrNotFound got %v", err)
  450. }
  451. }
  452. // TC-0298: 事务内更新
  453. func TestSysUserRoleModel_UpdateWithTx(t *testing.T) {
  454. ctx := context.Background()
  455. conn := testutil.GetTestSqlConn()
  456. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  457. userId := randUserRoleId()
  458. roleId := randUserRoleId()
  459. ts := time.Now().Unix()
  460. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  461. if err != nil {
  462. t.Fatalf("Insert: %v", err)
  463. }
  464. id, _ := res.LastInsertId()
  465. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  466. newTs := ts + 100
  467. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  468. return m.UpdateWithTx(c, session, &SysUserRole{
  469. Id: id, UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: newTs,
  470. })
  471. })
  472. if err != nil {
  473. t.Fatalf("UpdateWithTx: %v", err)
  474. }
  475. got, err := m.FindOne(ctx, id)
  476. if err != nil {
  477. t.Fatalf("FindOne: %v", err)
  478. }
  479. if got.UpdateTime != newTs {
  480. t.Fatalf("UpdateTime want %d got %d", newTs, got.UpdateTime)
  481. }
  482. }
  483. // TC-0300: 记录不存在
  484. func TestSysUserRoleModel_Delete_NotFound(t *testing.T) {
  485. conn := testutil.GetTestSqlConn()
  486. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  487. err := m.Delete(context.Background(), 999999999)
  488. if err != ErrNotFound {
  489. t.Fatalf("want ErrNotFound got %v", err)
  490. }
  491. }
  492. // TC-0301: 事务内删除
  493. func TestSysUserRoleModel_DeleteWithTx(t *testing.T) {
  494. ctx := context.Background()
  495. conn := testutil.GetTestSqlConn()
  496. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  497. userId := randUserRoleId()
  498. roleId := randUserRoleId()
  499. ts := time.Now().Unix()
  500. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  501. if err != nil {
  502. t.Fatalf("Insert: %v", err)
  503. }
  504. id, _ := res.LastInsertId()
  505. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  506. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  507. return m.DeleteWithTx(c, session, id)
  508. })
  509. if err != nil {
  510. t.Fatalf("DeleteWithTx: %v", err)
  511. }
  512. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  513. t.Fatalf("after DeleteWithTx want ErrNotFound got %v", err)
  514. }
  515. }
  516. // TC-0303: fn返回错误
  517. func TestSysUserRoleModel_TransactCtx_Rollback(t *testing.T) {
  518. ctx := context.Background()
  519. conn := testutil.GetTestSqlConn()
  520. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  521. userId := randUserRoleId()
  522. roleId := randUserRoleId()
  523. ts := time.Now().Unix()
  524. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  525. if err != nil {
  526. t.Fatalf("Insert: %v", err)
  527. }
  528. id, _ := res.LastInsertId()
  529. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  530. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  531. if err := m.DeleteWithTx(c, session, id); err != nil {
  532. return err
  533. }
  534. return errors.New("rollback")
  535. })
  536. if err == nil || err.Error() != "rollback" {
  537. t.Fatalf("want rollback error got %v", err)
  538. }
  539. got, err := m.FindOne(ctx, id)
  540. if err != nil {
  541. t.Fatalf("after rollback FindOne: %v", err)
  542. }
  543. if got.Id != id {
  544. t.Fatalf("row should still exist")
  545. }
  546. }
  547. // TC-0304: 获取表名
  548. func TestSysUserRoleModel_TableName(t *testing.T) {
  549. conn := testutil.GetTestSqlConn()
  550. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  551. if m.TableName() != "`sys_user_role`" {
  552. t.Fatalf("want `sys_user_role` got %s", m.TableName())
  553. }
  554. }
  555. // TC-0306: 单条记录
  556. func TestSysUserRoleModel_BatchInsert_Single(t *testing.T) {
  557. ctx := context.Background()
  558. conn := testutil.GetTestSqlConn()
  559. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  560. userId := randUserRoleId()
  561. roleId := randUserRoleId()
  562. ts := time.Now().Unix()
  563. if err := m.BatchInsert(ctx, []*SysUserRole{
  564. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  565. }); err != nil {
  566. t.Fatalf("BatchInsert: %v", err)
  567. }
  568. got, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
  569. if err != nil {
  570. t.Fatalf("FindOneByUserIdRoleId: %v", err)
  571. }
  572. defer testutil.CleanTable(ctx, conn, "sys_user_role", got.Id)
  573. if got.UserId != userId || got.RoleId != roleId {
  574. t.Fatalf("mismatch: %+v", got)
  575. }
  576. }
  577. // TC-0307: 多条记录(3条)
  578. func TestSysUserRoleModel_BatchInsert_Multi(t *testing.T) {
  579. ctx := context.Background()
  580. conn := testutil.GetTestSqlConn()
  581. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  582. u1 := randUserRoleId()
  583. r1, r2, r3 := randUserRoleId(), randUserRoleId(), randUserRoleId()
  584. ts := time.Now().Unix()
  585. if err := m.BatchInsert(ctx, []*SysUserRole{
  586. {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
  587. {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
  588. {UserId: u1, RoleId: r3, CreateTime: ts, UpdateTime: ts},
  589. }); err != nil {
  590. t.Fatalf("BatchInsert: %v", err)
  591. }
  592. var ids []int64
  593. for _, roleId := range []int64{r1, r2, r3} {
  594. got, err := m.FindOneByUserIdRoleId(ctx, u1, roleId)
  595. if err != nil {
  596. t.Fatalf("FindOneByUserIdRoleId(%d,%d): %v", u1, roleId, err)
  597. }
  598. ids = append(ids, got.Id)
  599. }
  600. defer func() {
  601. for _, id := range ids {
  602. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  603. }
  604. }()
  605. if len(ids) != 3 {
  606. t.Fatalf("want 3 rows got %d", len(ids))
  607. }
  608. }
  609. // TC-0309: 唯一索引冲突
  610. func TestSysUserRoleModel_BatchInsert_UniqueConflict(t *testing.T) {
  611. ctx := context.Background()
  612. conn := testutil.GetTestSqlConn()
  613. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  614. userId := randUserRoleId()
  615. roleId := randUserRoleId()
  616. ts := time.Now().Unix()
  617. err := m.BatchInsert(ctx, []*SysUserRole{
  618. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  619. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  620. })
  621. if err == nil {
  622. t.Fatal("want error for duplicate")
  623. }
  624. var me *mysql.MySQLError
  625. if !errors.As(err, &me) || me.Number != 1062 {
  626. t.Fatalf("want duplicate key 1062, got %v", err)
  627. }
  628. }
  629. // TC-0314: 空列表
  630. func TestSysUserRoleModel_BatchUpdate_Empty(t *testing.T) {
  631. conn := testutil.GetTestSqlConn()
  632. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  633. if err := m.BatchUpdate(context.Background(), nil); err != nil {
  634. t.Fatalf("nil: %v", err)
  635. }
  636. if err := m.BatchUpdate(context.Background(), []*SysUserRole{}); err != nil {
  637. t.Fatalf("empty: %v", err)
  638. }
  639. }
  640. // TC-0316: 多条记录(3条)
  641. func TestSysUserRoleModel_BatchUpdate_Multi(t *testing.T) {
  642. ctx := context.Background()
  643. conn := testutil.GetTestSqlConn()
  644. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  645. userId := randUserRoleId()
  646. r1, r2 := randUserRoleId(), randUserRoleId()
  647. ts := time.Now().Unix()
  648. res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  649. if err != nil {
  650. t.Fatalf("Insert1: %v", err)
  651. }
  652. id1, _ := res1.LastInsertId()
  653. res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
  654. if err != nil {
  655. t.Fatalf("Insert2: %v", err)
  656. }
  657. id2, _ := res2.LastInsertId()
  658. defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  659. newTs := ts + 100
  660. newR1, newR2 := randUserRoleId(), randUserRoleId()
  661. err = m.BatchUpdate(ctx, []*SysUserRole{
  662. {Id: id1, UserId: userId, RoleId: newR1, CreateTime: ts, UpdateTime: newTs},
  663. {Id: id2, UserId: userId, RoleId: newR2, CreateTime: ts, UpdateTime: newTs},
  664. })
  665. if err != nil {
  666. t.Fatalf("BatchUpdate: %v", err)
  667. }
  668. got1, err := m.FindOne(ctx, id1)
  669. if err != nil {
  670. t.Fatalf("FindOne1: %v", err)
  671. }
  672. if got1.UpdateTime != newTs || got1.RoleId != newR1 {
  673. t.Fatalf("got1 mismatch: %+v", got1)
  674. }
  675. got2, err := m.FindOne(ctx, id2)
  676. if err != nil {
  677. t.Fatalf("FindOne2: %v", err)
  678. }
  679. if got2.UpdateTime != newTs || got2.RoleId != newR2 {
  680. t.Fatalf("got2 mismatch: %+v", got2)
  681. }
  682. }
  683. // TC-0326: 多个id(3个)
  684. func TestSysUserRoleModel_BatchDelete_Multi(t *testing.T) {
  685. ctx := context.Background()
  686. conn := testutil.GetTestSqlConn()
  687. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  688. ts := time.Now().Unix()
  689. var ids []int64
  690. for i := 0; i < 3; i++ {
  691. res, err := m.Insert(ctx, &SysUserRole{
  692. UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
  693. })
  694. if err != nil {
  695. t.Fatalf("Insert: %v", err)
  696. }
  697. id, _ := res.LastInsertId()
  698. ids = append(ids, id)
  699. }
  700. defer func() {
  701. for _, id := range ids {
  702. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  703. }
  704. }()
  705. if err := m.BatchDelete(ctx, ids); err != nil {
  706. t.Fatalf("BatchDelete: %v", err)
  707. }
  708. for _, id := range ids {
  709. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  710. t.Fatalf("id %d should be deleted: %v", id, err)
  711. }
  712. }
  713. }
  714. // TC-0325: 单个id
  715. func TestSysUserRoleModel_BatchDelete_Single(t *testing.T) {
  716. ctx := context.Background()
  717. conn := testutil.GetTestSqlConn()
  718. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  719. userId := randUserRoleId()
  720. roleId := randUserRoleId()
  721. ts := time.Now().Unix()
  722. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  723. if err != nil {
  724. t.Fatalf("Insert: %v", err)
  725. }
  726. id, _ := res.LastInsertId()
  727. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  728. if err := m.BatchDelete(ctx, []int64{id}); err != nil {
  729. t.Fatalf("BatchDelete: %v", err)
  730. }
  731. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  732. t.Fatalf("want ErrNotFound got %v", err)
  733. }
  734. }
  735. // TC-0327: 包含不存在id
  736. func TestSysUserRoleModel_BatchDelete_ContainsNonExist(t *testing.T) {
  737. ctx := context.Background()
  738. conn := testutil.GetTestSqlConn()
  739. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  740. userId := randUserRoleId()
  741. roleId := randUserRoleId()
  742. ts := time.Now().Unix()
  743. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  744. if err != nil {
  745. t.Fatalf("Insert: %v", err)
  746. }
  747. id, _ := res.LastInsertId()
  748. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  749. if err := m.BatchDelete(ctx, []int64{id, 999999999}); err != nil {
  750. t.Fatalf("BatchDelete: %v", err)
  751. }
  752. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  753. t.Fatalf("want ErrNotFound got %v", err)
  754. }
  755. }
  756. // TC-0312: 正常多条
  757. func TestSysUserRoleModel_BatchInsertWithTx_Normal(t *testing.T) {
  758. ctx := context.Background()
  759. conn := testutil.GetTestSqlConn()
  760. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  761. u1 := randUserRoleId()
  762. r1, r2 := randUserRoleId(), randUserRoleId()
  763. ts := time.Now().Unix()
  764. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  765. return m.BatchInsertWithTx(c, session, []*SysUserRole{
  766. {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
  767. {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
  768. })
  769. })
  770. if err != nil {
  771. t.Fatalf("BatchInsertWithTx: %v", err)
  772. }
  773. got1, err := m.FindOneByUserIdRoleId(ctx, u1, r1)
  774. if err != nil {
  775. t.Fatalf("FindOne r1: %v", err)
  776. }
  777. got2, err := m.FindOneByUserIdRoleId(ctx, u1, r2)
  778. if err != nil {
  779. t.Fatalf("FindOne r2: %v", err)
  780. }
  781. defer testutil.CleanTable(ctx, conn, "sys_user_role", got1.Id, got2.Id)
  782. }
  783. // TC-0311: 空列表
  784. func TestSysUserRoleModel_BatchInsertWithTx_Empty(t *testing.T) {
  785. ctx := context.Background()
  786. conn := testutil.GetTestSqlConn()
  787. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  788. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  789. if err := m.BatchInsertWithTx(c, session, nil); err != nil {
  790. return err
  791. }
  792. return m.BatchInsertWithTx(c, session, []*SysUserRole{})
  793. })
  794. if err != nil {
  795. t.Fatalf("BatchInsertWithTx empty: %v", err)
  796. }
  797. }
  798. // TC-0313: 事务回滚
  799. func TestSysUserRoleModel_BatchInsertWithTx_Rollback(t *testing.T) {
  800. ctx := context.Background()
  801. conn := testutil.GetTestSqlConn()
  802. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  803. userId := randUserRoleId()
  804. roleId := randUserRoleId()
  805. ts := time.Now().Unix()
  806. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  807. if err := m.BatchInsertWithTx(c, session, []*SysUserRole{
  808. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  809. }); err != nil {
  810. return err
  811. }
  812. return errors.New("rollback")
  813. })
  814. if err == nil || err.Error() != "rollback" {
  815. t.Fatalf("want rollback error got %v", err)
  816. }
  817. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  818. if err != ErrNotFound {
  819. t.Fatalf("after rollback want ErrNotFound got %v", err)
  820. }
  821. }
  822. // TC-0320: 正常多条
  823. func TestSysUserRoleModel_BatchUpdateWithTx_Normal(t *testing.T) {
  824. ctx := context.Background()
  825. conn := testutil.GetTestSqlConn()
  826. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  827. userId := randUserRoleId()
  828. r1, r2 := randUserRoleId(), randUserRoleId()
  829. ts := time.Now().Unix()
  830. res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  831. if err != nil {
  832. t.Fatalf("Insert1: %v", err)
  833. }
  834. id1, _ := res1.LastInsertId()
  835. res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
  836. if err != nil {
  837. t.Fatalf("Insert2: %v", err)
  838. }
  839. id2, _ := res2.LastInsertId()
  840. defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  841. newTs := ts + 200
  842. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  843. return m.BatchUpdateWithTx(c, session, []*SysUserRole{
  844. {Id: id1, UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: newTs},
  845. {Id: id2, UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: newTs},
  846. })
  847. })
  848. if err != nil {
  849. t.Fatalf("BatchUpdateWithTx: %v", err)
  850. }
  851. got1, err := m.FindOne(ctx, id1)
  852. if err != nil {
  853. t.Fatalf("FindOne1: %v", err)
  854. }
  855. if got1.UpdateTime != newTs {
  856. t.Fatalf("got1 UpdateTime want %d got %d", newTs, got1.UpdateTime)
  857. }
  858. }
  859. // TC-0319: 空列表
  860. func TestSysUserRoleModel_BatchUpdateWithTx_Empty(t *testing.T) {
  861. ctx := context.Background()
  862. conn := testutil.GetTestSqlConn()
  863. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  864. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  865. if err := m.BatchUpdateWithTx(c, session, nil); err != nil {
  866. return err
  867. }
  868. return m.BatchUpdateWithTx(c, session, []*SysUserRole{})
  869. })
  870. if err != nil {
  871. t.Fatalf("BatchUpdateWithTx empty: %v", err)
  872. }
  873. }
  874. // TC-0329: 正常多条
  875. func TestSysUserRoleModel_BatchDeleteWithTx_Normal(t *testing.T) {
  876. ctx := context.Background()
  877. conn := testutil.GetTestSqlConn()
  878. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  879. ts := time.Now().Unix()
  880. var ids []int64
  881. for i := 0; i < 2; i++ {
  882. res, err := m.Insert(ctx, &SysUserRole{
  883. UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
  884. })
  885. if err != nil {
  886. t.Fatalf("Insert: %v", err)
  887. }
  888. id, _ := res.LastInsertId()
  889. ids = append(ids, id)
  890. }
  891. defer func() {
  892. for _, id := range ids {
  893. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  894. }
  895. }()
  896. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  897. return m.BatchDeleteWithTx(c, session, ids)
  898. })
  899. if err != nil {
  900. t.Fatalf("BatchDeleteWithTx: %v", err)
  901. }
  902. for _, id := range ids {
  903. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  904. t.Fatalf("id %d should be deleted: %v", id, err)
  905. }
  906. }
  907. }
  908. // TC-0328: 空ids
  909. func TestSysUserRoleModel_BatchDeleteWithTx_Empty(t *testing.T) {
  910. ctx := context.Background()
  911. conn := testutil.GetTestSqlConn()
  912. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  913. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  914. if err := m.BatchDeleteWithTx(c, session, nil); err != nil {
  915. return err
  916. }
  917. return m.BatchDeleteWithTx(c, session, []int64{})
  918. })
  919. if err != nil {
  920. t.Fatalf("BatchDeleteWithTx empty: %v", err)
  921. }
  922. }
  923. // TC-0294: 事务内可见性
  924. func TestSysUserRoleModel_FindOneWithTx_InsertThenFind(t *testing.T) {
  925. ctx := context.Background()
  926. conn := testutil.GetTestSqlConn()
  927. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  928. userId := randUserRoleId()
  929. roleId := randUserRoleId()
  930. ts := time.Now().Unix()
  931. var insertedId int64
  932. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  933. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  934. if err != nil {
  935. return err
  936. }
  937. insertedId, err = res.LastInsertId()
  938. if err != nil {
  939. return err
  940. }
  941. got, err := m.FindOneWithTx(c, session, insertedId)
  942. require.NoError(t, err)
  943. assert.Equal(t, insertedId, got.Id)
  944. assert.Equal(t, userId, got.UserId)
  945. assert.Equal(t, roleId, got.RoleId)
  946. return nil
  947. })
  948. require.NoError(t, err)
  949. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  950. }
  951. // TC-0293: 事务内记录不存在
  952. func TestSysUserRoleModel_FindOneWithTx_NotFound(t *testing.T) {
  953. ctx := context.Background()
  954. m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  955. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  956. _, err := m.FindOneWithTx(c, session, 999999999999)
  957. require.ErrorIs(t, err, ErrNotFound)
  958. return nil
  959. })
  960. require.NoError(t, err)
  961. }
  962. // TC-0360: FindOneByUserIdRoleIdWithTx
  963. func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_InsertThenFind(t *testing.T) {
  964. ctx := context.Background()
  965. conn := testutil.GetTestSqlConn()
  966. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  967. userId := randUserRoleId()
  968. roleId := randUserRoleId()
  969. ts := time.Now().Unix()
  970. var insertedId int64
  971. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  972. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  973. if err != nil {
  974. return err
  975. }
  976. insertedId, err = res.LastInsertId()
  977. if err != nil {
  978. return err
  979. }
  980. got, err := m.FindOneByUserIdRoleIdWithTx(c, session, userId, roleId)
  981. require.NoError(t, err)
  982. assert.Equal(t, insertedId, got.Id)
  983. assert.Equal(t, userId, got.UserId)
  984. assert.Equal(t, roleId, got.RoleId)
  985. return nil
  986. })
  987. require.NoError(t, err)
  988. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  989. }
  990. // TC-0361: FindOneByUserIdRoleIdWithTx
  991. func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_NotFound(t *testing.T) {
  992. ctx := context.Background()
  993. m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  994. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  995. _, err := m.FindOneByUserIdRoleIdWithTx(c, session, 999999999, 999999999)
  996. require.ErrorIs(t, err, ErrNotFound)
  997. return nil
  998. })
  999. require.NoError(t, err)
  1000. }
  1001. // TC-0451: FindUserIdsByRoleId 正常返回角色下用户ID列表
  1002. func TestSysUserRoleModel_FindUserIdsByRoleId_Normal(t *testing.T) {
  1003. ctx := context.Background()
  1004. conn := testutil.GetTestSqlConn()
  1005. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  1006. roleId := randUserRoleId()
  1007. u1, u2 := randUserRoleId(), randUserRoleId()
  1008. ts := time.Now().Unix()
  1009. var ids []int64
  1010. for _, userId := range []int64{u1, u2} {
  1011. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  1012. require.NoError(t, err)
  1013. id, _ := res.LastInsertId()
  1014. ids = append(ids, id)
  1015. }
  1016. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", ids...) })
  1017. got, err := m.FindUserIdsByRoleId(ctx, roleId)
  1018. require.NoError(t, err)
  1019. require.Len(t, got, 2)
  1020. assert.True(t, int64SliceEqualIgnoreOrder(got, []int64{u1, u2}))
  1021. }
  1022. // TC-0452: FindUserIdsByRoleId 无绑定返回空
  1023. func TestSysUserRoleModel_FindUserIdsByRoleId_Empty(t *testing.T) {
  1024. conn := testutil.GetTestSqlConn()
  1025. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  1026. got, err := m.FindUserIdsByRoleId(context.Background(), 999999998)
  1027. require.NoError(t, err)
  1028. require.Empty(t, got)
  1029. }