usage_handler_search_users_test.go 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. package admin
  2. import (
  3. "context"
  4. "encoding/json"
  5. "net/http"
  6. "net/http/httptest"
  7. "testing"
  8. "time"
  9. "github.com/Wei-Shaw/sub2api/internal/service"
  10. "github.com/gin-gonic/gin"
  11. "github.com/stretchr/testify/require"
  12. )
  13. // 捕获 ListUsers 入参、返回一个已删用户的 admin service 桩。
  14. type searchUsersAdminStub struct {
  15. service.AdminService
  16. gotFilters service.UserListFilters
  17. }
  18. func (s *searchUsersAdminStub) ListUsers(ctx context.Context, page, pageSize int, filters service.UserListFilters, sortBy, sortOrder string) ([]service.User, int64, error) {
  19. s.gotFilters = filters
  20. ts := time.Date(2026, 5, 28, 0, 0, 0, 0, time.UTC)
  21. return []service.User{
  22. {ID: 1, Email: "active@test.com"},
  23. {ID: 2, Email: "deleted@test.com", DeletedAt: &ts},
  24. }, 2, nil
  25. }
  26. func TestAdminUsageSearchUsers_IncludesDeletedAndFlags(t *testing.T) {
  27. gin.SetMode(gin.TestMode)
  28. stub := &searchUsersAdminStub{}
  29. handler := NewUsageHandler(nil, nil, stub, nil)
  30. router := gin.New()
  31. router.GET("/admin/usage/search-users", handler.SearchUsers)
  32. req := httptest.NewRequest(http.MethodGet, "/admin/usage/search-users?q=test", nil)
  33. rec := httptest.NewRecorder()
  34. router.ServeHTTP(rec, req)
  35. require.Equal(t, http.StatusOK, rec.Code)
  36. require.True(t, stub.gotFilters.IncludeDeleted, "SearchUsers 必须请求 IncludeDeleted")
  37. var resp struct {
  38. Data []struct {
  39. ID int64 `json:"id"`
  40. Email string `json:"email"`
  41. Deleted bool `json:"deleted"`
  42. } `json:"data"`
  43. }
  44. require.NoError(t, json.Unmarshal(rec.Body.Bytes(), &resp))
  45. require.Len(t, resp.Data, 2)
  46. require.False(t, resp.Data[0].Deleted)
  47. require.True(t, resp.Data[1].Deleted, "已删用户必须标记 deleted=true")
  48. }