deleteDeptLogic_test.go 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. package dept
  2. import (
  3. "database/sql"
  4. "errors"
  5. "testing"
  6. "time"
  7. userModel "perms-system-server/internal/model/user"
  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-0094: 正常删除(无子部门)
  17. func TestDeleteDept_NoChildren(t *testing.T) {
  18. ctx := ctxhelper.SuperAdminCtx()
  19. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  20. conn := testutil.GetTestSqlConn()
  21. id, err := insertDeptRaw(ctx, svcCtx, 0, "del_"+testutil.UniqueId(), "/")
  22. require.NoError(t, err)
  23. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", id) })
  24. l := NewDeleteDeptLogic(ctx, svcCtx)
  25. err = l.DeleteDept(&types.DeleteDeptReq{Id: id})
  26. require.NoError(t, err)
  27. _, err = svcCtx.SysDeptModel.FindOne(ctx, id)
  28. assert.Error(t, err)
  29. }
  30. // TC-0096: 不存在的部门
  31. func TestDeleteDept_NonExistentDept(t *testing.T) {
  32. ctx := ctxhelper.SuperAdminCtx()
  33. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  34. l := NewDeleteDeptLogic(ctx, svcCtx)
  35. err := l.DeleteDept(&types.DeleteDeptReq{Id: 999999999})
  36. assert.NoError(t, err)
  37. }
  38. // TC-0095: 有子部门
  39. func TestDeleteDept_HasChildren(t *testing.T) {
  40. ctx := ctxhelper.SuperAdminCtx()
  41. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  42. conn := testutil.GetTestSqlConn()
  43. parentId, err := insertDeptRaw(ctx, svcCtx, 0, "del_par_"+testutil.UniqueId(), "/")
  44. require.NoError(t, err)
  45. parent, _ := svcCtx.SysDeptModel.FindOne(ctx, parentId)
  46. childId, err := insertDeptRaw(ctx, svcCtx, parentId, "del_child_"+testutil.UniqueId(), parent.Path)
  47. require.NoError(t, err)
  48. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", childId, parentId) })
  49. l := NewDeleteDeptLogic(ctx, svcCtx)
  50. err = l.DeleteDept(&types.DeleteDeptReq{Id: parentId})
  51. require.Error(t, err)
  52. var ce *response.CodeError
  53. require.True(t, errors.As(err, &ce))
  54. assert.Equal(t, 400, ce.Code())
  55. assert.Contains(t, ce.Error(), "该部门下存在子部门,无法删除")
  56. _, err = svcCtx.SysDeptModel.FindOne(ctx, parentId)
  57. assert.NoError(t, err)
  58. }
  59. // TC-0097: 部门下有关联用户
  60. func TestDeleteDept_HasAssociatedUsers(t *testing.T) {
  61. ctx := ctxhelper.SuperAdminCtx()
  62. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  63. conn := testutil.GetTestSqlConn()
  64. deptId, err := insertDeptRaw(ctx, svcCtx, 0, "has_users_"+testutil.UniqueId(), "/")
  65. require.NoError(t, err)
  66. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_dept`", deptId) })
  67. now := time.Now().Unix()
  68. userRes, err := svcCtx.SysUserModel.Insert(ctx, &userModel.SysUser{
  69. Username: "dept_user_" + testutil.UniqueId(),
  70. Password: testutil.HashPassword("pass123456"),
  71. Nickname: "test",
  72. Avatar: sql.NullString{},
  73. DeptId: deptId,
  74. IsSuperAdmin: 2,
  75. MustChangePassword: 2,
  76. Status: 1,
  77. CreateTime: now,
  78. UpdateTime: now,
  79. })
  80. require.NoError(t, err)
  81. userId, _ := userRes.LastInsertId()
  82. t.Cleanup(func() { testutil.CleanTable(ctx, conn, "`sys_user`", userId) })
  83. l := NewDeleteDeptLogic(ctx, svcCtx)
  84. err = l.DeleteDept(&types.DeleteDeptReq{Id: deptId})
  85. require.Error(t, err)
  86. var ce *response.CodeError
  87. require.True(t, errors.As(err, &ce))
  88. assert.Equal(t, 400, ce.Code())
  89. assert.Equal(t, "该部门下仍有关联用户,无法删除", ce.Error())
  90. _, err = svcCtx.SysDeptModel.FindOne(ctx, deptId)
  91. assert.NoError(t, err)
  92. }
  93. // TC-0513: deleteDept非超管拒绝
  94. func TestDeleteDept_NonSuperAdminRejected(t *testing.T) {
  95. ctx := ctxhelper.AdminCtx("test_product")
  96. svcCtx := svc.NewServiceContext(testutil.GetTestConfig())
  97. l := NewDeleteDeptLogic(ctx, svcCtx)
  98. err := l.DeleteDept(&types.DeleteDeptReq{Id: 1})
  99. require.Error(t, err)
  100. var ce *response.CodeError
  101. require.True(t, errors.As(err, &ce))
  102. assert.Equal(t, 403, ce.Code())
  103. }