updateUserLogic_test.go 12 KB

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