userDetailLogic_test.go 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105
  1. package user
  2. import (
  3. "database/sql"
  4. "errors"
  5. "testing"
  6. "time"
  7. userModel "perms-system-server/internal/model/user"
  8. "perms-system-server/internal/model/userrole"
  9. "perms-system-server/internal/response"
  10. "perms-system-server/internal/svc"
  11. "perms-system-server/internal/testutil"
  12. "perms-system-server/internal/testutil/ctxhelper"
  13. "perms-system-server/internal/types"
  14. "github.com/stretchr/testify/assert"
  15. "github.com/stretchr/testify/require"
  16. )
  17. // TC-0181: 正常查询 —— 超管在具体产品上下文下仅应返回该产品下的 roleIds(audit M-3 修复后的行为)
  18. func TestUserDetail_Success(t *testing.T) {
  19. ctx := ctxhelper.SuperAdminCtx()
  20. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  21. conn := testutil.GetTestSqlConn()
  22. username := testutil.UniqueId()
  23. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  24. // 插入两条"当前产品"下的真实 sys_role 以及一条属于其它产品的 sys_role,
  25. // 用户同时绑定这三个角色。超管在 test_product 上下文下应当只看到前两个。
  26. roleInCurrent1 := insertTestRole(t, svcCtx, "test_product", 1)
  27. roleInCurrent2 := insertTestRole(t, svcCtx, "test_product", 1)
  28. roleInOther := insertTestRole(t, svcCtx, "other_product", 1)
  29. now := time.Now().Unix()
  30. var roleRecordIds []int64
  31. for _, roleId := range []int64{roleInCurrent1, roleInCurrent2, roleInOther} {
  32. res, err := svcCtx.SysUserRoleModel.Insert(ctx, &userrole.SysUserRole{
  33. UserId: userId,
  34. RoleId: roleId,
  35. CreateTime: now,
  36. UpdateTime: now,
  37. })
  38. require.NoError(t, err)
  39. id, _ := res.LastInsertId()
  40. roleRecordIds = append(roleRecordIds, id)
  41. }
  42. t.Cleanup(func() {
  43. testutil.CleanTable(ctx, conn, "`sys_user_role`", roleRecordIds...)
  44. testutil.CleanTable(ctx, conn, "`sys_role`", roleInCurrent1, roleInCurrent2, roleInOther)
  45. testutil.CleanTable(ctx, conn, "`sys_user`", userId)
  46. })
  47. logic := NewUserDetailLogic(ctx, svcCtx)
  48. resp, err := logic.UserDetail(&types.UserDetailReq{Id: userId})
  49. require.NoError(t, err)
  50. require.NotNil(t, resp)
  51. assert.Equal(t, userId, resp.Id)
  52. assert.Equal(t, username, resp.Username)
  53. // 修复后:超管在产品上下文里只看到 test_product 的角色;other_product 的角色不应返回
  54. assert.ElementsMatch(t, []int64{roleInCurrent1, roleInCurrent2}, resp.RoleIds)
  55. assert.NotContains(t, resp.RoleIds, roleInOther, "超管在具体产品上下文不应返回其它产品的 roleIds (audit M-3)")
  56. }
  57. // TC-0182: 正常查询-含Avatar
  58. func TestUserDetail_WithAvatar(t *testing.T) {
  59. ctx := ctxhelper.SuperAdminCtx()
  60. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  61. conn := testutil.GetTestSqlConn()
  62. userId := insertTestUserFull(t, ctx, &userModel.SysUser{
  63. Username: testutil.UniqueId(),
  64. Password: testutil.HashPassword("pass"),
  65. Nickname: "avatar_user",
  66. Avatar: sql.NullString{String: "https://example.com/avatar.png", Valid: true},
  67. IsSuperAdmin: 2,
  68. MustChangePassword: 2,
  69. Status: 1,
  70. })
  71. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  72. logic := NewUserDetailLogic(ctx, svcCtx)
  73. resp, err := logic.UserDetail(&types.UserDetailReq{Id: userId})
  74. require.NoError(t, err)
  75. require.NotNil(t, resp)
  76. assert.Equal(t, "https://example.com/avatar.png", resp.Avatar)
  77. }
  78. // TC-0183: 不存在
  79. func TestUserDetail_NotFound(t *testing.T) {
  80. ctx := ctxhelper.SuperAdminCtx()
  81. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  82. logic := NewUserDetailLogic(ctx, svcCtx)
  83. _, err := logic.UserDetail(&types.UserDetailReq{Id: 999999999})
  84. require.Error(t, err)
  85. var codeErr *response.CodeError
  86. require.True(t, errors.As(err, &codeErr))
  87. assert.Equal(t, 404, codeErr.Code())
  88. assert.Equal(t, "用户不存在", codeErr.Error())
  89. }