| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284 |
- package userrole
- import (
- "context"
- "errors"
- "fmt"
- "github.com/go-sql-driver/mysql"
- "github.com/stretchr/testify/assert"
- "github.com/stretchr/testify/require"
- "github.com/zeromicro/go-zero/core/stores/redis"
- "github.com/zeromicro/go-zero/core/stores/sqlx"
- "math/rand"
- "perms-system-server/internal/testutil"
- "sort"
- "testing"
- "time"
- )
- func randUserRoleId() int64 {
- return int64(900000 + rand.Intn(100000))
- }
- func sortedInt64Copy(a []int64) []int64 {
- b := append([]int64(nil), a...)
- sort.Slice(b, func(i, j int) bool { return b[i] < b[j] })
- return b
- }
- func int64SliceEqualIgnoreOrder(a, b []int64) bool {
- if len(a) != len(b) {
- return false
- }
- aa, bb := sortedInt64Copy(a), sortedInt64Copy(b)
- for i := range aa {
- if aa[i] != bb[i] {
- return false
- }
- }
- return true
- }
- func insertTestRole(t *testing.T, ctx context.Context, conn sqlx.SqlConn, productCode, name string) int64 {
- t.Helper()
- ts := time.Now().Unix()
- q := "INSERT INTO `sys_role` (`productCode`, `name`, `remark`, `status`, `permsLevel`, `createTime`, `updateTime`) VALUES (?, ?, '', 1, 0, ?, ?)"
- res, err := conn.ExecCtx(ctx, q, productCode, name, ts, ts)
- if err != nil {
- t.Fatalf("insert sys_role: %v", err)
- }
- id, err := res.LastInsertId()
- if err != nil {
- t.Fatalf("role LastInsertId: %v", err)
- }
- return id
- }
- // TC-0310: 正常插入
- func TestSysUserRoleModel_CRUD(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- data := &SysUserRole{
- UserId: userId,
- RoleId: roleId,
- CreateTime: ts,
- UpdateTime: ts,
- }
- res, err := m.Insert(ctx, data)
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, err := res.LastInsertId()
- if err != nil {
- t.Fatalf("LastInsertId: %v", err)
- }
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- got, err := m.FindOne(ctx, id)
- if err != nil {
- t.Fatalf("FindOne: %v", err)
- }
- if got.UserId != userId || got.RoleId != roleId {
- t.Fatalf("FindOne mismatch: %+v", got)
- }
- byPair, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
- if err != nil {
- t.Fatalf("FindOneByUserIdRoleId: %v", err)
- }
- if byPair.Id != id {
- t.Fatalf("FindOneByUserIdRoleId id want %d got %d", id, byPair.Id)
- }
- newTs := ts + 1
- got.UpdateTime = newTs
- if err := m.Update(ctx, got); err != nil {
- t.Fatalf("Update: %v", err)
- }
- updated, err := m.FindOne(ctx, id)
- if err != nil {
- t.Fatalf("FindOne after update: %v", err)
- }
- if updated.UpdateTime != newTs {
- t.Fatalf("UpdateTime want %d got %d", newTs, updated.UpdateTime)
- }
- if err := m.Delete(ctx, id); err != nil {
- t.Fatalf("Delete: %v", err)
- }
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("after Delete want ErrNotFound got %v", err)
- }
- }
- // TC-0467: 正常查询
- func TestSysUserRoleModel_FindRoleIdsByUserId(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- r1, r2 := randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- var ids []int64
- for _, roleId := range []int64{r1, r2} {
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- ids = append(ids, id)
- }
- defer func() {
- for _, id := range ids {
- testutil.CleanTable(ctx, conn, "sys_user_role", id)
- }
- }()
- got, err := m.FindRoleIdsByUserId(ctx, userId)
- if err != nil {
- t.Fatalf("FindRoleIdsByUserId: %v", err)
- }
- if !int64SliceEqualIgnoreOrder(got, []int64{r1, r2}) {
- t.Fatalf("got %v want %v", got, []int64{r1, r2})
- }
- }
- // TC-0444: 正常查询
- // TC-0469: 正常删除
- func TestSysUserRoleModel_DeleteByRoleIdTx(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByRoleIdTx(c, session, roleId)
- })
- if err != nil {
- t.Fatalf("DeleteByRoleIdTx: %v", err)
- }
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("after tx delete want ErrNotFound got %v", err)
- }
- }
- // TC-0319: 记录不存在
- func TestSysUserRoleModel_FindOne_NotFound(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- _, err := m.FindOne(context.Background(), 999999999999)
- if err != ErrNotFound {
- t.Fatalf("want ErrNotFound got %v", err)
- }
- }
- // TC-0388: FindOneByUserIdRoleId
- func TestSysUserRoleModel_FindOneByUserIdRoleId_NotFound(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- _, err := m.FindOneByUserIdRoleId(context.Background(), 999999999, 999999999)
- if err != ErrNotFound {
- t.Fatalf("want ErrNotFound got %v", err)
- }
- }
- // TC-0468: 无绑定
- func TestSysUserRoleModel_FindRoleIdsByUserId_Empty(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- got, err := m.FindRoleIdsByUserId(context.Background(), 999999999)
- if err != nil {
- t.Fatalf("err: %v", err)
- }
- if len(got) != 0 {
- t.Fatalf("want empty got %v", got)
- }
- }
- // TC-0470: 事务内跨产品删除
- func TestSysUserRoleModel_DeleteByUserIdForProductTx(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- pcA := "t_urt_a_" + testutil.UniqueId()
- pcB := "t_urt_b_" + testutil.UniqueId()
- nameA := "n_ta_" + testutil.UniqueId()
- nameB := "n_tb_" + testutil.UniqueId()
- roleA := insertTestRole(t, ctx, conn, pcA, nameA)
- roleB := insertTestRole(t, ctx, conn, pcB, nameB)
- defer testutil.CleanTable(ctx, conn, "sys_role", roleA, roleB)
- userId := randUserRoleId()
- ts := time.Now().Unix()
- resA, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleA, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert A: %v", err)
- }
- idA, _ := resA.LastInsertId()
- resB, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleB, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert B: %v", err)
- }
- idB, _ := resB.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", idA, idB)
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByUserIdForProductTx(c, session, userId, pcA)
- })
- if err != nil {
- t.Fatalf("DeleteByUserIdForProductTx: %v", err)
- }
- if _, err := m.FindOne(ctx, idA); err != ErrNotFound {
- t.Fatalf("pcA row should delete: %v", err)
- }
- if _, err := m.FindOne(ctx, idB); err != nil {
- t.Fatalf("pcB row should remain: %v", err)
- }
- }
- // TC-0334: 空列表
- func TestSysUserRoleModel_BatchInsert_Empty(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- if err := m.BatchInsert(context.Background(), nil); err != nil {
- t.Fatalf("nil: %v", err)
- }
- if err := m.BatchInsert(context.Background(), []*SysUserRole{}); err != nil {
- t.Fatalf("empty: %v", err)
- }
- }
- // TC-0353: 空ids
- func TestSysUserRoleModel_BatchDelete_Empty(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- if err := m.BatchDelete(context.Background(), nil); err != nil {
- t.Fatalf("nil: %v", err)
- }
- if err := m.BatchDelete(context.Background(), []int64{}); err != nil {
- t.Fatalf("empty: %v", err)
- }
- }
- // TC-0312: 唯一索引冲突
- func TestSysUserRoleModel_Insert_UniqueConflict(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- _, err = m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err == nil {
- t.Fatal("second Insert want error")
- }
- var me *mysql.MySQLError
- if !errors.As(err, &me) || me.Number != 1062 {
- t.Fatalf("want duplicate key 1062, got %v", err)
- }
- }
- // TC-0314: 事务内插入
- func TestSysUserRoleModel_InsertWithTx_Normal(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- var insertedId int64
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- return err
- }
- insertedId, _ = res.LastInsertId()
- return nil
- })
- if err != nil {
- t.Fatalf("TransactCtx: %v", err)
- }
- defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
- got, err := m.FindOne(ctx, insertedId)
- if err != nil {
- t.Fatalf("FindOne: %v", err)
- }
- if got.UserId != userId || got.RoleId != roleId {
- t.Fatalf("mismatch: %+v", got)
- }
- }
- // TC-0316: 事务回滚后无数据
- func TestSysUserRoleModel_InsertWithTx_Rollback(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- _, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- return err
- }
- return errors.New("rollback")
- })
- if err == nil || err.Error() != "rollback" {
- t.Fatalf("want rollback error got %v", err)
- }
- _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
- if err != ErrNotFound {
- t.Fatalf("after rollback want ErrNotFound got %v", err)
- }
- }
- // TC-0326: 记录不存在
- func TestSysUserRoleModel_Update_NotFound(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- ts := time.Now().Unix()
- err := m.Update(context.Background(), &SysUserRole{
- Id: 999999999, UserId: 1, RoleId: 1, CreateTime: ts, UpdateTime: ts,
- })
- if err != ErrNotFound {
- t.Fatalf("want ErrNotFound got %v", err)
- }
- }
- // TC-0327: 事务内更新
- func TestSysUserRoleModel_UpdateWithTx(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- newTs := ts + 100
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.UpdateWithTx(c, session, &SysUserRole{
- Id: id, UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: newTs,
- })
- })
- if err != nil {
- t.Fatalf("UpdateWithTx: %v", err)
- }
- got, err := m.FindOne(ctx, id)
- if err != nil {
- t.Fatalf("FindOne: %v", err)
- }
- if got.UpdateTime != newTs {
- t.Fatalf("UpdateTime want %d got %d", newTs, got.UpdateTime)
- }
- }
- // TC-0329: 记录不存在
- func TestSysUserRoleModel_Delete_NotFound(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- err := m.Delete(context.Background(), 999999999)
- if err != ErrNotFound {
- t.Fatalf("want ErrNotFound got %v", err)
- }
- }
- // TC-0330: 事务内删除
- func TestSysUserRoleModel_DeleteWithTx(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteWithTx(c, session, id)
- })
- if err != nil {
- t.Fatalf("DeleteWithTx: %v", err)
- }
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("after DeleteWithTx want ErrNotFound got %v", err)
- }
- }
- // TC-0332: fn返回错误
- func TestSysUserRoleModel_TransactCtx_Rollback(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- if err := m.DeleteWithTx(c, session, id); err != nil {
- return err
- }
- return errors.New("rollback")
- })
- if err == nil || err.Error() != "rollback" {
- t.Fatalf("want rollback error got %v", err)
- }
- got, err := m.FindOne(ctx, id)
- if err != nil {
- t.Fatalf("after rollback FindOne: %v", err)
- }
- if got.Id != id {
- t.Fatalf("row should still exist")
- }
- }
- // TC-0333: 获取表名
- func TestSysUserRoleModel_TableName(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- if m.TableName() != "`sys_user_role`" {
- t.Fatalf("want `sys_user_role` got %s", m.TableName())
- }
- }
- // TC-0335: 单条记录
- func TestSysUserRoleModel_BatchInsert_Single(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- if err := m.BatchInsert(ctx, []*SysUserRole{
- {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
- }); err != nil {
- t.Fatalf("BatchInsert: %v", err)
- }
- got, err := m.FindOneByUserIdRoleId(ctx, userId, roleId)
- if err != nil {
- t.Fatalf("FindOneByUserIdRoleId: %v", err)
- }
- defer testutil.CleanTable(ctx, conn, "sys_user_role", got.Id)
- if got.UserId != userId || got.RoleId != roleId {
- t.Fatalf("mismatch: %+v", got)
- }
- }
- // TC-0336: 多条记录(3条)
- func TestSysUserRoleModel_BatchInsert_Multi(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- u1 := randUserRoleId()
- r1, r2, r3 := randUserRoleId(), randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- if err := m.BatchInsert(ctx, []*SysUserRole{
- {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
- {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
- {UserId: u1, RoleId: r3, CreateTime: ts, UpdateTime: ts},
- }); err != nil {
- t.Fatalf("BatchInsert: %v", err)
- }
- var ids []int64
- for _, roleId := range []int64{r1, r2, r3} {
- got, err := m.FindOneByUserIdRoleId(ctx, u1, roleId)
- if err != nil {
- t.Fatalf("FindOneByUserIdRoleId(%d,%d): %v", u1, roleId, err)
- }
- ids = append(ids, got.Id)
- }
- defer func() {
- for _, id := range ids {
- testutil.CleanTable(ctx, conn, "sys_user_role", id)
- }
- }()
- if len(ids) != 3 {
- t.Fatalf("want 3 rows got %d", len(ids))
- }
- }
- // TC-0338: 唯一索引冲突
- func TestSysUserRoleModel_BatchInsert_UniqueConflict(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- err := m.BatchInsert(ctx, []*SysUserRole{
- {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
- {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
- })
- if err == nil {
- t.Fatal("want error for duplicate")
- }
- var me *mysql.MySQLError
- if !errors.As(err, &me) || me.Number != 1062 {
- t.Fatalf("want duplicate key 1062, got %v", err)
- }
- }
- // TC-0343: 空列表
- func TestSysUserRoleModel_BatchUpdate_Empty(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- if err := m.BatchUpdate(context.Background(), nil); err != nil {
- t.Fatalf("nil: %v", err)
- }
- if err := m.BatchUpdate(context.Background(), []*SysUserRole{}); err != nil {
- t.Fatalf("empty: %v", err)
- }
- }
- // TC-0345: 多条记录(3条)
- func TestSysUserRoleModel_BatchUpdate_Multi(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- r1, r2 := randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert1: %v", err)
- }
- id1, _ := res1.LastInsertId()
- res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert2: %v", err)
- }
- id2, _ := res2.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
- newTs := ts + 100
- newR1, newR2 := randUserRoleId(), randUserRoleId()
- err = m.BatchUpdate(ctx, []*SysUserRole{
- {Id: id1, UserId: userId, RoleId: newR1, CreateTime: ts, UpdateTime: newTs},
- {Id: id2, UserId: userId, RoleId: newR2, CreateTime: ts, UpdateTime: newTs},
- })
- if err != nil {
- t.Fatalf("BatchUpdate: %v", err)
- }
- got1, err := m.FindOne(ctx, id1)
- if err != nil {
- t.Fatalf("FindOne1: %v", err)
- }
- if got1.UpdateTime != newTs || got1.RoleId != newR1 {
- t.Fatalf("got1 mismatch: %+v", got1)
- }
- got2, err := m.FindOne(ctx, id2)
- if err != nil {
- t.Fatalf("FindOne2: %v", err)
- }
- if got2.UpdateTime != newTs || got2.RoleId != newR2 {
- t.Fatalf("got2 mismatch: %+v", got2)
- }
- }
- // TC-0355: 多个id(3个)
- func TestSysUserRoleModel_BatchDelete_Multi(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- ts := time.Now().Unix()
- var ids []int64
- for i := 0; i < 3; i++ {
- res, err := m.Insert(ctx, &SysUserRole{
- UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
- })
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- ids = append(ids, id)
- }
- defer func() {
- for _, id := range ids {
- testutil.CleanTable(ctx, conn, "sys_user_role", id)
- }
- }()
- if err := m.BatchDelete(ctx, ids); err != nil {
- t.Fatalf("BatchDelete: %v", err)
- }
- for _, id := range ids {
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("id %d should be deleted: %v", id, err)
- }
- }
- }
- // TC-0354: 单个id
- func TestSysUserRoleModel_BatchDelete_Single(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- if err := m.BatchDelete(ctx, []int64{id}); err != nil {
- t.Fatalf("BatchDelete: %v", err)
- }
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("want ErrNotFound got %v", err)
- }
- }
- // TC-0356: 包含不存在id
- func TestSysUserRoleModel_BatchDelete_ContainsNonExist(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- if err := m.BatchDelete(ctx, []int64{id, 999999999}); err != nil {
- t.Fatalf("BatchDelete: %v", err)
- }
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("want ErrNotFound got %v", err)
- }
- }
- // TC-0341: 正常多条
- func TestSysUserRoleModel_BatchInsertWithTx_Normal(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- u1 := randUserRoleId()
- r1, r2 := randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.BatchInsertWithTx(c, session, []*SysUserRole{
- {UserId: u1, RoleId: r1, CreateTime: ts, UpdateTime: ts},
- {UserId: u1, RoleId: r2, CreateTime: ts, UpdateTime: ts},
- })
- })
- if err != nil {
- t.Fatalf("BatchInsertWithTx: %v", err)
- }
- got1, err := m.FindOneByUserIdRoleId(ctx, u1, r1)
- if err != nil {
- t.Fatalf("FindOne r1: %v", err)
- }
- got2, err := m.FindOneByUserIdRoleId(ctx, u1, r2)
- if err != nil {
- t.Fatalf("FindOne r2: %v", err)
- }
- defer testutil.CleanTable(ctx, conn, "sys_user_role", got1.Id, got2.Id)
- }
- // TC-0340: 空列表
- func TestSysUserRoleModel_BatchInsertWithTx_Empty(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- if err := m.BatchInsertWithTx(c, session, nil); err != nil {
- return err
- }
- return m.BatchInsertWithTx(c, session, []*SysUserRole{})
- })
- if err != nil {
- t.Fatalf("BatchInsertWithTx empty: %v", err)
- }
- }
- // TC-0342: 事务回滚
- func TestSysUserRoleModel_BatchInsertWithTx_Rollback(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- if err := m.BatchInsertWithTx(c, session, []*SysUserRole{
- {UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts},
- }); err != nil {
- return err
- }
- return errors.New("rollback")
- })
- if err == nil || err.Error() != "rollback" {
- t.Fatalf("want rollback error got %v", err)
- }
- _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
- if err != ErrNotFound {
- t.Fatalf("after rollback want ErrNotFound got %v", err)
- }
- }
- // TC-0349: 正常多条
- func TestSysUserRoleModel_BatchUpdateWithTx_Normal(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- r1, r2 := randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- res1, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert1: %v", err)
- }
- id1, _ := res1.LastInsertId()
- res2, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- t.Fatalf("Insert2: %v", err)
- }
- id2, _ := res2.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
- newTs := ts + 200
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.BatchUpdateWithTx(c, session, []*SysUserRole{
- {Id: id1, UserId: userId, RoleId: r1, CreateTime: ts, UpdateTime: newTs},
- {Id: id2, UserId: userId, RoleId: r2, CreateTime: ts, UpdateTime: newTs},
- })
- })
- if err != nil {
- t.Fatalf("BatchUpdateWithTx: %v", err)
- }
- got1, err := m.FindOne(ctx, id1)
- if err != nil {
- t.Fatalf("FindOne1: %v", err)
- }
- if got1.UpdateTime != newTs {
- t.Fatalf("got1 UpdateTime want %d got %d", newTs, got1.UpdateTime)
- }
- }
- // TC-0348: 空列表
- func TestSysUserRoleModel_BatchUpdateWithTx_Empty(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- if err := m.BatchUpdateWithTx(c, session, nil); err != nil {
- return err
- }
- return m.BatchUpdateWithTx(c, session, []*SysUserRole{})
- })
- if err != nil {
- t.Fatalf("BatchUpdateWithTx empty: %v", err)
- }
- }
- // TC-0358: 正常多条
- func TestSysUserRoleModel_BatchDeleteWithTx_Normal(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- ts := time.Now().Unix()
- var ids []int64
- for i := 0; i < 2; i++ {
- res, err := m.Insert(ctx, &SysUserRole{
- UserId: randUserRoleId(), RoleId: randUserRoleId(), CreateTime: ts, UpdateTime: ts,
- })
- if err != nil {
- t.Fatalf("Insert: %v", err)
- }
- id, _ := res.LastInsertId()
- ids = append(ids, id)
- }
- defer func() {
- for _, id := range ids {
- testutil.CleanTable(ctx, conn, "sys_user_role", id)
- }
- }()
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.BatchDeleteWithTx(c, session, ids)
- })
- if err != nil {
- t.Fatalf("BatchDeleteWithTx: %v", err)
- }
- for _, id := range ids {
- if _, err := m.FindOne(ctx, id); err != ErrNotFound {
- t.Fatalf("id %d should be deleted: %v", id, err)
- }
- }
- }
- // TC-0357: 空ids
- func TestSysUserRoleModel_BatchDeleteWithTx_Empty(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- if err := m.BatchDeleteWithTx(c, session, nil); err != nil {
- return err
- }
- return m.BatchDeleteWithTx(c, session, []int64{})
- })
- if err != nil {
- t.Fatalf("BatchDeleteWithTx empty: %v", err)
- }
- }
- // TC-0323: 事务内可见性
- func TestSysUserRoleModel_FindOneWithTx_InsertThenFind(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- var insertedId int64
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- return err
- }
- insertedId, err = res.LastInsertId()
- if err != nil {
- return err
- }
- got, err := m.FindOneWithTx(c, session, insertedId)
- require.NoError(t, err)
- assert.Equal(t, insertedId, got.Id)
- assert.Equal(t, userId, got.UserId)
- assert.Equal(t, roleId, got.RoleId)
- return nil
- })
- require.NoError(t, err)
- defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
- }
- // TC-0322: 事务内记录不存在
- func TestSysUserRoleModel_FindOneWithTx_NotFound(t *testing.T) {
- ctx := context.Background()
- m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- _, err := m.FindOneWithTx(c, session, 999999999999)
- require.ErrorIs(t, err, ErrNotFound)
- return nil
- })
- require.NoError(t, err)
- }
- // TC-0389: FindOneByUserIdRoleIdWithTx
- func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_InsertThenFind(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- var insertedId int64
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- res, err := m.InsertWithTx(c, session, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- if err != nil {
- return err
- }
- insertedId, err = res.LastInsertId()
- if err != nil {
- return err
- }
- got, err := m.FindOneByUserIdRoleIdWithTx(c, session, userId, roleId)
- require.NoError(t, err)
- assert.Equal(t, insertedId, got.Id)
- assert.Equal(t, userId, got.UserId)
- assert.Equal(t, roleId, got.RoleId)
- return nil
- })
- require.NoError(t, err)
- defer testutil.CleanTable(ctx, conn, "sys_user_role", insertedId)
- }
- // TC-0390: FindOneByUserIdRoleIdWithTx
- func TestSysUserRoleModel_FindOneByUserIdRoleIdWithTx_NotFound(t *testing.T) {
- ctx := context.Background()
- m := NewSysUserRoleModel(testutil.GetTestSqlConn(), testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- _, err := m.FindOneByUserIdRoleIdWithTx(c, session, 999999999, 999999999)
- require.ErrorIs(t, err, ErrNotFound)
- return nil
- })
- require.NoError(t, err)
- }
- // TC-0472: FindUserIdsByRoleId 正常返回角色下用户ID列表
- func TestSysUserRoleModel_FindUserIdsByRoleId_Normal(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- roleId := randUserRoleId()
- u1, u2 := randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- var ids []int64
- for _, userId := range []int64{u1, u2} {
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- require.NoError(t, err)
- id, _ := res.LastInsertId()
- ids = append(ids, id)
- }
- t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", ids...) })
- got, err := m.FindUserIdsByRoleId(ctx, roleId)
- require.NoError(t, err)
- require.Len(t, got, 2)
- assert.True(t, int64SliceEqualIgnoreOrder(got, []int64{u1, u2}))
- }
- // TC-0473: FindUserIdsByRoleId 无绑定返回空
- func TestSysUserRoleModel_FindUserIdsByRoleId_Empty(t *testing.T) {
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- got, err := m.FindUserIdsByRoleId(context.Background(), 999999998)
- require.NoError(t, err)
- require.Empty(t, got)
- }
- // =============================================================================
- // audit 回归:FindRoleIdsByUserIdForProduct 必须过滤 r.status=1(禁用角色不返回)
- // =============================================================================
- // TC-0706: 同一产品下同时存在启用/禁用两个角色,接口只返回启用的那个
- func TestSysUserRoleModel_FindRoleIdsByUserIdForProduct_FiltersDisabledRole(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- productCode := testutil.UniqueId()
- ts := time.Now().Unix()
- // 启用角色
- enabledRoleId := insertTestRole(t, ctx, conn, productCode, "enabled_"+testutil.UniqueId())
- // 禁用角色(默认 insertTestRole 状态=1,需要手动改为 2)
- disabledRoleId := insertTestRole(t, ctx, conn, productCode, "disabled_"+testutil.UniqueId())
- _, err := conn.ExecCtx(ctx, "UPDATE `sys_role` SET `status`=2 WHERE `id`=?", disabledRoleId)
- require.NoError(t, err)
- var recIds []int64
- for _, rid := range []int64{enabledRoleId, disabledRoleId} {
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: rid, CreateTime: ts, UpdateTime: ts})
- require.NoError(t, err)
- id, _ := res.LastInsertId()
- recIds = append(recIds, id)
- }
- defer func() {
- for _, id := range recIds {
- testutil.CleanTable(ctx, conn, "sys_user_role", id)
- }
- testutil.CleanTable(ctx, conn, "sys_role", enabledRoleId, disabledRoleId)
- }()
- got, err := m.FindRoleIdsByUserIdForProduct(ctx, userId, productCode)
- require.NoError(t, err)
- // 修复后:仅返回启用角色
- assert.ElementsMatch(t, []int64{enabledRoleId}, got,
- "FindRoleIdsByUserIdForProduct 必须过滤 r.status=1,禁用角色不应出现在返回中")
- assert.NotContains(t, got, disabledRoleId,
- "禁用角色的 id 不应被返回")
- }
- // =============================================================================
- // audit 回归:DeleteByUserIdAndRoleIdsTx 批量删除正确性
- // =============================================================================
- // TC-0707: 批量删除指定 (userId, roleIds) 组合;未命中的绑定不受影响
- func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- r1, r2, r3 := randUserRoleId(), randUserRoleId(), randUserRoleId()
- ts := time.Now().Unix()
- var recIds []int64
- for _, rid := range []int64{r1, r2, r3} {
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: rid, CreateTime: ts, UpdateTime: ts})
- require.NoError(t, err)
- id, _ := res.LastInsertId()
- recIds = append(recIds, id)
- }
- defer func() {
- for _, id := range recIds {
- testutil.CleanTable(ctx, conn, "sys_user_role", id)
- }
- }()
- // 删除 r1 和 r3,保留 r2
- err := m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByUserIdAndRoleIdsTx(c, session, userId, []int64{r1, r3})
- })
- require.NoError(t, err)
- got, err := m.FindRoleIdsByUserId(ctx, userId)
- require.NoError(t, err)
- assert.ElementsMatch(t, []int64{r2}, got, "仅保留未被删除的 r2")
- }
- // TC-0708: DeleteByUserIdAndRoleIdsTx 空列表时应立即返回 nil,不执行 SQL
- func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx_EmptyIsNoop(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- userId := randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- require.NoError(t, err)
- id, _ := res.LastInsertId()
- defer testutil.CleanTable(ctx, conn, "sys_user_role", id)
- // 空列表
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByUserIdAndRoleIdsTx(c, session, userId, []int64{})
- })
- require.NoError(t, err)
- // 原有绑定未被删除
- got, err := m.FindRoleIdsByUserId(ctx, userId)
- require.NoError(t, err)
- assert.Contains(t, got, roleId, "空 roleIds 列表不应删除任何数据")
- }
- // TC-0709: DeleteByUserIdAndRoleIdsTx 不影响其它 userId 的同 roleId 绑定(WHERE userId 约束)
- func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx_OtherUserNotAffected(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- u1, u2 := randUserRoleId(), randUserRoleId()
- roleId := randUserRoleId()
- ts := time.Now().Unix()
- res1, err := m.Insert(ctx, &SysUserRole{UserId: u1, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- require.NoError(t, err)
- id1, _ := res1.LastInsertId()
- res2, err := m.Insert(ctx, &SysUserRole{UserId: u2, RoleId: roleId, CreateTime: ts, UpdateTime: ts})
- require.NoError(t, err)
- id2, _ := res2.LastInsertId()
- defer func() {
- testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2)
- }()
- err = m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByUserIdAndRoleIdsTx(c, session, u1, []int64{roleId})
- })
- require.NoError(t, err)
- // u1 的被删,u2 的保留
- got1, _ := m.FindRoleIdsByUserId(ctx, u1)
- got2, _ := m.FindRoleIdsByUserId(ctx, u2)
- assert.NotContains(t, got1, roleId)
- assert.Contains(t, got2, roleId)
- }
- func seedUserRoleWithPrimedCache(t *testing.T, ctx context.Context, m SysUserRoleModel, userId, roleId int64) int64 {
- t.Helper()
- ts := time.Now().Unix()
- res, err := m.Insert(ctx, &SysUserRole{
- UserId: userId, RoleId: roleId, CreateTime: ts, UpdateTime: ts,
- })
- require.NoError(t, err)
- id, err := res.LastInsertId()
- require.NoError(t, err)
- _, err = m.FindOne(ctx, id)
- require.NoError(t, err, "FindOne 为 id 维度缓存预热")
- _, err = m.FindOneByUserIdRoleId(ctx, userId, roleId)
- require.NoError(t, err, "FindOneByUserIdRoleId 为复合维度缓存预热")
- return id
- }
- func assertCacheKeysGone(t *testing.T, rds *redis.Redis, idKey, compositeKey string) {
- t.Helper()
- got, err := rds.Get(idKey)
- require.NoError(t, err)
- assert.Empty(t, got, "id 维度缓存 key %q 应被 DELETE 一并失效", idKey)
- got, err = rds.Get(compositeKey)
- require.NoError(t, err)
- assert.Empty(t, got, "复合维度缓存 key %q 应被 DELETE 一并失效", compositeKey)
- }
- // TC-1062: DeleteByRoleIdTx 对多行同时失效 id + composite 缓存
- func TestSysUserRoleModel_DeleteByRoleIdTx_InvalidatesAllKeyCols(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- rds := redis.MustNewRedis(testutil.GetTestConfig().CacheRedis.Nodes[0].RedisConf)
- cachePrefix := testutil.GetTestCachePrefix()
- roleId := randUserRoleId()
- u1, u2 := randUserRoleId(), randUserRoleId()
- if u1 == u2 {
- u2 = u1 + 1
- }
- id1 := seedUserRoleWithPrimedCache(t, ctx, m, u1, roleId)
- id2 := seedUserRoleWithPrimedCache(t, ctx, m, u2, roleId)
- t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2) })
- require.NoError(t,
- m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByRoleIdTx(c, session, roleId)
- }))
- // 裁剪 SELECT 为三列后,构造缓存 key 需要的 id/userId/roleId 仍被携带回来。
- // 任何一条遗漏都会让这里的断言炸掉(旧行的缓存还在,FindOne 就能从缓存拿到"幽灵行")。
- for _, c := range []struct {
- id, uid int64
- }{{id1, u1}, {id2, u2}} {
- idKey := fmt.Sprintf("%s:cache:sysUserRole:id:%v", cachePrefix, c.id)
- compKey := fmt.Sprintf("%s:cache:sysUserRole:userId:roleId:%v:%v", cachePrefix, c.uid, roleId)
- assertCacheKeysGone(t, rds, idKey, compKey)
- }
- // 终态真相:两行都不应再存在于 DB。
- for _, id := range []int64{id1, id2} {
- _, err := m.FindOne(ctx, id)
- assert.ErrorIs(t, err, ErrNotFound,
- "DELETE 已提交,FindOne 必须回落 DB 读到 ErrNotFound;"+
- "若仍查到旧行说明裁剪后的 SELECT 漏掉了 id 列,id 维度缓存仍在")
- }
- }
- // TC-1063: DeleteByUserIdAndRoleIdsTx 的批量 IN 路径
- func TestSysUserRoleModel_DeleteByUserIdAndRoleIdsTx_InvalidatesAllKeyCols(t *testing.T) {
- ctx := context.Background()
- conn := testutil.GetTestSqlConn()
- m := NewSysUserRoleModel(conn, testutil.GetTestCacheConf(), testutil.GetTestCachePrefix())
- rds := redis.MustNewRedis(testutil.GetTestConfig().CacheRedis.Nodes[0].RedisConf)
- cachePrefix := testutil.GetTestCachePrefix()
- userId := randUserRoleId()
- r1, r2, r3 := randUserRoleId(), randUserRoleId()+1, randUserRoleId()+2
- id1 := seedUserRoleWithPrimedCache(t, ctx, m, userId, r1)
- id2 := seedUserRoleWithPrimedCache(t, ctx, m, userId, r2)
- // r3 作为"不在删除集合内"的对照组:这一行不得被误伤
- id3 := seedUserRoleWithPrimedCache(t, ctx, m, userId, r3)
- t.Cleanup(func() { testutil.CleanTable(ctx, conn, "sys_user_role", id1, id2, id3) })
- require.NoError(t,
- m.TransactCtx(ctx, func(c context.Context, session sqlx.Session) error {
- return m.DeleteByUserIdAndRoleIdsTx(c, session, userId, []int64{r1, r2})
- }))
- // 被删的两条:id+composite 都必须失效。
- for _, c := range []struct {
- id, rid int64
- }{{id1, r1}, {id2, r2}} {
- idKey := fmt.Sprintf("%s:cache:sysUserRole:id:%v", cachePrefix, c.id)
- compKey := fmt.Sprintf("%s:cache:sysUserRole:userId:roleId:%v:%v", cachePrefix, userId, c.rid)
- assertCacheKeysGone(t, rds, idKey, compKey)
- _, err := m.FindOne(ctx, c.id)
- assert.ErrorIs(t, err, ErrNotFound)
- }
- // 未被删的第 3 条:DB 仍在,FindOne 必须成功。
- got, err := m.FindOne(ctx, id3)
- require.NoError(t, err,
- "防误伤:IN (r1, r2) 不得把 r3 的行带走;"+
- "若失败,说明裁剪后 SELECT 把对照组的 key 也返回并被 ExecCtx 一并失效")
- assert.Equal(t, id3, got.Id)
- assert.Equal(t, r3, got.RoleId)
- }
|