sysUserRoleModel_test.go 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033
  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-0310: 正常插入
  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-0467: 正常查询
  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. // TC-0469: 正常删除
  137. func TestSysUserRoleModel_DeleteByRoleIdTx(t *testing.T) {
  138. ctx := context.Background()
  139. conn := testutil.GetTestSqlConn()
  140. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  141. userId := randUserRoleId()
  142. roleId := randUserRoleId()
  143. ts := time.Now().Unix()
  144. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  145. if err != nil {
  146. t.Fatalf("Insert: %v", err)
  147. }
  148. id, _ := res.LastInsertId()
  149. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  150. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  151. return m.DeleteByRoleIdTx(c, session, roleId)
  152. })
  153. if err != nil {
  154. t.Fatalf("DeleteByRoleIdTx: %v", err)
  155. }
  156. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  157. t.Fatalf("after tx delete want ErrNotFound got %v", err)
  158. }
  159. }
  160. // TC-0319: 记录不存在
  161. func TestSysUserRoleModel_FindOne_NotFound(t *testing.T) {
  162. conn := testutil.GetTestSqlConn()
  163. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  164. _, err := m.FindOne(context.Background(), 999999999999)
  165. if err != ErrNotFound {
  166. t.Fatalf("want ErrNotFound got %v", err)
  167. }
  168. }
  169. // TC-0388: FindOneByUserIdRoleId
  170. func TestSysUserRoleModel_FindOneByUserIdRoleId_NotFound(t *testing.T) {
  171. conn := testutil.GetTestSqlConn()
  172. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  173. _, err := m.FindOneByUserIdRoleId(context.Background(), 999999999, 999999999)
  174. if err != ErrNotFound {
  175. t.Fatalf("want ErrNotFound got %v", err)
  176. }
  177. }
  178. // TC-0468: 无绑定
  179. func TestSysUserRoleModel_FindRoleIdsByUserId_Empty(t *testing.T) {
  180. conn := testutil.GetTestSqlConn()
  181. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  182. got, err := m.FindRoleIdsByUserId(context.Background(), 999999999)
  183. if err != nil {
  184. t.Fatalf("err: %v", err)
  185. }
  186. if len(got) != 0 {
  187. t.Fatalf("want empty got %v", got)
  188. }
  189. }
  190. // TC-0470: 事务内跨产品删除
  191. func TestSysUserRoleModel_DeleteByUserIdForProductTx(t *testing.T) {
  192. ctx := context.Background()
  193. conn := testutil.GetTestSqlConn()
  194. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  195. pcA := "t_urt_a_" + testutil.UniqueId()
  196. pcB := "t_urt_b_" + testutil.UniqueId()
  197. nameA := "n_ta_" + testutil.UniqueId()
  198. nameB := "n_tb_" + testutil.UniqueId()
  199. roleA := insertTestRole(t, ctx, conn, pcA, nameA)
  200. roleB := insertTestRole(t, ctx, conn, pcB, nameB)
  201. defer testutil.CleanTable(ctx, conn, "sys_role", roleA, roleB)
  202. userId := randUserRoleId()
  203. ts := time.Now().Unix()
  204. resA, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleA, CreateTime: ts, UpdateTime: ts})
  205. if err != nil {
  206. t.Fatalf("Insert A: %v", err)
  207. }
  208. idA, _ := resA.LastInsertId()
  209. resB, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleB, CreateTime: ts, UpdateTime: ts})
  210. if err != nil {
  211. t.Fatalf("Insert B: %v", err)
  212. }
  213. idB, _ := resB.LastInsertId()
  214. defer testutil.CleanTable(ctx, conn, "sys_user_role", idA, idB)
  215. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  216. return m.DeleteByUserIdForProductTx(c, session, userId, pcA)
  217. })
  218. if err != nil {
  219. t.Fatalf("DeleteByUserIdForProductTx: %v", err)
  220. }
  221. if _, err := m.FindOne(ctx, idA); err != ErrNotFound {
  222. t.Fatalf("pcA row should delete: %v", err)
  223. }
  224. if _, err := m.FindOne(ctx, idB); err != nil {
  225. t.Fatalf("pcB row should remain: %v", err)
  226. }
  227. }
  228. // TC-0334: 空列表
  229. func TestSysUserRoleModel_BatchInsert_Empty(t *testing.T) {
  230. conn := testutil.GetTestSqlConn()
  231. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  232. if err := m.BatchInsert(context.Background(), nil); err != nil {
  233. t.Fatalf("nil: %v", err)
  234. }
  235. if err := m.BatchInsert(context.Background(), []*SysUserRole{}); err != nil {
  236. t.Fatalf("empty: %v", err)
  237. }
  238. }
  239. // TC-0353: 空ids
  240. func TestSysUserRoleModel_BatchDelete_Empty(t *testing.T) {
  241. conn := testutil.GetTestSqlConn()
  242. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  243. if err := m.BatchDelete(context.Background(), nil); err != nil {
  244. t.Fatalf("nil: %v", err)
  245. }
  246. if err := m.BatchDelete(context.Background(), []int64{}); err != nil {
  247. t.Fatalf("empty: %v", err)
  248. }
  249. }
  250. // TC-0312: 唯一索引冲突
  251. func TestSysUserRoleModel_Insert_UniqueConflict(t *testing.T) {
  252. ctx := context.Background()
  253. conn := testutil.GetTestSqlConn()
  254. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  255. userId := randUserRoleId()
  256. roleId := randUserRoleId()
  257. ts := time.Now().Unix()
  258. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  259. if err != nil {
  260. t.Fatalf("Insert: %v", err)
  261. }
  262. id, _ := res.LastInsertId()
  263. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  264. _, err = m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  265. if err == nil {
  266. t.Fatal("second Insert want error")
  267. }
  268. var me *mysql.MySQLError
  269. if !errors.As(err, &me) || me.Number != 1062 {
  270. t.Fatalf("want duplicate key 1062, got %v", err)
  271. }
  272. }
  273. // TC-0314: 事务内插入
  274. func TestSysUserRoleModel_InsertWithTx_Normal(t *testing.T) {
  275. ctx := context.Background()
  276. conn := testutil.GetTestSqlConn()
  277. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  278. userId := randUserRoleId()
  279. roleId := randUserRoleId()
  280. ts := time.Now().Unix()
  281. var insertedId int64
  282. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  283. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  284. if err != nil {
  285. return err
  286. }
  287. insertedId, _ = res.LastInsertId()
  288. return nil
  289. })
  290. if err != nil {
  291. t.Fatalf("TransactCtx: %v", err)
  292. }
  293. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  294. got, err := m.FindOne(ctx, insertedId)
  295. if err != nil {
  296. t.Fatalf("FindOne: %v", err)
  297. }
  298. if got.UserId != userId || got.RoleId != roleId {
  299. t.Fatalf("mismatch: %+v", got)
  300. }
  301. }
  302. // TC-0316: 事务回滚后无数据
  303. func TestSysUserRoleModel_InsertWithTx_Rollback(t *testing.T) {
  304. ctx := context.Background()
  305. conn := testutil.GetTestSqlConn()
  306. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  307. userId := randUserRoleId()
  308. roleId := randUserRoleId()
  309. ts := time.Now().Unix()
  310. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  311. _, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  312. if err != nil {
  313. return err
  314. }
  315. return errors.New("rollback")
  316. })
  317. if err == nil || err.Error() != "rollback" {
  318. t.Fatalf("want rollback error got %v", err)
  319. }
  320. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  321. if err != ErrNotFound {
  322. t.Fatalf("after rollback want ErrNotFound got %v", err)
  323. }
  324. }
  325. // TC-0326: 记录不存在
  326. func TestSysUserRoleModel_Update_NotFound(t *testing.T) {
  327. conn := testutil.GetTestSqlConn()
  328. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  329. ts := time.Now().Unix()
  330. err := m.Update(context.Background(), &SysUserRole{
  331. Id: 999999999, UserId: 1, RoleId: 1, CreateTime: ts, UpdateTime: ts,
  332. })
  333. if err != ErrNotFound {
  334. t.Fatalf("want ErrNotFound got %v", err)
  335. }
  336. }
  337. // TC-0327: 事务内更新
  338. func TestSysUserRoleModel_UpdateWithTx(t *testing.T) {
  339. ctx := context.Background()
  340. conn := testutil.GetTestSqlConn()
  341. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  342. userId := randUserRoleId()
  343. roleId := randUserRoleId()
  344. ts := time.Now().Unix()
  345. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  346. if err != nil {
  347. t.Fatalf("Insert: %v", err)
  348. }
  349. id, _ := res.LastInsertId()
  350. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  351. newTs := ts + 100
  352. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  353. return m.UpdateWithTx(c, session, &SysUserRole{
  354. Id: id, UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: newTs,
  355. })
  356. })
  357. if err != nil {
  358. t.Fatalf("UpdateWithTx: %v", err)
  359. }
  360. got, err := m.FindOne(ctx, id)
  361. if err != nil {
  362. t.Fatalf("FindOne: %v", err)
  363. }
  364. if got.UpdateTime != newTs {
  365. t.Fatalf("UpdateTime want %d got %d", newTs, got.UpdateTime)
  366. }
  367. }
  368. // TC-0329: 记录不存在
  369. func TestSysUserRoleModel_Delete_NotFound(t *testing.T) {
  370. conn := testutil.GetTestSqlConn()
  371. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  372. err := m.Delete(context.Background(), 999999999)
  373. if err != ErrNotFound {
  374. t.Fatalf("want ErrNotFound got %v", err)
  375. }
  376. }
  377. // TC-0330: 事务内删除
  378. func TestSysUserRoleModel_DeleteWithTx(t *testing.T) {
  379. ctx := context.Background()
  380. conn := testutil.GetTestSqlConn()
  381. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  382. userId := randUserRoleId()
  383. roleId := randUserRoleId()
  384. ts := time.Now().Unix()
  385. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  386. if err != nil {
  387. t.Fatalf("Insert: %v", err)
  388. }
  389. id, _ := res.LastInsertId()
  390. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  391. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  392. return m.DeleteWithTx(c, session, id)
  393. })
  394. if err != nil {
  395. t.Fatalf("DeleteWithTx: %v", err)
  396. }
  397. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  398. t.Fatalf("after DeleteWithTx want ErrNotFound got %v", err)
  399. }
  400. }
  401. // TC-0332: fn返回错误
  402. func TestSysUserRoleModel_TransactCtx_Rollback(t *testing.T) {
  403. ctx := context.Background()
  404. conn := testutil.GetTestSqlConn()
  405. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  406. userId := randUserRoleId()
  407. roleId := randUserRoleId()
  408. ts := time.Now().Unix()
  409. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  410. if err != nil {
  411. t.Fatalf("Insert: %v", err)
  412. }
  413. id, _ := res.LastInsertId()
  414. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  415. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  416. if err := m.DeleteWithTx(c, session, id); err != nil {
  417. return err
  418. }
  419. return errors.New("rollback")
  420. })
  421. if err == nil || err.Error() != "rollback" {
  422. t.Fatalf("want rollback error got %v", err)
  423. }
  424. got, err := m.FindOne(ctx, id)
  425. if err != nil {
  426. t.Fatalf("after rollback FindOne: %v", err)
  427. }
  428. if got.Id != id {
  429. t.Fatalf("row should still exist")
  430. }
  431. }
  432. // TC-0333: 获取表名
  433. func TestSysUserRoleModel_TableName(t *testing.T) {
  434. conn := testutil.GetTestSqlConn()
  435. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  436. if m.TableName() != "`sys_user_role`" {
  437. t.Fatalf("want `sys_user_role` got %s", m.TableName())
  438. }
  439. }
  440. // TC-0335: 单条记录
  441. func TestSysUserRoleModel_BatchInsert_Single(t *testing.T) {
  442. ctx := context.Background()
  443. conn := testutil.GetTestSqlConn()
  444. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  445. userId := randUserRoleId()
  446. roleId := randUserRoleId()
  447. ts := time.Now().Unix()
  448. if err := m.BatchInsert(ctx, []*SysUserRole{
  449. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  450. }); err != nil {
  451. t.Fatalf("BatchInsert: %v", err)
  452. }
  453. got, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
  454. if err != nil {
  455. t.Fatalf("FindOneByUserIdRoleId: %v", err)
  456. }
  457. defer testutil.CleanTable(ctx, conn, "sys_user_role", got.Id)
  458. if got.UserId != userId || got.RoleId != roleId {
  459. t.Fatalf("mismatch: %+v", got)
  460. }
  461. }
  462. // TC-0336: 多条记录(3条)
  463. func TestSysUserRoleModel_BatchInsert_Multi(t *testing.T) {
  464. ctx := context.Background()
  465. conn := testutil.GetTestSqlConn()
  466. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  467. u1 := randUserRoleId()
  468. r1, r2, r3 := randUserRoleId(), randUserRoleId(), randUserRoleId()
  469. ts := time.Now().Unix()
  470. if err := m.BatchInsert(ctx, []*SysUserRole{
  471. {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
  472. {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
  473. {UserId: u1, RoleId: r3, CreateTime: ts, UpdateTime: ts},
  474. }); err != nil {
  475. t.Fatalf("BatchInsert: %v", err)
  476. }
  477. var ids []int64
  478. for _, roleId := range []int64{r1, r2, r3} {
  479. got, err := m.FindOneByUserIdRoleId(ctx, u1, roleId)
  480. if err != nil {
  481. t.Fatalf("FindOneByUserIdRoleId(%d,%d): %v", u1, roleId, err)
  482. }
  483. ids = append(ids, got.Id)
  484. }
  485. defer func() {
  486. for _, id := range ids {
  487. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  488. }
  489. }()
  490. if len(ids) != 3 {
  491. t.Fatalf("want 3 rows got %d", len(ids))
  492. }
  493. }
  494. // TC-0338: 唯一索引冲突
  495. func TestSysUserRoleModel_BatchInsert_UniqueConflict(t *testing.T) {
  496. ctx := context.Background()
  497. conn := testutil.GetTestSqlConn()
  498. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  499. userId := randUserRoleId()
  500. roleId := randUserRoleId()
  501. ts := time.Now().Unix()
  502. err := m.BatchInsert(ctx, []*SysUserRole{
  503. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  504. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  505. })
  506. if err == nil {
  507. t.Fatal("want error for duplicate")
  508. }
  509. var me *mysql.MySQLError
  510. if !errors.As(err, &me) || me.Number != 1062 {
  511. t.Fatalf("want duplicate key 1062, got %v", err)
  512. }
  513. }
  514. // TC-0343: 空列表
  515. func TestSysUserRoleModel_BatchUpdate_Empty(t *testing.T) {
  516. conn := testutil.GetTestSqlConn()
  517. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  518. if err := m.BatchUpdate(context.Background(), nil); err != nil {
  519. t.Fatalf("nil: %v", err)
  520. }
  521. if err := m.BatchUpdate(context.Background(), []*SysUserRole{}); err != nil {
  522. t.Fatalf("empty: %v", err)
  523. }
  524. }
  525. // TC-0345: 多条记录(3条)
  526. func TestSysUserRoleModel_BatchUpdate_Multi(t *testing.T) {
  527. ctx := context.Background()
  528. conn := testutil.GetTestSqlConn()
  529. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  530. userId := randUserRoleId()
  531. r1, r2 := randUserRoleId(), randUserRoleId()
  532. ts := time.Now().Unix()
  533. res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  534. if err != nil {
  535. t.Fatalf("Insert1: %v", err)
  536. }
  537. id1, _ := res1.LastInsertId()
  538. res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
  539. if err != nil {
  540. t.Fatalf("Insert2: %v", err)
  541. }
  542. id2, _ := res2.LastInsertId()
  543. defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  544. newTs := ts + 100
  545. newR1, newR2 := randUserRoleId(), randUserRoleId()
  546. err = m.BatchUpdate(ctx, []*SysUserRole{
  547. {Id: id1, UserId: userId, RoleId: newR1, CreateTime: ts, UpdateTime: newTs},
  548. {Id: id2, UserId: userId, RoleId: newR2, CreateTime: ts, UpdateTime: newTs},
  549. })
  550. if err != nil {
  551. t.Fatalf("BatchUpdate: %v", err)
  552. }
  553. got1, err := m.FindOne(ctx, id1)
  554. if err != nil {
  555. t.Fatalf("FindOne1: %v", err)
  556. }
  557. if got1.UpdateTime != newTs || got1.RoleId != newR1 {
  558. t.Fatalf("got1 mismatch: %+v", got1)
  559. }
  560. got2, err := m.FindOne(ctx, id2)
  561. if err != nil {
  562. t.Fatalf("FindOne2: %v", err)
  563. }
  564. if got2.UpdateTime != newTs || got2.RoleId != newR2 {
  565. t.Fatalf("got2 mismatch: %+v", got2)
  566. }
  567. }
  568. // TC-0355: 多个id(3个)
  569. func TestSysUserRoleModel_BatchDelete_Multi(t *testing.T) {
  570. ctx := context.Background()
  571. conn := testutil.GetTestSqlConn()
  572. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  573. ts := time.Now().Unix()
  574. var ids []int64
  575. for i := 0; i < 3; i++ {
  576. res, err := m.Insert(ctx, &SysUserRole{
  577. UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
  578. })
  579. if err != nil {
  580. t.Fatalf("Insert: %v", err)
  581. }
  582. id, _ := res.LastInsertId()
  583. ids = append(ids, id)
  584. }
  585. defer func() {
  586. for _, id := range ids {
  587. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  588. }
  589. }()
  590. if err := m.BatchDelete(ctx, ids); err != nil {
  591. t.Fatalf("BatchDelete: %v", err)
  592. }
  593. for _, id := range ids {
  594. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  595. t.Fatalf("id %d should be deleted: %v", id, err)
  596. }
  597. }
  598. }
  599. // TC-0354: 单个id
  600. func TestSysUserRoleModel_BatchDelete_Single(t *testing.T) {
  601. ctx := context.Background()
  602. conn := testutil.GetTestSqlConn()
  603. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  604. userId := randUserRoleId()
  605. roleId := randUserRoleId()
  606. ts := time.Now().Unix()
  607. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  608. if err != nil {
  609. t.Fatalf("Insert: %v", err)
  610. }
  611. id, _ := res.LastInsertId()
  612. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  613. if err := m.BatchDelete(ctx, []int64{id}); err != nil {
  614. t.Fatalf("BatchDelete: %v", err)
  615. }
  616. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  617. t.Fatalf("want ErrNotFound got %v", err)
  618. }
  619. }
  620. // TC-0356: 包含不存在id
  621. func TestSysUserRoleModel_BatchDelete_ContainsNonExist(t *testing.T) {
  622. ctx := context.Background()
  623. conn := testutil.GetTestSqlConn()
  624. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  625. userId := randUserRoleId()
  626. roleId := randUserRoleId()
  627. ts := time.Now().Unix()
  628. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  629. if err != nil {
  630. t.Fatalf("Insert: %v", err)
  631. }
  632. id, _ := res.LastInsertId()
  633. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  634. if err := m.BatchDelete(ctx, []int64{id, 999999999}); err != nil {
  635. t.Fatalf("BatchDelete: %v", err)
  636. }
  637. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  638. t.Fatalf("want ErrNotFound got %v", err)
  639. }
  640. }
  641. // TC-0341: 正常多条
  642. func TestSysUserRoleModel_BatchInsertWithTx_Normal(t *testing.T) {
  643. ctx := context.Background()
  644. conn := testutil.GetTestSqlConn()
  645. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  646. u1 := randUserRoleId()
  647. r1, r2 := randUserRoleId(), randUserRoleId()
  648. ts := time.Now().Unix()
  649. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  650. return m.BatchInsertWithTx(c, session, []*SysUserRole{
  651. {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
  652. {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
  653. })
  654. })
  655. if err != nil {
  656. t.Fatalf("BatchInsertWithTx: %v", err)
  657. }
  658. got1, err := m.FindOneByUserIdRoleId(ctx, u1, r1)
  659. if err != nil {
  660. t.Fatalf("FindOne r1: %v", err)
  661. }
  662. got2, err := m.FindOneByUserIdRoleId(ctx, u1, r2)
  663. if err != nil {
  664. t.Fatalf("FindOne r2: %v", err)
  665. }
  666. defer testutil.CleanTable(ctx, conn, "sys_user_role", got1.Id, got2.Id)
  667. }
  668. // TC-0340: 空列表
  669. func TestSysUserRoleModel_BatchInsertWithTx_Empty(t *testing.T) {
  670. ctx := context.Background()
  671. conn := testutil.GetTestSqlConn()
  672. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  673. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  674. if err := m.BatchInsertWithTx(c, session, nil); err != nil {
  675. return err
  676. }
  677. return m.BatchInsertWithTx(c, session, []*SysUserRole{})
  678. })
  679. if err != nil {
  680. t.Fatalf("BatchInsertWithTx empty: %v", err)
  681. }
  682. }
  683. // TC-0342: 事务回滚
  684. func TestSysUserRoleModel_BatchInsertWithTx_Rollback(t *testing.T) {
  685. ctx := context.Background()
  686. conn := testutil.GetTestSqlConn()
  687. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  688. userId := randUserRoleId()
  689. roleId := randUserRoleId()
  690. ts := time.Now().Unix()
  691. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  692. if err := m.BatchInsertWithTx(c, session, []*SysUserRole{
  693. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  694. }); err != nil {
  695. return err
  696. }
  697. return errors.New("rollback")
  698. })
  699. if err == nil || err.Error() != "rollback" {
  700. t.Fatalf("want rollback error got %v", err)
  701. }
  702. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  703. if err != ErrNotFound {
  704. t.Fatalf("after rollback want ErrNotFound got %v", err)
  705. }
  706. }
  707. // TC-0349: 正常多条
  708. func TestSysUserRoleModel_BatchUpdateWithTx_Normal(t *testing.T) {
  709. ctx := context.Background()
  710. conn := testutil.GetTestSqlConn()
  711. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  712. userId := randUserRoleId()
  713. r1, r2 := randUserRoleId(), randUserRoleId()
  714. ts := time.Now().Unix()
  715. res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  716. if err != nil {
  717. t.Fatalf("Insert1: %v", err)
  718. }
  719. id1, _ := res1.LastInsertId()
  720. res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
  721. if err != nil {
  722. t.Fatalf("Insert2: %v", err)
  723. }
  724. id2, _ := res2.LastInsertId()
  725. defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  726. newTs := ts + 200
  727. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  728. return m.BatchUpdateWithTx(c, session, []*SysUserRole{
  729. {Id: id1, UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: newTs},
  730. {Id: id2, UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: newTs},
  731. })
  732. })
  733. if err != nil {
  734. t.Fatalf("BatchUpdateWithTx: %v", err)
  735. }
  736. got1, err := m.FindOne(ctx, id1)
  737. if err != nil {
  738. t.Fatalf("FindOne1: %v", err)
  739. }
  740. if got1.UpdateTime != newTs {
  741. t.Fatalf("got1 UpdateTime want %d got %d", newTs, got1.UpdateTime)
  742. }
  743. }
  744. // TC-0348: 空列表
  745. func TestSysUserRoleModel_BatchUpdateWithTx_Empty(t *testing.T) {
  746. ctx := context.Background()
  747. conn := testutil.GetTestSqlConn()
  748. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  749. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  750. if err := m.BatchUpdateWithTx(c, session, nil); err != nil {
  751. return err
  752. }
  753. return m.BatchUpdateWithTx(c, session, []*SysUserRole{})
  754. })
  755. if err != nil {
  756. t.Fatalf("BatchUpdateWithTx empty: %v", err)
  757. }
  758. }
  759. // TC-0358: 正常多条
  760. func TestSysUserRoleModel_BatchDeleteWithTx_Normal(t *testing.T) {
  761. ctx := context.Background()
  762. conn := testutil.GetTestSqlConn()
  763. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  764. ts := time.Now().Unix()
  765. var ids []int64
  766. for i := 0; i < 2; i++ {
  767. res, err := m.Insert(ctx, &SysUserRole{
  768. UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
  769. })
  770. if err != nil {
  771. t.Fatalf("Insert: %v", err)
  772. }
  773. id, _ := res.LastInsertId()
  774. ids = append(ids, id)
  775. }
  776. defer func() {
  777. for _, id := range ids {
  778. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  779. }
  780. }()
  781. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  782. return m.BatchDeleteWithTx(c, session, ids)
  783. })
  784. if err != nil {
  785. t.Fatalf("BatchDeleteWithTx: %v", err)
  786. }
  787. for _, id := range ids {
  788. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  789. t.Fatalf("id %d should be deleted: %v", id, err)
  790. }
  791. }
  792. }
  793. // TC-0357: 空ids
  794. func TestSysUserRoleModel_BatchDeleteWithTx_Empty(t *testing.T) {
  795. ctx := context.Background()
  796. conn := testutil.GetTestSqlConn()
  797. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  798. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  799. if err := m.BatchDeleteWithTx(c, session, nil); err != nil {
  800. return err
  801. }
  802. return m.BatchDeleteWithTx(c, session, []int64{})
  803. })
  804. if err != nil {
  805. t.Fatalf("BatchDeleteWithTx empty: %v", err)
  806. }
  807. }
  808. // TC-0323: 事务内可见性
  809. func TestSysUserRoleModel_FindOneWithTx_InsertThenFind(t *testing.T) {
  810. ctx := context.Background()
  811. conn := testutil.GetTestSqlConn()
  812. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  813. userId := randUserRoleId()
  814. roleId := randUserRoleId()
  815. ts := time.Now().Unix()
  816. var insertedId int64
  817. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  818. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  819. if err != nil {
  820. return err
  821. }
  822. insertedId, err = res.LastInsertId()
  823. if err != nil {
  824. return err
  825. }
  826. got, err := m.FindOneWithTx(c, session, insertedId)
  827. require.NoError(t, err)
  828. assert.Equal(t, insertedId, got.Id)
  829. assert.Equal(t, userId, got.UserId)
  830. assert.Equal(t, roleId, got.RoleId)
  831. return nil
  832. })
  833. require.NoError(t, err)
  834. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  835. }
  836. // TC-0322: 事务内记录不存在
  837. func TestSysUserRoleModel_FindOneWithTx_NotFound(t *testing.T) {
  838. ctx := context.Background()
  839. m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  840. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  841. _, err := m.FindOneWithTx(c, session, 999999999999)
  842. require.ErrorIs(t, err, ErrNotFound)
  843. return nil
  844. })
  845. require.NoError(t, err)
  846. }
  847. // TC-0389: FindOneByUserIdRoleIdWithTx
  848. func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_InsertThenFind(t *testing.T) {
  849. ctx := context.Background()
  850. conn := testutil.GetTestSqlConn()
  851. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  852. userId := randUserRoleId()
  853. roleId := randUserRoleId()
  854. ts := time.Now().Unix()
  855. var insertedId int64
  856. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  857. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  858. if err != nil {
  859. return err
  860. }
  861. insertedId, err = res.LastInsertId()
  862. if err != nil {
  863. return err
  864. }
  865. got, err := m.FindOneByUserIdRoleIdWithTx(c, session, userId, roleId)
  866. require.NoError(t, err)
  867. assert.Equal(t, insertedId, got.Id)
  868. assert.Equal(t, userId, got.UserId)
  869. assert.Equal(t, roleId, got.RoleId)
  870. return nil
  871. })
  872. require.NoError(t, err)
  873. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  874. }
  875. // TC-0390: FindOneByUserIdRoleIdWithTx
  876. func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_NotFound(t *testing.T) {
  877. ctx := context.Background()
  878. m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  879. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  880. _, err := m.FindOneByUserIdRoleIdWithTx(c, session, 999999999, 999999999)
  881. require.ErrorIs(t, err, ErrNotFound)
  882. return nil
  883. })
  884. require.NoError(t, err)
  885. }
  886. // TC-0472: FindUserIdsByRoleId 正常返回角色下用户ID列表
  887. func TestSysUserRoleModel_FindUserIdsByRoleId_Normal(t *testing.T) {
  888. ctx := context.Background()
  889. conn := testutil.GetTestSqlConn()
  890. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  891. roleId := randUserRoleId()
  892. u1, u2 := randUserRoleId(), randUserRoleId()
  893. ts := time.Now().Unix()
  894. var ids []int64
  895. for _, userId := range []int64{u1, u2} {
  896. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  897. require.NoError(t, err)
  898. id, _ := res.LastInsertId()
  899. ids = append(ids, id)
  900. }
  901. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", ids...) })
  902. got, err := m.FindUserIdsByRoleId(ctx, roleId)
  903. require.NoError(t, err)
  904. require.Len(t, got, 2)
  905. assert.True(t, int64SliceEqualIgnoreOrder(got, []int64{u1, u2}))
  906. }
  907. // TC-0473: FindUserIdsByRoleId 无绑定返回空
  908. func TestSysUserRoleModel_FindUserIdsByRoleId_Empty(t *testing.T) {
  909. conn := testutil.GetTestSqlConn()
  910. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  911. got, err := m.FindUserIdsByRoleId(context.Background(), 999999998)
  912. require.NoError(t, err)
  913. require.Empty(t, got)
  914. }