sysUserRoleModel_test.go 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303
  1. package userrole
  2. import (
  3. "context"
  4. "errors"
  5. "fmt"
  6. "github.com/go-sql-driver/mysql"
  7. "github.com/stretchr/testify/assert"
  8. "github.com/stretchr/testify/require"
  9. "github.com/zeromicro/go-zero/core/stores/redis"
  10. "github.com/zeromicro/go-zero/core/stores/sqlx"
  11. "math/rand"
  12. "perms-system-server/internal/testutil"
  13. "sort"
  14. "testing"
  15. "time"
  16. )
  17. func randUserRoleId() int64 {
  18. return int64(900000 + rand.Intn(100000))
  19. }
  20. func sortedInt64Copy(a []int64) []int64 {
  21. b := append([]int64(nil), a...)
  22. sort.Slice(b, func(i, j int) bool { return b[i] < b[j] })
  23. return b
  24. }
  25. func int64SliceEqualIgnoreOrder(a, b []int64) bool {
  26. if len(a) != len(b) {
  27. return false
  28. }
  29. aa, bb := sortedInt64Copy(a), sortedInt64Copy(b)
  30. for i := range aa {
  31. if aa[i] != bb[i] {
  32. return false
  33. }
  34. }
  35. return true
  36. }
  37. func insertTestRole(t *testing.T, ctx context.Context, conn sqlx.SqlConn, productCode, name string) int64 {
  38. t.Helper()
  39. ts := time.Now().Unix()
  40. q := "INSERT INTO `sys_role` (`productCode`, `name`, `remark`, `status`, `permsLevel`, `createTime`, `updateTime`) VALUES (?, ?, '', 1, 0, ?, ?)"
  41. res, err := conn.ExecCtx(ctx, q, productCode, name, ts, ts)
  42. if err != nil {
  43. t.Fatalf("insert sys_role: %v", err)
  44. }
  45. id, err := res.LastInsertId()
  46. if err != nil {
  47. t.Fatalf("role LastInsertId: %v", err)
  48. }
  49. return id
  50. }
  51. // TC-0310: 正常插入
  52. func TestSysUserRoleModel_CRUD(t *testing.T) {
  53. ctx := context.Background()
  54. conn := testutil.GetTestSqlConn()
  55. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  56. userId := randUserRoleId()
  57. roleId := randUserRoleId()
  58. ts := time.Now().Unix()
  59. data := &SysUserRole{
  60. UserId: userId,
  61. RoleId: roleId,
  62. CreateTime: ts,
  63. UpdateTime: ts,
  64. }
  65. res, err := m.Insert(ctx, data)
  66. if err != nil {
  67. t.Fatalf("Insert: %v", err)
  68. }
  69. id, err := res.LastInsertId()
  70. if err != nil {
  71. t.Fatalf("LastInsertId: %v", err)
  72. }
  73. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  74. got, err := m.FindOne(ctx, id)
  75. if err != nil {
  76. t.Fatalf("FindOne: %v", err)
  77. }
  78. if got.UserId != userId || got.RoleId != roleId {
  79. t.Fatalf("FindOne mismatch: %+v", got)
  80. }
  81. byPair, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
  82. if err != nil {
  83. t.Fatalf("FindOneByUserIdRoleId: %v", err)
  84. }
  85. if byPair.Id != id {
  86. t.Fatalf("FindOneByUserIdRoleId id want %d got %d", id, byPair.Id)
  87. }
  88. newTs := ts + 1
  89. got.UpdateTime = newTs
  90. if err := m.Update(ctx, got); err != nil {
  91. t.Fatalf("Update: %v", err)
  92. }
  93. updated, err := m.FindOne(ctx, id)
  94. if err != nil {
  95. t.Fatalf("FindOne after update: %v", err)
  96. }
  97. if updated.UpdateTime != newTs {
  98. t.Fatalf("UpdateTime want %d got %d", newTs, updated.UpdateTime)
  99. }
  100. if err := m.Delete(ctx, id); err != nil {
  101. t.Fatalf("Delete: %v", err)
  102. }
  103. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  104. t.Fatalf("after Delete want ErrNotFound got %v", err)
  105. }
  106. }
  107. // TC-0467: 正常查询
  108. func TestSysUserRoleModel_FindRoleIdsByUserId(t *testing.T) {
  109. ctx := context.Background()
  110. conn := testutil.GetTestSqlConn()
  111. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  112. productCode := testutil.UniqueId()
  113. userId := randUserRoleId()
  114. r1 := insertTestRole(t, ctx, conn, productCode, "role1_"+testutil.UniqueId())
  115. r2 := insertTestRole(t, ctx, conn, productCode, "role2_"+testutil.UniqueId())
  116. defer func() {
  117. conn.ExecCtx(ctx, "DELETE FROM `sys_role` WHERE `id` IN (?, ?)", r1, r2)
  118. }()
  119. ts := time.Now().Unix()
  120. var ids []int64
  121. for _, roleId := range []int64{r1, r2} {
  122. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  123. if err != nil {
  124. t.Fatalf("Insert: %v", err)
  125. }
  126. id, _ := res.LastInsertId()
  127. ids = append(ids, id)
  128. }
  129. defer func() {
  130. for _, id := range ids {
  131. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  132. }
  133. }()
  134. got, err := m.FindRoleIdsByUserId(ctx, userId)
  135. if err != nil {
  136. t.Fatalf("FindRoleIdsByUserId: %v", err)
  137. }
  138. if !int64SliceEqualIgnoreOrder(got, []int64{r1, r2}) {
  139. t.Fatalf("got %v want %v", got, []int64{r1, r2})
  140. }
  141. }
  142. // TC-0444: 正常查询
  143. // TC-0469: 正常删除
  144. func TestSysUserRoleModel_DeleteByRoleIdTx(t *testing.T) {
  145. ctx := context.Background()
  146. conn := testutil.GetTestSqlConn()
  147. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  148. userId := randUserRoleId()
  149. roleId := randUserRoleId()
  150. ts := time.Now().Unix()
  151. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  152. if err != nil {
  153. t.Fatalf("Insert: %v", err)
  154. }
  155. id, _ := res.LastInsertId()
  156. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  157. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  158. return m.DeleteByRoleIdTx(c, session, roleId)
  159. })
  160. if err != nil {
  161. t.Fatalf("DeleteByRoleIdTx: %v", err)
  162. }
  163. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  164. t.Fatalf("after tx delete want ErrNotFound got %v", err)
  165. }
  166. }
  167. // TC-0319: 记录不存在
  168. func TestSysUserRoleModel_FindOne_NotFound(t *testing.T) {
  169. conn := testutil.GetTestSqlConn()
  170. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  171. _, err := m.FindOne(context.Background(), 999999999999)
  172. if err != ErrNotFound {
  173. t.Fatalf("want ErrNotFound got %v", err)
  174. }
  175. }
  176. // TC-0388: FindOneByUserIdRoleId
  177. func TestSysUserRoleModel_FindOneByUserIdRoleId_NotFound(t *testing.T) {
  178. conn := testutil.GetTestSqlConn()
  179. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  180. _, err := m.FindOneByUserIdRoleId(context.Background(), 999999999, 999999999)
  181. if err != ErrNotFound {
  182. t.Fatalf("want ErrNotFound got %v", err)
  183. }
  184. }
  185. // TC-0468: 无绑定
  186. func TestSysUserRoleModel_FindRoleIdsByUserId_Empty(t *testing.T) {
  187. conn := testutil.GetTestSqlConn()
  188. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  189. got, err := m.FindRoleIdsByUserId(context.Background(), 999999999)
  190. if err != nil {
  191. t.Fatalf("err: %v", err)
  192. }
  193. if len(got) != 0 {
  194. t.Fatalf("want empty got %v", got)
  195. }
  196. }
  197. // TC-0470: 事务内跨产品删除
  198. func TestSysUserRoleModel_DeleteByUserIdForProductTx(t *testing.T) {
  199. ctx := context.Background()
  200. conn := testutil.GetTestSqlConn()
  201. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  202. pcA := "t_urt_a_" + testutil.UniqueId()
  203. pcB := "t_urt_b_" + testutil.UniqueId()
  204. nameA := "n_ta_" + testutil.UniqueId()
  205. nameB := "n_tb_" + testutil.UniqueId()
  206. roleA := insertTestRole(t, ctx, conn, pcA, nameA)
  207. roleB := insertTestRole(t, ctx, conn, pcB, nameB)
  208. defer testutil.CleanTable(ctx, conn, "sys_role", roleA, roleB)
  209. userId := randUserRoleId()
  210. ts := time.Now().Unix()
  211. resA, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleA, CreateTime: ts, UpdateTime: ts})
  212. if err != nil {
  213. t.Fatalf("Insert A: %v", err)
  214. }
  215. idA, _ := resA.LastInsertId()
  216. resB, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleB, CreateTime: ts, UpdateTime: ts})
  217. if err != nil {
  218. t.Fatalf("Insert B: %v", err)
  219. }
  220. idB, _ := resB.LastInsertId()
  221. defer testutil.CleanTable(ctx, conn, "sys_user_role", idA, idB)
  222. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  223. return m.DeleteByUserIdForProductTx(c, session, userId, pcA)
  224. })
  225. if err != nil {
  226. t.Fatalf("DeleteByUserIdForProductTx: %v", err)
  227. }
  228. if _, err := m.FindOne(ctx, idA); err != ErrNotFound {
  229. t.Fatalf("pcA row should delete: %v", err)
  230. }
  231. if _, err := m.FindOne(ctx, idB); err != nil {
  232. t.Fatalf("pcB row should remain: %v", err)
  233. }
  234. }
  235. // TC-0334: 空列表
  236. func TestSysUserRoleModel_BatchInsert_Empty(t *testing.T) {
  237. conn := testutil.GetTestSqlConn()
  238. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  239. if err := m.BatchInsert(context.Background(), nil); err != nil {
  240. t.Fatalf("nil: %v", err)
  241. }
  242. if err := m.BatchInsert(context.Background(), []*SysUserRole{}); err != nil {
  243. t.Fatalf("empty: %v", err)
  244. }
  245. }
  246. // TC-0353: 空ids
  247. func TestSysUserRoleModel_BatchDelete_Empty(t *testing.T) {
  248. conn := testutil.GetTestSqlConn()
  249. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  250. if err := m.BatchDelete(context.Background(), nil); err != nil {
  251. t.Fatalf("nil: %v", err)
  252. }
  253. if err := m.BatchDelete(context.Background(), []int64{}); err != nil {
  254. t.Fatalf("empty: %v", err)
  255. }
  256. }
  257. // TC-0312: 唯一索引冲突
  258. func TestSysUserRoleModel_Insert_UniqueConflict(t *testing.T) {
  259. ctx := context.Background()
  260. conn := testutil.GetTestSqlConn()
  261. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  262. userId := randUserRoleId()
  263. roleId := randUserRoleId()
  264. ts := time.Now().Unix()
  265. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  266. if err != nil {
  267. t.Fatalf("Insert: %v", err)
  268. }
  269. id, _ := res.LastInsertId()
  270. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  271. _, err = m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  272. if err == nil {
  273. t.Fatal("second Insert want error")
  274. }
  275. var me *mysql.MySQLError
  276. if !errors.As(err, &me) || me.Number != 1062 {
  277. t.Fatalf("want duplicate key 1062, got %v", err)
  278. }
  279. }
  280. // TC-0314: 事务内插入
  281. func TestSysUserRoleModel_InsertWithTx_Normal(t *testing.T) {
  282. ctx := context.Background()
  283. conn := testutil.GetTestSqlConn()
  284. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  285. userId := randUserRoleId()
  286. roleId := randUserRoleId()
  287. ts := time.Now().Unix()
  288. var insertedId int64
  289. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  290. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  291. if err != nil {
  292. return err
  293. }
  294. insertedId, _ = res.LastInsertId()
  295. return nil
  296. })
  297. if err != nil {
  298. t.Fatalf("TransactCtx: %v", err)
  299. }
  300. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  301. got, err := m.FindOne(ctx, insertedId)
  302. if err != nil {
  303. t.Fatalf("FindOne: %v", err)
  304. }
  305. if got.UserId != userId || got.RoleId != roleId {
  306. t.Fatalf("mismatch: %+v", got)
  307. }
  308. }
  309. // TC-0316: 事务回滚后无数据
  310. func TestSysUserRoleModel_InsertWithTx_Rollback(t *testing.T) {
  311. ctx := context.Background()
  312. conn := testutil.GetTestSqlConn()
  313. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  314. userId := randUserRoleId()
  315. roleId := randUserRoleId()
  316. ts := time.Now().Unix()
  317. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  318. _, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  319. if err != nil {
  320. return err
  321. }
  322. return errors.New("rollback")
  323. })
  324. if err == nil || err.Error() != "rollback" {
  325. t.Fatalf("want rollback error got %v", err)
  326. }
  327. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  328. if err != ErrNotFound {
  329. t.Fatalf("after rollback want ErrNotFound got %v", err)
  330. }
  331. }
  332. // TC-0326: 记录不存在
  333. func TestSysUserRoleModel_Update_NotFound(t *testing.T) {
  334. conn := testutil.GetTestSqlConn()
  335. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  336. ts := time.Now().Unix()
  337. err := m.Update(context.Background(), &SysUserRole{
  338. Id: 999999999, UserId: 1, RoleId: 1, CreateTime: ts, UpdateTime: ts,
  339. })
  340. if err != ErrNotFound {
  341. t.Fatalf("want ErrNotFound got %v", err)
  342. }
  343. }
  344. // TC-0327: 事务内更新
  345. func TestSysUserRoleModel_UpdateWithTx(t *testing.T) {
  346. ctx := context.Background()
  347. conn := testutil.GetTestSqlConn()
  348. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  349. userId := randUserRoleId()
  350. roleId := randUserRoleId()
  351. ts := time.Now().Unix()
  352. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  353. if err != nil {
  354. t.Fatalf("Insert: %v", err)
  355. }
  356. id, _ := res.LastInsertId()
  357. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  358. newTs := ts + 100
  359. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  360. return m.UpdateWithTx(c, session, &SysUserRole{
  361. Id: id, UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: newTs,
  362. })
  363. })
  364. if err != nil {
  365. t.Fatalf("UpdateWithTx: %v", err)
  366. }
  367. got, err := m.FindOne(ctx, id)
  368. if err != nil {
  369. t.Fatalf("FindOne: %v", err)
  370. }
  371. if got.UpdateTime != newTs {
  372. t.Fatalf("UpdateTime want %d got %d", newTs, got.UpdateTime)
  373. }
  374. }
  375. // TC-0329: 记录不存在
  376. func TestSysUserRoleModel_Delete_NotFound(t *testing.T) {
  377. conn := testutil.GetTestSqlConn()
  378. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  379. err := m.Delete(context.Background(), 999999999)
  380. if err != ErrNotFound {
  381. t.Fatalf("want ErrNotFound got %v", err)
  382. }
  383. }
  384. // TC-0330: 事务内删除
  385. func TestSysUserRoleModel_DeleteWithTx(t *testing.T) {
  386. ctx := context.Background()
  387. conn := testutil.GetTestSqlConn()
  388. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  389. userId := randUserRoleId()
  390. roleId := randUserRoleId()
  391. ts := time.Now().Unix()
  392. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  393. if err != nil {
  394. t.Fatalf("Insert: %v", err)
  395. }
  396. id, _ := res.LastInsertId()
  397. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  398. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  399. return m.DeleteWithTx(c, session, id)
  400. })
  401. if err != nil {
  402. t.Fatalf("DeleteWithTx: %v", err)
  403. }
  404. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  405. t.Fatalf("after DeleteWithTx want ErrNotFound got %v", err)
  406. }
  407. }
  408. // TC-0332: fn返回错误
  409. func TestSysUserRoleModel_TransactCtx_Rollback(t *testing.T) {
  410. ctx := context.Background()
  411. conn := testutil.GetTestSqlConn()
  412. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  413. userId := randUserRoleId()
  414. roleId := randUserRoleId()
  415. ts := time.Now().Unix()
  416. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  417. if err != nil {
  418. t.Fatalf("Insert: %v", err)
  419. }
  420. id, _ := res.LastInsertId()
  421. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  422. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  423. if err := m.DeleteWithTx(c, session, id); err != nil {
  424. return err
  425. }
  426. return errors.New("rollback")
  427. })
  428. if err == nil || err.Error() != "rollback" {
  429. t.Fatalf("want rollback error got %v", err)
  430. }
  431. got, err := m.FindOne(ctx, id)
  432. if err != nil {
  433. t.Fatalf("after rollback FindOne: %v", err)
  434. }
  435. if got.Id != id {
  436. t.Fatalf("row should still exist")
  437. }
  438. }
  439. // TC-0333: 获取表名
  440. func TestSysUserRoleModel_TableName(t *testing.T) {
  441. conn := testutil.GetTestSqlConn()
  442. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  443. if m.TableName() != "`sys_user_role`" {
  444. t.Fatalf("want `sys_user_role` got %s", m.TableName())
  445. }
  446. }
  447. // TC-0335: 单条记录
  448. func TestSysUserRoleModel_BatchInsert_Single(t *testing.T) {
  449. ctx := context.Background()
  450. conn := testutil.GetTestSqlConn()
  451. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  452. userId := randUserRoleId()
  453. roleId := randUserRoleId()
  454. ts := time.Now().Unix()
  455. if err := m.BatchInsert(ctx, []*SysUserRole{
  456. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  457. }); err != nil {
  458. t.Fatalf("BatchInsert: %v", err)
  459. }
  460. got, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
  461. if err != nil {
  462. t.Fatalf("FindOneByUserIdRoleId: %v", err)
  463. }
  464. defer testutil.CleanTable(ctx, conn, "sys_user_role", got.Id)
  465. if got.UserId != userId || got.RoleId != roleId {
  466. t.Fatalf("mismatch: %+v", got)
  467. }
  468. }
  469. // TC-0336: 多条记录(3条)
  470. func TestSysUserRoleModel_BatchInsert_Multi(t *testing.T) {
  471. ctx := context.Background()
  472. conn := testutil.GetTestSqlConn()
  473. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  474. u1 := randUserRoleId()
  475. r1, r2, r3 := randUserRoleId(), randUserRoleId(), randUserRoleId()
  476. ts := time.Now().Unix()
  477. if err := m.BatchInsert(ctx, []*SysUserRole{
  478. {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
  479. {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
  480. {UserId: u1, RoleId: r3, CreateTime: ts, UpdateTime: ts},
  481. }); err != nil {
  482. t.Fatalf("BatchInsert: %v", err)
  483. }
  484. var ids []int64
  485. for _, roleId := range []int64{r1, r2, r3} {
  486. got, err := m.FindOneByUserIdRoleId(ctx, u1, roleId)
  487. if err != nil {
  488. t.Fatalf("FindOneByUserIdRoleId(%d,%d): %v", u1, roleId, err)
  489. }
  490. ids = append(ids, got.Id)
  491. }
  492. defer func() {
  493. for _, id := range ids {
  494. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  495. }
  496. }()
  497. if len(ids) != 3 {
  498. t.Fatalf("want 3 rows got %d", len(ids))
  499. }
  500. }
  501. // TC-0338: 唯一索引冲突
  502. func TestSysUserRoleModel_BatchInsert_UniqueConflict(t *testing.T) {
  503. ctx := context.Background()
  504. conn := testutil.GetTestSqlConn()
  505. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  506. userId := randUserRoleId()
  507. roleId := randUserRoleId()
  508. ts := time.Now().Unix()
  509. err := m.BatchInsert(ctx, []*SysUserRole{
  510. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  511. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  512. })
  513. if err == nil {
  514. t.Fatal("want error for duplicate")
  515. }
  516. var me *mysql.MySQLError
  517. if !errors.As(err, &me) || me.Number != 1062 {
  518. t.Fatalf("want duplicate key 1062, got %v", err)
  519. }
  520. }
  521. // TC-0343: 空列表
  522. func TestSysUserRoleModel_BatchUpdate_Empty(t *testing.T) {
  523. conn := testutil.GetTestSqlConn()
  524. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  525. if err := m.BatchUpdate(context.Background(), nil); err != nil {
  526. t.Fatalf("nil: %v", err)
  527. }
  528. if err := m.BatchUpdate(context.Background(), []*SysUserRole{}); err != nil {
  529. t.Fatalf("empty: %v", err)
  530. }
  531. }
  532. // TC-0345: 多条记录(3条)
  533. func TestSysUserRoleModel_BatchUpdate_Multi(t *testing.T) {
  534. ctx := context.Background()
  535. conn := testutil.GetTestSqlConn()
  536. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  537. userId := randUserRoleId()
  538. r1, r2 := randUserRoleId(), randUserRoleId()
  539. ts := time.Now().Unix()
  540. res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  541. if err != nil {
  542. t.Fatalf("Insert1: %v", err)
  543. }
  544. id1, _ := res1.LastInsertId()
  545. res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
  546. if err != nil {
  547. t.Fatalf("Insert2: %v", err)
  548. }
  549. id2, _ := res2.LastInsertId()
  550. defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  551. newTs := ts + 100
  552. newR1, newR2 := randUserRoleId(), randUserRoleId()
  553. err = m.BatchUpdate(ctx, []*SysUserRole{
  554. {Id: id1, UserId: userId, RoleId: newR1, CreateTime: ts, UpdateTime: newTs},
  555. {Id: id2, UserId: userId, RoleId: newR2, CreateTime: ts, UpdateTime: newTs},
  556. })
  557. if err != nil {
  558. t.Fatalf("BatchUpdate: %v", err)
  559. }
  560. got1, err := m.FindOne(ctx, id1)
  561. if err != nil {
  562. t.Fatalf("FindOne1: %v", err)
  563. }
  564. if got1.UpdateTime != newTs || got1.RoleId != newR1 {
  565. t.Fatalf("got1 mismatch: %+v", got1)
  566. }
  567. got2, err := m.FindOne(ctx, id2)
  568. if err != nil {
  569. t.Fatalf("FindOne2: %v", err)
  570. }
  571. if got2.UpdateTime != newTs || got2.RoleId != newR2 {
  572. t.Fatalf("got2 mismatch: %+v", got2)
  573. }
  574. }
  575. // TC-0355: 多个id(3个)
  576. func TestSysUserRoleModel_BatchDelete_Multi(t *testing.T) {
  577. ctx := context.Background()
  578. conn := testutil.GetTestSqlConn()
  579. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  580. ts := time.Now().Unix()
  581. var ids []int64
  582. for i := 0; i < 3; i++ {
  583. res, err := m.Insert(ctx, &SysUserRole{
  584. UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
  585. })
  586. if err != nil {
  587. t.Fatalf("Insert: %v", err)
  588. }
  589. id, _ := res.LastInsertId()
  590. ids = append(ids, id)
  591. }
  592. defer func() {
  593. for _, id := range ids {
  594. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  595. }
  596. }()
  597. if err := m.BatchDelete(ctx, ids); err != nil {
  598. t.Fatalf("BatchDelete: %v", err)
  599. }
  600. for _, id := range ids {
  601. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  602. t.Fatalf("id %d should be deleted: %v", id, err)
  603. }
  604. }
  605. }
  606. // TC-0354: 单个id
  607. func TestSysUserRoleModel_BatchDelete_Single(t *testing.T) {
  608. ctx := context.Background()
  609. conn := testutil.GetTestSqlConn()
  610. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  611. userId := randUserRoleId()
  612. roleId := randUserRoleId()
  613. ts := time.Now().Unix()
  614. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  615. if err != nil {
  616. t.Fatalf("Insert: %v", err)
  617. }
  618. id, _ := res.LastInsertId()
  619. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  620. if err := m.BatchDelete(ctx, []int64{id}); err != nil {
  621. t.Fatalf("BatchDelete: %v", err)
  622. }
  623. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  624. t.Fatalf("want ErrNotFound got %v", err)
  625. }
  626. }
  627. // TC-0356: 包含不存在id
  628. func TestSysUserRoleModel_BatchDelete_ContainsNonExist(t *testing.T) {
  629. ctx := context.Background()
  630. conn := testutil.GetTestSqlConn()
  631. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  632. userId := randUserRoleId()
  633. roleId := randUserRoleId()
  634. ts := time.Now().Unix()
  635. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  636. if err != nil {
  637. t.Fatalf("Insert: %v", err)
  638. }
  639. id, _ := res.LastInsertId()
  640. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  641. if err := m.BatchDelete(ctx, []int64{id, 999999999}); err != nil {
  642. t.Fatalf("BatchDelete: %v", err)
  643. }
  644. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  645. t.Fatalf("want ErrNotFound got %v", err)
  646. }
  647. }
  648. // TC-0341: 正常多条
  649. func TestSysUserRoleModel_BatchInsertWithTx_Normal(t *testing.T) {
  650. ctx := context.Background()
  651. conn := testutil.GetTestSqlConn()
  652. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  653. u1 := randUserRoleId()
  654. r1, r2 := randUserRoleId(), randUserRoleId()
  655. ts := time.Now().Unix()
  656. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  657. return m.BatchInsertWithTx(c, session, []*SysUserRole{
  658. {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
  659. {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
  660. })
  661. })
  662. if err != nil {
  663. t.Fatalf("BatchInsertWithTx: %v", err)
  664. }
  665. got1, err := m.FindOneByUserIdRoleId(ctx, u1, r1)
  666. if err != nil {
  667. t.Fatalf("FindOne r1: %v", err)
  668. }
  669. got2, err := m.FindOneByUserIdRoleId(ctx, u1, r2)
  670. if err != nil {
  671. t.Fatalf("FindOne r2: %v", err)
  672. }
  673. defer testutil.CleanTable(ctx, conn, "sys_user_role", got1.Id, got2.Id)
  674. }
  675. // TC-0340: 空列表
  676. func TestSysUserRoleModel_BatchInsertWithTx_Empty(t *testing.T) {
  677. ctx := context.Background()
  678. conn := testutil.GetTestSqlConn()
  679. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  680. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  681. if err := m.BatchInsertWithTx(c, session, nil); err != nil {
  682. return err
  683. }
  684. return m.BatchInsertWithTx(c, session, []*SysUserRole{})
  685. })
  686. if err != nil {
  687. t.Fatalf("BatchInsertWithTx empty: %v", err)
  688. }
  689. }
  690. // TC-0342: 事务回滚
  691. func TestSysUserRoleModel_BatchInsertWithTx_Rollback(t *testing.T) {
  692. ctx := context.Background()
  693. conn := testutil.GetTestSqlConn()
  694. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  695. userId := randUserRoleId()
  696. roleId := randUserRoleId()
  697. ts := time.Now().Unix()
  698. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  699. if err := m.BatchInsertWithTx(c, session, []*SysUserRole{
  700. {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
  701. }); err != nil {
  702. return err
  703. }
  704. return errors.New("rollback")
  705. })
  706. if err == nil || err.Error() != "rollback" {
  707. t.Fatalf("want rollback error got %v", err)
  708. }
  709. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  710. if err != ErrNotFound {
  711. t.Fatalf("after rollback want ErrNotFound got %v", err)
  712. }
  713. }
  714. // TC-0349: 正常多条
  715. func TestSysUserRoleModel_BatchUpdateWithTx_Normal(t *testing.T) {
  716. ctx := context.Background()
  717. conn := testutil.GetTestSqlConn()
  718. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  719. userId := randUserRoleId()
  720. r1, r2 := randUserRoleId(), randUserRoleId()
  721. ts := time.Now().Unix()
  722. res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
  723. if err != nil {
  724. t.Fatalf("Insert1: %v", err)
  725. }
  726. id1, _ := res1.LastInsertId()
  727. res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
  728. if err != nil {
  729. t.Fatalf("Insert2: %v", err)
  730. }
  731. id2, _ := res2.LastInsertId()
  732. defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  733. newTs := ts + 200
  734. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  735. return m.BatchUpdateWithTx(c, session, []*SysUserRole{
  736. {Id: id1, UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: newTs},
  737. {Id: id2, UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: newTs},
  738. })
  739. })
  740. if err != nil {
  741. t.Fatalf("BatchUpdateWithTx: %v", err)
  742. }
  743. got1, err := m.FindOne(ctx, id1)
  744. if err != nil {
  745. t.Fatalf("FindOne1: %v", err)
  746. }
  747. if got1.UpdateTime != newTs {
  748. t.Fatalf("got1 UpdateTime want %d got %d", newTs, got1.UpdateTime)
  749. }
  750. }
  751. // TC-0348: 空列表
  752. func TestSysUserRoleModel_BatchUpdateWithTx_Empty(t *testing.T) {
  753. ctx := context.Background()
  754. conn := testutil.GetTestSqlConn()
  755. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  756. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  757. if err := m.BatchUpdateWithTx(c, session, nil); err != nil {
  758. return err
  759. }
  760. return m.BatchUpdateWithTx(c, session, []*SysUserRole{})
  761. })
  762. if err != nil {
  763. t.Fatalf("BatchUpdateWithTx empty: %v", err)
  764. }
  765. }
  766. // TC-0358: 正常多条
  767. func TestSysUserRoleModel_BatchDeleteWithTx_Normal(t *testing.T) {
  768. ctx := context.Background()
  769. conn := testutil.GetTestSqlConn()
  770. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  771. ts := time.Now().Unix()
  772. var ids []int64
  773. for i := 0; i < 2; i++ {
  774. res, err := m.Insert(ctx, &SysUserRole{
  775. UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
  776. })
  777. if err != nil {
  778. t.Fatalf("Insert: %v", err)
  779. }
  780. id, _ := res.LastInsertId()
  781. ids = append(ids, id)
  782. }
  783. defer func() {
  784. for _, id := range ids {
  785. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  786. }
  787. }()
  788. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  789. return m.BatchDeleteWithTx(c, session, ids)
  790. })
  791. if err != nil {
  792. t.Fatalf("BatchDeleteWithTx: %v", err)
  793. }
  794. for _, id := range ids {
  795. if _, err := m.FindOne(ctx, id); err != ErrNotFound {
  796. t.Fatalf("id %d should be deleted: %v", id, err)
  797. }
  798. }
  799. }
  800. // TC-0357: 空ids
  801. func TestSysUserRoleModel_BatchDeleteWithTx_Empty(t *testing.T) {
  802. ctx := context.Background()
  803. conn := testutil.GetTestSqlConn()
  804. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  805. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  806. if err := m.BatchDeleteWithTx(c, session, nil); err != nil {
  807. return err
  808. }
  809. return m.BatchDeleteWithTx(c, session, []int64{})
  810. })
  811. if err != nil {
  812. t.Fatalf("BatchDeleteWithTx empty: %v", err)
  813. }
  814. }
  815. // TC-0323: 事务内可见性
  816. func TestSysUserRoleModel_FindOneWithTx_InsertThenFind(t *testing.T) {
  817. ctx := context.Background()
  818. conn := testutil.GetTestSqlConn()
  819. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  820. userId := randUserRoleId()
  821. roleId := randUserRoleId()
  822. ts := time.Now().Unix()
  823. var insertedId int64
  824. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  825. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  826. if err != nil {
  827. return err
  828. }
  829. insertedId, err = res.LastInsertId()
  830. if err != nil {
  831. return err
  832. }
  833. got, err := m.FindOneWithTx(c, session, insertedId)
  834. require.NoError(t, err)
  835. assert.Equal(t, insertedId, got.Id)
  836. assert.Equal(t, userId, got.UserId)
  837. assert.Equal(t, roleId, got.RoleId)
  838. return nil
  839. })
  840. require.NoError(t, err)
  841. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  842. }
  843. // TC-0322: 事务内记录不存在
  844. func TestSysUserRoleModel_FindOneWithTx_NotFound(t *testing.T) {
  845. ctx := context.Background()
  846. m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  847. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  848. _, err := m.FindOneWithTx(c, session, 999999999999)
  849. require.ErrorIs(t, err, ErrNotFound)
  850. return nil
  851. })
  852. require.NoError(t, err)
  853. }
  854. // TC-0389: FindOneByUserIdRoleIdWithTx
  855. func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_InsertThenFind(t *testing.T) {
  856. ctx := context.Background()
  857. conn := testutil.GetTestSqlConn()
  858. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  859. userId := randUserRoleId()
  860. roleId := randUserRoleId()
  861. ts := time.Now().Unix()
  862. var insertedId int64
  863. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  864. res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  865. if err != nil {
  866. return err
  867. }
  868. insertedId, err = res.LastInsertId()
  869. if err != nil {
  870. return err
  871. }
  872. got, err := m.FindOneByUserIdRoleIdWithTx(c, session, userId, roleId)
  873. require.NoError(t, err)
  874. assert.Equal(t, insertedId, got.Id)
  875. assert.Equal(t, userId, got.UserId)
  876. assert.Equal(t, roleId, got.RoleId)
  877. return nil
  878. })
  879. require.NoError(t, err)
  880. defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
  881. }
  882. // TC-0390: FindOneByUserIdRoleIdWithTx
  883. func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_NotFound(t *testing.T) {
  884. ctx := context.Background()
  885. m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  886. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  887. _, err := m.FindOneByUserIdRoleIdWithTx(c, session, 999999999, 999999999)
  888. require.ErrorIs(t, err, ErrNotFound)
  889. return nil
  890. })
  891. require.NoError(t, err)
  892. }
  893. // TC-0472: FindUserIdsByRoleId 正常返回角色下用户ID列表
  894. func TestSysUserRoleModel_FindUserIdsByRoleId_Normal(t *testing.T) {
  895. ctx := context.Background()
  896. conn := testutil.GetTestSqlConn()
  897. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  898. roleId := randUserRoleId()
  899. u1, u2 := randUserRoleId(), randUserRoleId()
  900. ts := time.Now().Unix()
  901. var ids []int64
  902. for _, userId := range []int64{u1, u2} {
  903. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  904. require.NoError(t, err)
  905. id, _ := res.LastInsertId()
  906. ids = append(ids, id)
  907. }
  908. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", ids...) })
  909. got, err := m.FindUserIdsByRoleId(ctx, roleId)
  910. require.NoError(t, err)
  911. require.Len(t, got, 2)
  912. assert.True(t, int64SliceEqualIgnoreOrder(got, []int64{u1, u2}))
  913. }
  914. // TC-0473: FindUserIdsByRoleId 无绑定返回空
  915. func TestSysUserRoleModel_FindUserIdsByRoleId_Empty(t *testing.T) {
  916. conn := testutil.GetTestSqlConn()
  917. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  918. got, err := m.FindUserIdsByRoleId(context.Background(), 999999998)
  919. require.NoError(t, err)
  920. require.Empty(t, got)
  921. }
  922. // =============================================================================
  923. // audit 回归:FindRoleIdsByUserIdForProduct 必须过滤 r.status=1(禁用角色不返回)
  924. // =============================================================================
  925. // TC-0706: 同一产品下同时存在启用/禁用两个角色,接口只返回启用的那个
  926. func TestSysUserRoleModel_FindRoleIdsByUserIdForProduct_FiltersDisabledRole(t *testing.T) {
  927. ctx := context.Background()
  928. conn := testutil.GetTestSqlConn()
  929. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  930. userId := randUserRoleId()
  931. productCode := testutil.UniqueId()
  932. ts := time.Now().Unix()
  933. // 启用角色
  934. enabledRoleId := insertTestRole(t, ctx, conn, productCode, "enabled_"+testutil.UniqueId())
  935. // 禁用角色(默认 insertTestRole 状态=1,需要手动改为 2)
  936. disabledRoleId := insertTestRole(t, ctx, conn, productCode, "disabled_"+testutil.UniqueId())
  937. _, err := conn.ExecCtx(ctx, "UPDATE `sys_role` SET `status`=2 WHERE `id`=?", disabledRoleId)
  938. require.NoError(t, err)
  939. var recIds []int64
  940. for _, rid := range []int64{enabledRoleId, disabledRoleId} {
  941. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: rid, CreateTime: ts, UpdateTime: ts})
  942. require.NoError(t, err)
  943. id, _ := res.LastInsertId()
  944. recIds = append(recIds, id)
  945. }
  946. defer func() {
  947. for _, id := range recIds {
  948. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  949. }
  950. testutil.CleanTable(ctx, conn, "sys_role", enabledRoleId, disabledRoleId)
  951. }()
  952. got, err := m.FindRoleIdsByUserIdForProduct(ctx, userId, productCode)
  953. require.NoError(t, err)
  954. // 修复后:仅返回启用角色
  955. assert.ElementsMatch(t, []int64{enabledRoleId}, got,
  956. "FindRoleIdsByUserIdForProduct 必须过滤 r.status=1,禁用角色不应出现在返回中")
  957. assert.NotContains(t, got, disabledRoleId,
  958. "禁用角色的 id 不应被返回")
  959. }
  960. // =============================================================================
  961. // audit 回归:DeleteByUserIdAndRoleIdsTx 批量删除正确性
  962. // =============================================================================
  963. // TC-0707: 批量删除指定 (userId, roleIds) 组合;未命中的绑定不受影响
  964. func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx(t *testing.T) {
  965. ctx := context.Background()
  966. conn := testutil.GetTestSqlConn()
  967. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  968. productCode := testutil.UniqueId()
  969. userId := randUserRoleId()
  970. r1 := insertTestRole(t, ctx, conn, productCode, "r1_"+testutil.UniqueId())
  971. r2 := insertTestRole(t, ctx, conn, productCode, "r2_"+testutil.UniqueId())
  972. r3 := insertTestRole(t, ctx, conn, productCode, "r3_"+testutil.UniqueId())
  973. defer func() {
  974. conn.ExecCtx(ctx, "DELETE FROM `sys_role` WHERE `id` IN (?, ?, ?)", r1, r2, r3)
  975. }()
  976. ts := time.Now().Unix()
  977. var recIds []int64
  978. for _, rid := range []int64{r1, r2, r3} {
  979. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: rid, CreateTime: ts, UpdateTime: ts})
  980. require.NoError(t, err)
  981. id, _ := res.LastInsertId()
  982. recIds = append(recIds, id)
  983. }
  984. defer func() {
  985. for _, id := range recIds {
  986. testutil.CleanTable(ctx, conn, "sys_user_role", id)
  987. }
  988. }()
  989. // 删除 r1 和 r3,保留 r2
  990. err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  991. return m.DeleteByUserIdAndRoleIdsTx(c, session, userId, []int64{r1, r3})
  992. })
  993. require.NoError(t, err)
  994. got, err := m.FindRoleIdsByUserId(ctx, userId)
  995. require.NoError(t, err)
  996. assert.ElementsMatch(t, []int64{r2}, got, "仅保留未被删除的 r2")
  997. }
  998. // TC-0708: DeleteByUserIdAndRoleIdsTx 空列表时应立即返回 nil,不执行 SQL
  999. func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx_EmptyIsNoop(t *testing.T) {
  1000. ctx := context.Background()
  1001. conn := testutil.GetTestSqlConn()
  1002. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  1003. productCode := testutil.UniqueId()
  1004. userId := randUserRoleId()
  1005. roleId := insertTestRole(t, ctx, conn, productCode, "role_"+testutil.UniqueId())
  1006. defer func() {
  1007. conn.ExecCtx(ctx, "DELETE FROM `sys_role` WHERE `id` = ?", roleId)
  1008. }()
  1009. ts := time.Now().Unix()
  1010. res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  1011. require.NoError(t, err)
  1012. id, _ := res.LastInsertId()
  1013. defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
  1014. // 空列表
  1015. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  1016. return m.DeleteByUserIdAndRoleIdsTx(c, session, userId, []int64{})
  1017. })
  1018. require.NoError(t, err)
  1019. // 原有绑定未被删除
  1020. got, err := m.FindRoleIdsByUserId(ctx, userId)
  1021. require.NoError(t, err)
  1022. assert.Contains(t, got, roleId, "空 roleIds 列表不应删除任何数据")
  1023. }
  1024. // TC-0709: DeleteByUserIdAndRoleIdsTx 不影响其它 userId 的同 roleId 绑定(WHERE userId 约束)
  1025. func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx_OtherUserNotAffected(t *testing.T) {
  1026. ctx := context.Background()
  1027. conn := testutil.GetTestSqlConn()
  1028. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  1029. productCode := testutil.UniqueId()
  1030. u1, u2 := randUserRoleId(), randUserRoleId()
  1031. roleId := insertTestRole(t, ctx, conn, productCode, "role_"+testutil.UniqueId())
  1032. defer func() {
  1033. conn.ExecCtx(ctx, "DELETE FROM `sys_role` WHERE `id` = ?", roleId)
  1034. }()
  1035. ts := time.Now().Unix()
  1036. res1, err := m.Insert(ctx, &SysUserRole{UserId: u1, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  1037. require.NoError(t, err)
  1038. id1, _ := res1.LastInsertId()
  1039. res2, err := m.Insert(ctx, &SysUserRole{UserId: u2, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
  1040. require.NoError(t, err)
  1041. id2, _ := res2.LastInsertId()
  1042. defer func() {
  1043. testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
  1044. }()
  1045. err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  1046. return m.DeleteByUserIdAndRoleIdsTx(c, session, u1, []int64{roleId})
  1047. })
  1048. require.NoError(t, err)
  1049. // u1 的被删,u2 的保留
  1050. got1, _ := m.FindRoleIdsByUserId(ctx, u1)
  1051. got2, _ := m.FindRoleIdsByUserId(ctx, u2)
  1052. assert.NotContains(t, got1, roleId)
  1053. assert.Contains(t, got2, roleId)
  1054. }
  1055. func seedUserRoleWithPrimedCache(t *testing.T, ctx context.Context, m SysUserRoleModel, userId, roleId int64) int64 {
  1056. t.Helper()
  1057. ts := time.Now().Unix()
  1058. res, err := m.Insert(ctx, &SysUserRole{
  1059. UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts,
  1060. })
  1061. require.NoError(t, err)
  1062. id, err := res.LastInsertId()
  1063. require.NoError(t, err)
  1064. _, err = m.FindOne(ctx, id)
  1065. require.NoError(t, err, "FindOne 为 id 维度缓存预热")
  1066. _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
  1067. require.NoError(t, err, "FindOneByUserIdRoleId 为复合维度缓存预热")
  1068. return id
  1069. }
  1070. func assertCacheKeysGone(t *testing.T, rds *redis.Redis, idKey, compositeKey string) {
  1071. t.Helper()
  1072. got, err := rds.Get(idKey)
  1073. require.NoError(t, err)
  1074. assert.Empty(t, got, "id 维度缓存 key %q 应被 DELETE 一并失效", idKey)
  1075. got, err = rds.Get(compositeKey)
  1076. require.NoError(t, err)
  1077. assert.Empty(t, got, "复合维度缓存 key %q 应被 DELETE 一并失效", compositeKey)
  1078. }
  1079. // TC-1062: DeleteByRoleIdTx 对多行同时失效 id + composite 缓存
  1080. func TestSysUserRoleModel_DeleteByRoleIdTx_InvalidatesAllKeyCols(t *testing.T) {
  1081. ctx := context.Background()
  1082. conn := testutil.GetTestSqlConn()
  1083. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  1084. rds := redis.MustNewRedis(testutil.GetTestConfig().CacheRedis.Nodes[0].RedisConf)
  1085. cachePrefix := testutil.GetTestCachePrefix()
  1086. roleId := randUserRoleId()
  1087. u1, u2 := randUserRoleId(), randUserRoleId()
  1088. if u1 == u2 {
  1089. u2 = u1 + 1
  1090. }
  1091. id1 := seedUserRoleWithPrimedCache(t, ctx, m, u1, roleId)
  1092. id2 := seedUserRoleWithPrimedCache(t, ctx, m, u2, roleId)
  1093. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2) })
  1094. require.NoError(t,
  1095. m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  1096. return m.DeleteByRoleIdTx(c, session, roleId)
  1097. }))
  1098. // 裁剪 SELECT 为三列后,构造缓存 key 需要的 id/userId/roleId 仍被携带回来。
  1099. // 任何一条遗漏都会让这里的断言炸掉(旧行的缓存还在,FindOne 就能从缓存拿到"幽灵行")。
  1100. for _, c := range []struct {
  1101. id, uid int64
  1102. }{{id1, u1}, {id2, u2}} {
  1103. idKey := fmt.Sprintf("%s:cache:sysUserRole:id:%v", cachePrefix, c.id)
  1104. compKey := fmt.Sprintf("%s:cache:sysUserRole:userId:roleId:%v:%v", cachePrefix, c.uid, roleId)
  1105. assertCacheKeysGone(t, rds, idKey, compKey)
  1106. }
  1107. // 终态真相:两行都不应再存在于 DB。
  1108. for _, id := range []int64{id1, id2} {
  1109. _, err := m.FindOne(ctx, id)
  1110. assert.ErrorIs(t, err, ErrNotFound,
  1111. "DELETE 已提交,FindOne 必须回落 DB 读到 ErrNotFound;"+
  1112. "若仍查到旧行说明裁剪后的 SELECT 漏掉了 id 列,id 维度缓存仍在")
  1113. }
  1114. }
  1115. // TC-1063: DeleteByUserIdAndRoleIdsTx 的批量 IN 路径
  1116. func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx_InvalidatesAllKeyCols(t *testing.T) {
  1117. ctx := context.Background()
  1118. conn := testutil.GetTestSqlConn()
  1119. m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
  1120. rds := redis.MustNewRedis(testutil.GetTestConfig().CacheRedis.Nodes[0].RedisConf)
  1121. cachePrefix := testutil.GetTestCachePrefix()
  1122. userId := randUserRoleId()
  1123. r1, r2, r3 := randUserRoleId(), randUserRoleId()+1, randUserRoleId()+2
  1124. id1 := seedUserRoleWithPrimedCache(t, ctx, m, userId, r1)
  1125. id2 := seedUserRoleWithPrimedCache(t, ctx, m, userId, r2)
  1126. // r3 作为"不在删除集合内"的对照组:这一行不得被误伤
  1127. id3 := seedUserRoleWithPrimedCache(t, ctx, m, userId, r3)
  1128. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2, id3) })
  1129. require.NoError(t,
  1130. m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
  1131. return m.DeleteByUserIdAndRoleIdsTx(c, session, userId, []int64{r1, r2})
  1132. }))
  1133. // 被删的两条:id+composite 都必须失效。
  1134. for _, c := range []struct {
  1135. id, rid int64
  1136. }{{id1, r1}, {id2, r2}} {
  1137. idKey := fmt.Sprintf("%s:cache:sysUserRole:id:%v", cachePrefix, c.id)
  1138. compKey := fmt.Sprintf("%s:cache:sysUserRole:userId:roleId:%v:%v", cachePrefix, userId, c.rid)
  1139. assertCacheKeysGone(t, rds, idKey, compKey)
  1140. _, err := m.FindOne(ctx, c.id)
  1141. assert.ErrorIs(t, err, ErrNotFound)
  1142. }
  1143. // 未被删的第 3 条:DB 仍在,FindOne 必须成功。
  1144. got, err := m.FindOne(ctx, id3)
  1145. require.NoError(t, err,
  1146. "防误伤:IN (r1, r2) 不得把 r3 的行带走;"+
  1147. "若失败,说明裁剪后 SELECT 把对照组的 key 也返回并被 ExecCtx 一并失效")
  1148. assert.Equal(t, id3, got.Id)
  1149. assert.Equal(t, r3, got.RoleId)
  1150. }