updateUserLogic_test.go 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468
  1. package user
  2. import (
  3. "context"
  4. "errors"
  5. "testing"
  6. "time"
  7. "perms-system-server/internal/consts"
  8. "perms-system-server/internal/loaders"
  9. deptModel "perms-system-server/internal/model/dept"
  10. userModel "perms-system-server/internal/model/user"
  11. "perms-system-server/internal/response"
  12. "perms-system-server/internal/svc"
  13. "perms-system-server/internal/testutil"
  14. "perms-system-server/internal/testutil/ctxhelper"
  15. "perms-system-server/internal/types"
  16. "github.com/stretchr/testify/assert"
  17. "github.com/stretchr/testify/require"
  18. )
  19. func insertTestDept(t *testing.T, ctx context.Context, svcCtx *svc.ServiceContext) int64 {
  20. t.Helper()
  21. now := time.Now().Unix()
  22. res, err := svcCtx.SysDeptModel.Insert(ctx, &deptModel.SysDept{
  23. Name: testutil.UniqueId(),
  24. ParentId: 0,
  25. Path: "/",
  26. Sort: 0,
  27. DeptType: "NORMAL",
  28. Status: consts.StatusEnabled,
  29. CreateTime: now,
  30. UpdateTime: now,
  31. })
  32. require.NoError(t, err)
  33. id, _ := res.LastInsertId()
  34. return id
  35. }
  36. // TC-0112: 正常更新
  37. func TestUpdateUser_Success(t *testing.T) {
  38. ctx := ctxhelper.SuperAdminCtx()
  39. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  40. conn := testutil.GetTestSqlConn()
  41. username := testutil.UniqueId()
  42. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  43. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  44. deptId := insertTestDept(t, ctx, svcCtx)
  45. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", deptId) })
  46. logic := NewUpdateUserLogic(ctx, svcCtx)
  47. err := logic.UpdateUser(&types.UpdateUserReq{
  48. Id: userId,
  49. Nickname: strPtr("新昵称"),
  50. Email: strPtr("[email protected]"),
  51. Phone: strPtr("13900139000"),
  52. Remark: strPtr("更新备注"),
  53. DeptId: int64Ptr(deptId),
  54. })
  55. require.NoError(t, err)
  56. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  57. require.NoError(t, err)
  58. assert.Equal(t, "新昵称", user.Nickname)
  59. assert.Equal(t, "[email protected]", user.Email)
  60. assert.Equal(t, "13900139000", user.Phone)
  61. assert.Equal(t, "更新备注", user.Remark)
  62. assert.Equal(t, deptId, user.DeptId)
  63. }
  64. // TC-0113: 不存在
  65. func TestUpdateUser_NotFound(t *testing.T) {
  66. ctx := ctxhelper.SuperAdminCtx()
  67. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  68. logic := NewUpdateUserLogic(ctx, svcCtx)
  69. err := logic.UpdateUser(&types.UpdateUserReq{
  70. Id: 999999999,
  71. Nickname: strPtr("ghost"),
  72. })
  73. require.Error(t, err)
  74. var codeErr *response.CodeError
  75. require.True(t, errors.As(err, &codeErr))
  76. assert.Equal(t, 404, codeErr.Code())
  77. assert.Equal(t, "用户不存在", codeErr.Error())
  78. }
  79. // TC-0114: 仅传id
  80. func TestUpdateUser_OnlyId_NothingChanges(t *testing.T) {
  81. ctx := ctxhelper.SuperAdminCtx()
  82. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  83. conn := testutil.GetTestSqlConn()
  84. username := testutil.UniqueId()
  85. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  86. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  87. before, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  88. require.NoError(t, err)
  89. logic := NewUpdateUserLogic(ctx, svcCtx)
  90. err = logic.UpdateUser(&types.UpdateUserReq{Id: userId})
  91. require.NoError(t, err)
  92. after, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  93. require.NoError(t, err)
  94. assert.Equal(t, before.Nickname, after.Nickname)
  95. assert.Equal(t, before.Email, after.Email)
  96. assert.Equal(t, before.Phone, after.Phone)
  97. assert.Equal(t, before.DeptId, after.DeptId)
  98. }
  99. // TC-0115: 清空nickname
  100. func TestUpdateUser_ClearNickname(t *testing.T) {
  101. ctx := ctxhelper.SuperAdminCtx()
  102. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  103. conn := testutil.GetTestSqlConn()
  104. username := testutil.UniqueId()
  105. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  106. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  107. logic := NewUpdateUserLogic(ctx, svcCtx)
  108. err := logic.UpdateUser(&types.UpdateUserReq{
  109. Id: userId,
  110. Nickname: strPtr(""),
  111. })
  112. require.NoError(t, err)
  113. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  114. require.NoError(t, err)
  115. assert.Equal(t, "", user.Nickname)
  116. }
  117. // TC-0116: 清空email
  118. func TestUpdateUser_ClearEmail(t *testing.T) {
  119. ctx := ctxhelper.SuperAdminCtx()
  120. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  121. conn := testutil.GetTestSqlConn()
  122. username := testutil.UniqueId()
  123. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  124. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  125. logic := NewUpdateUserLogic(ctx, svcCtx)
  126. err := logic.UpdateUser(&types.UpdateUserReq{
  127. Id: userId,
  128. Email: strPtr(""),
  129. })
  130. require.NoError(t, err)
  131. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  132. require.NoError(t, err)
  133. assert.Equal(t, "", user.Email)
  134. }
  135. // TC-0118: 非法email格式
  136. func TestUpdateUser_InvalidEmail(t *testing.T) {
  137. ctx := ctxhelper.SuperAdminCtx()
  138. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  139. conn := testutil.GetTestSqlConn()
  140. username := testutil.UniqueId()
  141. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  142. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  143. logic := NewUpdateUserLogic(ctx, svcCtx)
  144. err := logic.UpdateUser(&types.UpdateUserReq{
  145. Id: userId,
  146. Email: strPtr("bad-email"),
  147. })
  148. require.Error(t, err)
  149. var codeErr *response.CodeError
  150. require.True(t, errors.As(err, &codeErr))
  151. assert.Equal(t, 400, codeErr.Code())
  152. assert.Equal(t, "邮箱格式不正确", codeErr.Error())
  153. }
  154. // TC-0122: DeptId设为0(取消部门)
  155. func TestUpdateUser_DeptIdZero_Clear(t *testing.T) {
  156. ctx := ctxhelper.SuperAdminCtx()
  157. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  158. conn := testutil.GetTestSqlConn()
  159. username := testutil.UniqueId()
  160. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  161. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  162. deptId := insertTestDept(t, ctx, svcCtx)
  163. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", deptId) })
  164. logic := NewUpdateUserLogic(ctx, svcCtx)
  165. err := logic.UpdateUser(&types.UpdateUserReq{
  166. Id: userId,
  167. DeptId: int64Ptr(deptId),
  168. })
  169. require.NoError(t, err)
  170. err = logic.UpdateUser(&types.UpdateUserReq{
  171. Id: userId,
  172. DeptId: int64Ptr(0),
  173. })
  174. require.NoError(t, err)
  175. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  176. require.NoError(t, err)
  177. assert.Equal(t, int64(0), user.DeptId)
  178. }
  179. // TC-0123: DeptId设为正值
  180. func TestUpdateUser_DeptIdSet(t *testing.T) {
  181. ctx := ctxhelper.SuperAdminCtx()
  182. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  183. conn := testutil.GetTestSqlConn()
  184. username := testutil.UniqueId()
  185. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  186. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  187. deptId := insertTestDept(t, ctx, svcCtx)
  188. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", deptId) })
  189. logic := NewUpdateUserLogic(ctx, svcCtx)
  190. err := logic.UpdateUser(&types.UpdateUserReq{
  191. Id: userId,
  192. DeptId: int64Ptr(deptId),
  193. })
  194. require.NoError(t, err)
  195. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  196. require.NoError(t, err)
  197. assert.Equal(t, deptId, user.DeptId)
  198. }
  199. // TC-0124: DeptId不传(nil)
  200. func TestUpdateUser_NilDeptId_Unchanged(t *testing.T) {
  201. ctx := ctxhelper.SuperAdminCtx()
  202. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  203. conn := testutil.GetTestSqlConn()
  204. username := testutil.UniqueId()
  205. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  206. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  207. deptId := insertTestDept(t, ctx, svcCtx)
  208. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", deptId) })
  209. logic := NewUpdateUserLogic(ctx, svcCtx)
  210. err := logic.UpdateUser(&types.UpdateUserReq{
  211. Id: userId,
  212. DeptId: int64Ptr(deptId),
  213. })
  214. require.NoError(t, err)
  215. err = logic.UpdateUser(&types.UpdateUserReq{
  216. Id: userId,
  217. Nickname: strPtr("changed"),
  218. })
  219. require.NoError(t, err)
  220. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  221. require.NoError(t, err)
  222. assert.Equal(t, deptId, user.DeptId)
  223. assert.Equal(t, "changed", user.Nickname)
  224. }
  225. // TC-0117: 清空remark
  226. func TestUpdateUser_ClearRemark(t *testing.T) {
  227. ctx := ctxhelper.SuperAdminCtx()
  228. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  229. conn := testutil.GetTestSqlConn()
  230. username := testutil.UniqueId()
  231. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  232. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  233. logic := NewUpdateUserLogic(ctx, svcCtx)
  234. err := logic.UpdateUser(&types.UpdateUserReq{
  235. Id: userId,
  236. Remark: strPtr("some remark"),
  237. })
  238. require.NoError(t, err)
  239. err = logic.UpdateUser(&types.UpdateUserReq{
  240. Id: userId,
  241. Remark: strPtr(""),
  242. })
  243. require.NoError(t, err)
  244. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  245. require.NoError(t, err)
  246. assert.Equal(t, "", user.Remark)
  247. }
  248. // TC-0120: 合法phone
  249. func TestUpdateUser_ValidInternationalPhone(t *testing.T) {
  250. ctx := ctxhelper.SuperAdminCtx()
  251. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  252. conn := testutil.GetTestSqlConn()
  253. username := testutil.UniqueId()
  254. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  255. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  256. logic := NewUpdateUserLogic(ctx, svcCtx)
  257. err := logic.UpdateUser(&types.UpdateUserReq{
  258. Id: userId,
  259. Phone: strPtr("+8613800138000"),
  260. })
  261. require.NoError(t, err)
  262. user, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  263. require.NoError(t, err)
  264. assert.Equal(t, "+8613800138000", user.Phone)
  265. }
  266. // TC-0119: 非法phone格式
  267. func TestUpdateUser_InvalidPhone(t *testing.T) {
  268. ctx := ctxhelper.SuperAdminCtx()
  269. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  270. conn := testutil.GetTestSqlConn()
  271. username := testutil.UniqueId()
  272. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  273. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  274. logic := NewUpdateUserLogic(ctx, svcCtx)
  275. err := logic.UpdateUser(&types.UpdateUserReq{
  276. Id: userId,
  277. Phone: strPtr("12345"),
  278. })
  279. require.Error(t, err)
  280. var codeErr *response.CodeError
  281. require.True(t, errors.As(err, &codeErr))
  282. assert.Equal(t, 400, codeErr.Code())
  283. assert.Equal(t, "手机号格式不正确", codeErr.Error())
  284. }
  285. // TC-0121: 不传email(nil)
  286. func TestUpdateUser_NilEmail_Unchanged(t *testing.T) {
  287. ctx := ctxhelper.SuperAdminCtx()
  288. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  289. conn := testutil.GetTestSqlConn()
  290. username := testutil.UniqueId()
  291. userId := insertTestUser(t, ctx, username, testutil.HashPassword("pass"))
  292. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  293. before, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  294. require.NoError(t, err)
  295. originalEmail := before.Email
  296. logic := NewUpdateUserLogic(ctx, svcCtx)
  297. err = logic.UpdateUser(&types.UpdateUserReq{
  298. Id: userId,
  299. Nickname: strPtr("changed-nick"),
  300. })
  301. require.NoError(t, err)
  302. after, err := svcCtx.SysUserModel.FindOne(ctx, userId)
  303. require.NoError(t, err)
  304. assert.Equal(t, originalEmail, after.Email)
  305. assert.Equal(t, "changed-nick", after.Nickname)
  306. }
  307. // TC-0491: 非本人非超管修改拒绝
  308. func TestUpdateUser_NonSelfNonSuperAdminRejected(t *testing.T) {
  309. ctx := ctxhelper.MemberCtx("test_product")
  310. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  311. logic := NewUpdateUserLogic(ctx, svcCtx)
  312. err := logic.UpdateUser(&types.UpdateUserReq{Id: 999, Nickname: strPtr("hacked")})
  313. require.Error(t, err)
  314. var ce *response.CodeError
  315. require.True(t, errors.As(err, &ce))
  316. assert.Equal(t, 403, ce.Code())
  317. assert.Contains(t, ce.Error(), "仅允许修改自己的信息或超管操作")
  318. }
  319. // TC-0511: updateUser自己修改DeptId被拒绝
  320. func TestUpdateUser_SelfEditDeptIdRejected(t *testing.T) {
  321. ctx := ctxhelper.CustomCtx(&loaders.UserDetails{
  322. UserId: 100,
  323. Username: "self_user",
  324. Status: consts.StatusEnabled,
  325. })
  326. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  327. logic := NewUpdateUserLogic(ctx, svcCtx)
  328. err := logic.UpdateUser(&types.UpdateUserReq{Id: 100, DeptId: int64Ptr(5)})
  329. require.Error(t, err)
  330. var ce *response.CodeError
  331. require.True(t, errors.As(err, &ce))
  332. assert.Equal(t, 403, ce.Code())
  333. assert.Equal(t, "不允许修改自己的部门和状态", ce.Error())
  334. }
  335. // TC-0512: updateUser自己修改Status被拒绝
  336. func TestUpdateUser_SelfEditStatusRejected(t *testing.T) {
  337. ctx := ctxhelper.CustomCtx(&loaders.UserDetails{
  338. UserId: 100,
  339. Username: "self_user",
  340. Status: consts.StatusEnabled,
  341. })
  342. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  343. logic := NewUpdateUserLogic(ctx, svcCtx)
  344. err := logic.UpdateUser(&types.UpdateUserReq{Id: 100, Status: 2})
  345. require.Error(t, err)
  346. var ce *response.CodeError
  347. require.True(t, errors.As(err, &ce))
  348. assert.Equal(t, 403, ce.Code())
  349. assert.Equal(t, "不允许修改自己的部门和状态", ce.Error())
  350. }
  351. // TC-0513: updateUser未登录被拒绝
  352. func TestUpdateUser_NotLoggedInRejected(t *testing.T) {
  353. ctx := context.Background()
  354. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  355. logic := NewUpdateUserLogic(ctx, svcCtx)
  356. err := logic.UpdateUser(&types.UpdateUserReq{Id: 1, Nickname: strPtr("hacked")})
  357. require.Error(t, err)
  358. var ce *response.CodeError
  359. require.True(t, errors.As(err, &ce))
  360. assert.Equal(t, 401, ce.Code())
  361. assert.Equal(t, "未登录", ce.Error())
  362. }
  363. // TC-0544: 超管A通过updateUser修改超管B的状态被拒绝(H-2修复验证)
  364. func TestUpdateUser_SuperAdminCannotFreezeOtherSuperAdmin(t *testing.T) {
  365. ctx := ctxhelper.SuperAdminCtx()
  366. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  367. conn := testutil.GetTestSqlConn()
  368. now := time.Now().Unix()
  369. uRes, err := svcCtx.SysUserModel.Insert(ctx, &userModel.SysUser{
  370. Username: testutil.UniqueId(),
  371. Password: testutil.HashPassword("pass"),
  372. Nickname: "super_b",
  373. IsSuperAdmin: consts.IsSuperAdminYes,
  374. MustChangePassword: 2,
  375. Status: consts.StatusEnabled,
  376. CreateTime: now,
  377. UpdateTime: now,
  378. })
  379. require.NoError(t, err)
  380. superBId, _ := uRes.LastInsertId()
  381. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", superBId) })
  382. logic := NewUpdateUserLogic(ctx, svcCtx)
  383. err = logic.UpdateUser(&types.UpdateUserReq{
  384. Id: superBId,
  385. Status: consts.StatusDisabled,
  386. })
  387. require.Error(t, err)
  388. var ce *response.CodeError
  389. require.True(t, errors.As(err, &ce))
  390. assert.Equal(t, 403, ce.Code())
  391. assert.Equal(t, "不能通过此接口修改超级管理员的状态", ce.Error())
  392. user, err := svcCtx.SysUserModel.FindOne(ctx, superBId)
  393. require.NoError(t, err)
  394. assert.Equal(t, int64(consts.StatusEnabled), user.Status, "超管B的状态不应被修改")
  395. }