updateUserLogic_test.go 13 KB

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