gitea源码

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112
  1. // Copyright 2025 The Gitea Authors. All rights reserved.
  2. // SPDX-License-Identifier: MIT
  3. package cmd
  4. import (
  5. "strconv"
  6. "strings"
  7. "testing"
  8. auth_model "code.gitea.io/gitea/models/auth"
  9. "code.gitea.io/gitea/models/db"
  10. "code.gitea.io/gitea/models/unittest"
  11. user_model "code.gitea.io/gitea/models/user"
  12. "github.com/stretchr/testify/require"
  13. )
  14. func TestAdminUserDelete(t *testing.T) {
  15. ctx := t.Context()
  16. defer func() {
  17. require.NoError(t, db.TruncateBeans(t.Context(), &user_model.User{}))
  18. require.NoError(t, db.TruncateBeans(t.Context(), &user_model.EmailAddress{}))
  19. require.NoError(t, db.TruncateBeans(t.Context(), &auth_model.AccessToken{}))
  20. }()
  21. setupTestUser := func(t *testing.T) {
  22. unittest.AssertNotExistsBean(t, &user_model.User{LowerName: "testuser"})
  23. err := microcmdUserCreate().Run(t.Context(), []string{"create", "--username", "testuser", "--email", "testuser@gitea.local", "--random-password"})
  24. require.NoError(t, err)
  25. }
  26. t.Run("delete user by id", func(t *testing.T) {
  27. setupTestUser(t)
  28. u := unittest.AssertExistsAndLoadBean(t, &user_model.User{LowerName: "testuser"})
  29. err := microcmdUserDelete().Run(ctx, []string{"delete-test", "--id", strconv.FormatInt(u.ID, 10)})
  30. require.NoError(t, err)
  31. unittest.AssertNotExistsBean(t, &user_model.User{LowerName: "testuser"})
  32. })
  33. t.Run("delete user by username", func(t *testing.T) {
  34. setupTestUser(t)
  35. err := microcmdUserDelete().Run(ctx, []string{"delete-test", "--username", "testuser"})
  36. require.NoError(t, err)
  37. unittest.AssertNotExistsBean(t, &user_model.User{LowerName: "testuser"})
  38. })
  39. t.Run("delete user by email", func(t *testing.T) {
  40. setupTestUser(t)
  41. err := microcmdUserDelete().Run(ctx, []string{"delete-test", "--email", "testuser@gitea.local"})
  42. require.NoError(t, err)
  43. unittest.AssertNotExistsBean(t, &user_model.User{LowerName: "testuser"})
  44. })
  45. t.Run("delete user by all 3 attributes", func(t *testing.T) {
  46. setupTestUser(t)
  47. u := unittest.AssertExistsAndLoadBean(t, &user_model.User{LowerName: "testuser"})
  48. err := microcmdUserDelete().Run(ctx, []string{"delete", "--id", strconv.FormatInt(u.ID, 10), "--username", "testuser", "--email", "testuser@gitea.local"})
  49. require.NoError(t, err)
  50. unittest.AssertNotExistsBean(t, &user_model.User{LowerName: "testuser"})
  51. })
  52. }
  53. func TestAdminUserDeleteFailure(t *testing.T) {
  54. testCases := []struct {
  55. name string
  56. args []string
  57. expectedErr string
  58. }{
  59. {
  60. name: "no user to delete",
  61. args: []string{"delete", "--username", "nonexistentuser"},
  62. expectedErr: "user does not exist",
  63. },
  64. {
  65. name: "user exists but provided username does not match",
  66. args: []string{"delete", "--email", "testuser@gitea.local", "--username", "wrongusername"},
  67. expectedErr: "the user testuser who has email testuser@gitea.local does not match the provided username wrongusername",
  68. },
  69. {
  70. name: "user exists but provided id does not match",
  71. args: []string{"delete", "--username", "testuser", "--id", "999"},
  72. expectedErr: "the user testuser does not match the provided id 999",
  73. },
  74. {
  75. name: "no required flags are provided",
  76. args: []string{"delete"},
  77. expectedErr: "You must provide the id, username or email of a user to delete",
  78. },
  79. }
  80. for _, tc := range testCases {
  81. t.Run(tc.name, func(t *testing.T) {
  82. ctx := t.Context()
  83. if strings.Contains(tc.name, "user exists") {
  84. unittest.AssertNotExistsBean(t, &user_model.User{LowerName: "testuser"})
  85. err := microcmdUserCreate().Run(t.Context(), []string{"create", "--username", "testuser", "--email", "testuser@gitea.local", "--random-password"})
  86. require.NoError(t, err)
  87. }
  88. err := microcmdUserDelete().Run(ctx, tc.args)
  89. require.Error(t, err)
  90. require.Contains(t, err.Error(), tc.expectedErr)
  91. })
  92. require.NoError(t, db.TruncateBeans(t.Context(), &user_model.User{}))
  93. require.NoError(t, db.TruncateBeans(t.Context(), &user_model.EmailAddress{}))
  94. require.NoError(t, db.TruncateBeans(t.Context(), &auth_model.AccessToken{}))
  95. }
  96. }