538 lines
18 KiB
Go
538 lines
18 KiB
Go
//go:build integration
|
|
|
|
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
dbent "github.com/Wei-Shaw/sub2api/ent"
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/pagination"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type UserRepoSuite struct {
|
|
suite.Suite
|
|
ctx context.Context
|
|
client *dbent.Client
|
|
repo *userRepository
|
|
}
|
|
|
|
func (s *UserRepoSuite) SetupTest() {
|
|
s.ctx = context.Background()
|
|
s.client = testEntClient(s.T())
|
|
s.repo = newUserRepositoryWithSQL(s.client, integrationDB)
|
|
|
|
// 清理测试数据，确保每个测试从干净状态开始
|
|
_, _ = integrationDB.ExecContext(s.ctx, "DELETE FROM user_subscriptions")
|
|
_, _ = integrationDB.ExecContext(s.ctx, "DELETE FROM user_allowed_groups")
|
|
_, _ = integrationDB.ExecContext(s.ctx, "DELETE FROM users")
|
|
}
|
|
|
|
func TestUserRepoSuite(t *testing.T) {
|
|
suite.Run(t, new(UserRepoSuite))
|
|
}
|
|
|
|
func (s *UserRepoSuite) mustCreateUser(u *service.User) *service.User {
|
|
s.T().Helper()
|
|
|
|
if u.Email == "" {
|
|
u.Email = "user-" + time.Now().Format(time.RFC3339Nano) + "@example.com"
|
|
}
|
|
if u.PasswordHash == "" {
|
|
u.PasswordHash = "test-password-hash"
|
|
}
|
|
if u.Role == "" {
|
|
u.Role = service.RoleUser
|
|
}
|
|
if u.Status == "" {
|
|
u.Status = service.StatusActive
|
|
}
|
|
if u.Concurrency == 0 {
|
|
u.Concurrency = 5
|
|
}
|
|
|
|
s.Require().NoError(s.repo.Create(s.ctx, u), "create user")
|
|
return u
|
|
}
|
|
|
|
func (s *UserRepoSuite) mustCreateGroup(name string) *service.Group {
|
|
s.T().Helper()
|
|
|
|
g, err := s.client.Group.Create().
|
|
SetName(name).
|
|
SetStatus(service.StatusActive).
|
|
Save(s.ctx)
|
|
s.Require().NoError(err, "create group")
|
|
return groupEntityToService(g)
|
|
}
|
|
|
|
func (s *UserRepoSuite) mustCreateSubscription(userID, groupID int64, mutate func(*dbent.UserSubscriptionCreate)) *dbent.UserSubscription {
|
|
s.T().Helper()
|
|
|
|
now := time.Now()
|
|
create := s.client.UserSubscription.Create().
|
|
SetUserID(userID).
|
|
SetGroupID(groupID).
|
|
SetStartsAt(now.Add(-1 * time.Hour)).
|
|
SetExpiresAt(now.Add(24 * time.Hour)).
|
|
SetStatus(service.SubscriptionStatusActive).
|
|
SetAssignedAt(now).
|
|
SetNotes("")
|
|
|
|
if mutate != nil {
|
|
mutate(create)
|
|
}
|
|
|
|
sub, err := create.Save(s.ctx)
|
|
s.Require().NoError(err, "create subscription")
|
|
return sub
|
|
}
|
|
|
|
// --- Create / GetByID / GetByEmail / Update / Delete ---
|
|
|
|
func (s *UserRepoSuite) TestCreate() {
|
|
user := s.mustCreateUser(&service.User{
|
|
Email: "create@test.com",
|
|
Username: "testuser",
|
|
PasswordHash: "test-password-hash",
|
|
Role: service.RoleUser,
|
|
Status: service.StatusActive,
|
|
})
|
|
|
|
s.Require().NotZero(user.ID, "expected ID to be set")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err, "GetByID")
|
|
s.Require().Equal("create@test.com", got.Email)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestGetByID_NotFound() {
|
|
_, err := s.repo.GetByID(s.ctx, 999999)
|
|
s.Require().Error(err, "expected error for non-existent ID")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestGetByEmail() {
|
|
user := s.mustCreateUser(&service.User{Email: "byemail@test.com"})
|
|
|
|
got, err := s.repo.GetByEmail(s.ctx, user.Email)
|
|
s.Require().NoError(err, "GetByEmail")
|
|
s.Require().Equal(user.ID, got.ID)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestGetByEmail_NotFound() {
|
|
_, err := s.repo.GetByEmail(s.ctx, "nonexistent@test.com")
|
|
s.Require().Error(err, "expected error for non-existent email")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestUpdate() {
|
|
user := s.mustCreateUser(&service.User{Email: "update@test.com", Username: "original"})
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
got.Username = "updated"
|
|
s.Require().NoError(s.repo.Update(s.ctx, got), "Update")
|
|
|
|
updated, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err, "GetByID after update")
|
|
s.Require().Equal("updated", updated.Username)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestDelete() {
|
|
user := s.mustCreateUser(&service.User{Email: "delete@test.com"})
|
|
|
|
err := s.repo.Delete(s.ctx, user.ID)
|
|
s.Require().NoError(err, "Delete")
|
|
|
|
_, err = s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().Error(err, "expected error after delete")
|
|
}
|
|
|
|
// --- List / ListWithFilters ---
|
|
|
|
func (s *UserRepoSuite) TestList() {
|
|
s.mustCreateUser(&service.User{Email: "list1@test.com"})
|
|
s.mustCreateUser(&service.User{Email: "list2@test.com"})
|
|
|
|
users, page, err := s.repo.List(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10})
|
|
s.Require().NoError(err, "List")
|
|
s.Require().Len(users, 2)
|
|
s.Require().Equal(int64(2), page.Total)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestListWithFilters_Status() {
|
|
s.mustCreateUser(&service.User{Email: "active@test.com", Status: service.StatusActive})
|
|
s.mustCreateUser(&service.User{Email: "disabled@test.com", Status: service.StatusDisabled})
|
|
|
|
users, _, err := s.repo.ListWithFilters(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10}, service.UserListFilters{Status: service.StatusActive})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(users, 1)
|
|
s.Require().Equal(service.StatusActive, users[0].Status)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestListWithFilters_Role() {
|
|
s.mustCreateUser(&service.User{Email: "user@test.com", Role: service.RoleUser})
|
|
s.mustCreateUser(&service.User{Email: "admin@test.com", Role: service.RoleAdmin})
|
|
|
|
users, _, err := s.repo.ListWithFilters(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10}, service.UserListFilters{Role: service.RoleAdmin})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(users, 1)
|
|
s.Require().Equal(service.RoleAdmin, users[0].Role)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestListWithFilters_Search() {
|
|
s.mustCreateUser(&service.User{Email: "alice@test.com", Username: "Alice"})
|
|
s.mustCreateUser(&service.User{Email: "bob@test.com", Username: "Bob"})
|
|
|
|
users, _, err := s.repo.ListWithFilters(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10}, service.UserListFilters{Search: "alice"})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(users, 1)
|
|
s.Require().Contains(users[0].Email, "alice")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestListWithFilters_SearchByUsername() {
|
|
s.mustCreateUser(&service.User{Email: "u1@test.com", Username: "JohnDoe"})
|
|
s.mustCreateUser(&service.User{Email: "u2@test.com", Username: "JaneSmith"})
|
|
|
|
users, _, err := s.repo.ListWithFilters(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10}, service.UserListFilters{Search: "john"})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(users, 1)
|
|
s.Require().Equal("JohnDoe", users[0].Username)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestListWithFilters_LoadsActiveSubscriptions() {
|
|
user := s.mustCreateUser(&service.User{Email: "sub@test.com", Status: service.StatusActive})
|
|
groupActive := s.mustCreateGroup("g-sub-active")
|
|
groupExpired := s.mustCreateGroup("g-sub-expired")
|
|
|
|
_ = s.mustCreateSubscription(user.ID, groupActive.ID, func(c *dbent.UserSubscriptionCreate) {
|
|
c.SetStatus(service.SubscriptionStatusActive)
|
|
c.SetExpiresAt(time.Now().Add(1 * time.Hour))
|
|
})
|
|
_ = s.mustCreateSubscription(user.ID, groupExpired.ID, func(c *dbent.UserSubscriptionCreate) {
|
|
c.SetStatus(service.SubscriptionStatusExpired)
|
|
c.SetExpiresAt(time.Now().Add(-1 * time.Hour))
|
|
})
|
|
|
|
users, _, err := s.repo.ListWithFilters(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10}, service.UserListFilters{Search: "sub@"})
|
|
s.Require().NoError(err, "ListWithFilters")
|
|
s.Require().Len(users, 1, "expected 1 user")
|
|
s.Require().Len(users[0].Subscriptions, 1, "expected 1 active subscription")
|
|
s.Require().NotNil(users[0].Subscriptions[0].Group, "expected subscription group preload")
|
|
s.Require().Equal(groupActive.ID, users[0].Subscriptions[0].Group.ID, "group ID mismatch")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestListWithFilters_CombinedFilters() {
|
|
s.mustCreateUser(&service.User{
|
|
Email: "a@example.com",
|
|
Username: "Alice",
|
|
Role: service.RoleUser,
|
|
Status: service.StatusActive,
|
|
Balance: 10,
|
|
})
|
|
target := s.mustCreateUser(&service.User{
|
|
Email: "b@example.com",
|
|
Username: "Bob",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusActive,
|
|
Balance: 1,
|
|
})
|
|
s.mustCreateUser(&service.User{
|
|
Email: "c@example.com",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusDisabled,
|
|
})
|
|
|
|
users, page, err := s.repo.ListWithFilters(s.ctx, pagination.PaginationParams{Page: 1, PageSize: 10}, service.UserListFilters{Status: service.StatusActive, Role: service.RoleAdmin, Search: "b@"})
|
|
s.Require().NoError(err, "ListWithFilters")
|
|
s.Require().Equal(int64(1), page.Total, "ListWithFilters total mismatch")
|
|
s.Require().Len(users, 1, "ListWithFilters len mismatch")
|
|
s.Require().Equal(target.ID, users[0].ID, "ListWithFilters result mismatch")
|
|
}
|
|
|
|
// --- Balance operations ---
|
|
|
|
func (s *UserRepoSuite) TestUpdateBalance() {
|
|
user := s.mustCreateUser(&service.User{Email: "bal@test.com", Balance: 10})
|
|
|
|
err := s.repo.UpdateBalance(s.ctx, user.ID, 2.5)
|
|
s.Require().NoError(err, "UpdateBalance")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().InDelta(12.5, got.Balance, 1e-6)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestUpdateBalance_Negative() {
|
|
user := s.mustCreateUser(&service.User{Email: "balneg@test.com", Balance: 10})
|
|
|
|
err := s.repo.UpdateBalance(s.ctx, user.ID, -3)
|
|
s.Require().NoError(err, "UpdateBalance with negative")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().InDelta(7.0, got.Balance, 1e-6)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestDeductBalance() {
|
|
user := s.mustCreateUser(&service.User{Email: "deduct@test.com", Balance: 10})
|
|
|
|
err := s.repo.DeductBalance(s.ctx, user.ID, 5)
|
|
s.Require().NoError(err, "DeductBalance")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().InDelta(5.0, got.Balance, 1e-6)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestDeductBalance_InsufficientFunds() {
|
|
user := s.mustCreateUser(&service.User{Email: "insuf@test.com", Balance: 5})
|
|
|
|
// 透支策略：允许扣除超过余额的金额
|
|
err := s.repo.DeductBalance(s.ctx, user.ID, 999)
|
|
s.Require().NoError(err, "DeductBalance should allow overdraft")
|
|
|
|
// 验证余额变为负数
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().InDelta(-994.0, got.Balance, 1e-6, "Balance should be negative after overdraft")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestDeductBalance_ExactAmount() {
|
|
user := s.mustCreateUser(&service.User{Email: "exact@test.com", Balance: 10})
|
|
|
|
err := s.repo.DeductBalance(s.ctx, user.ID, 10)
|
|
s.Require().NoError(err, "DeductBalance exact amount")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().InDelta(0.0, got.Balance, 1e-6)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestDeductBalance_AllowsOverdraft() {
|
|
user := s.mustCreateUser(&service.User{Email: "overdraft@test.com", Balance: 5.0})
|
|
|
|
// 扣除超过余额的金额 - 应该成功
|
|
err := s.repo.DeductBalance(s.ctx, user.ID, 10.0)
|
|
s.Require().NoError(err, "DeductBalance should allow overdraft")
|
|
|
|
// 验证余额为负
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().InDelta(-5.0, got.Balance, 1e-6, "Balance should be -5.0 after overdraft")
|
|
}
|
|
|
|
// --- Concurrency ---
|
|
|
|
func (s *UserRepoSuite) TestUpdateConcurrency() {
|
|
user := s.mustCreateUser(&service.User{Email: "conc@test.com", Concurrency: 5})
|
|
|
|
err := s.repo.UpdateConcurrency(s.ctx, user.ID, 3)
|
|
s.Require().NoError(err, "UpdateConcurrency")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(8, got.Concurrency)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestUpdateConcurrency_Negative() {
|
|
user := s.mustCreateUser(&service.User{Email: "concneg@test.com", Concurrency: 5})
|
|
|
|
err := s.repo.UpdateConcurrency(s.ctx, user.ID, -2)
|
|
s.Require().NoError(err, "UpdateConcurrency negative")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user.ID)
|
|
s.Require().NoError(err)
|
|
s.Require().Equal(3, got.Concurrency)
|
|
}
|
|
|
|
// --- ExistsByEmail ---
|
|
|
|
func (s *UserRepoSuite) TestExistsByEmail() {
|
|
s.mustCreateUser(&service.User{Email: "exists@test.com"})
|
|
|
|
exists, err := s.repo.ExistsByEmail(s.ctx, "exists@test.com")
|
|
s.Require().NoError(err, "ExistsByEmail")
|
|
s.Require().True(exists)
|
|
|
|
notExists, err := s.repo.ExistsByEmail(s.ctx, "notexists@test.com")
|
|
s.Require().NoError(err)
|
|
s.Require().False(notExists)
|
|
}
|
|
|
|
// --- RemoveGroupFromAllowedGroups ---
|
|
|
|
func (s *UserRepoSuite) TestRemoveGroupFromAllowedGroups() {
|
|
target := s.mustCreateGroup("target-42")
|
|
other := s.mustCreateGroup("other-7")
|
|
|
|
userA := s.mustCreateUser(&service.User{
|
|
Email: "a1@example.com",
|
|
AllowedGroups: []int64{target.ID, other.ID},
|
|
})
|
|
s.mustCreateUser(&service.User{
|
|
Email: "a2@example.com",
|
|
AllowedGroups: []int64{other.ID},
|
|
})
|
|
|
|
affected, err := s.repo.RemoveGroupFromAllowedGroups(s.ctx, target.ID)
|
|
s.Require().NoError(err, "RemoveGroupFromAllowedGroups")
|
|
s.Require().Equal(int64(1), affected, "expected 1 affected row")
|
|
|
|
got, err := s.repo.GetByID(s.ctx, userA.ID)
|
|
s.Require().NoError(err, "GetByID")
|
|
s.Require().NotContains(got.AllowedGroups, target.ID)
|
|
s.Require().Contains(got.AllowedGroups, other.ID)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestRemoveGroupFromAllowedGroups_NoMatch() {
|
|
groupA := s.mustCreateGroup("nomatch-a")
|
|
groupB := s.mustCreateGroup("nomatch-b")
|
|
|
|
s.mustCreateUser(&service.User{
|
|
Email: "nomatch@test.com",
|
|
AllowedGroups: []int64{groupA.ID, groupB.ID},
|
|
})
|
|
|
|
affected, err := s.repo.RemoveGroupFromAllowedGroups(s.ctx, 999999)
|
|
s.Require().NoError(err)
|
|
s.Require().Zero(affected, "expected no affected rows")
|
|
}
|
|
|
|
// --- GetFirstAdmin ---
|
|
|
|
func (s *UserRepoSuite) TestGetFirstAdmin() {
|
|
admin1 := s.mustCreateUser(&service.User{
|
|
Email: "admin1@example.com",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusActive,
|
|
})
|
|
s.mustCreateUser(&service.User{
|
|
Email: "admin2@example.com",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusActive,
|
|
})
|
|
|
|
got, err := s.repo.GetFirstAdmin(s.ctx)
|
|
s.Require().NoError(err, "GetFirstAdmin")
|
|
s.Require().Equal(admin1.ID, got.ID, "GetFirstAdmin mismatch")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestGetFirstAdmin_NoAdmin() {
|
|
s.mustCreateUser(&service.User{
|
|
Email: "user@example.com",
|
|
Role: service.RoleUser,
|
|
Status: service.StatusActive,
|
|
})
|
|
|
|
_, err := s.repo.GetFirstAdmin(s.ctx)
|
|
s.Require().Error(err, "expected error when no admin exists")
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestGetFirstAdmin_DisabledAdminIgnored() {
|
|
s.mustCreateUser(&service.User{
|
|
Email: "disabled@example.com",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusDisabled,
|
|
})
|
|
activeAdmin := s.mustCreateUser(&service.User{
|
|
Email: "active@example.com",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusActive,
|
|
})
|
|
|
|
got, err := s.repo.GetFirstAdmin(s.ctx)
|
|
s.Require().NoError(err, "GetFirstAdmin")
|
|
s.Require().Equal(activeAdmin.ID, got.ID, "should return only active admin")
|
|
}
|
|
|
|
// --- Combined ---
|
|
|
|
func (s *UserRepoSuite) TestCRUD_And_Filters_And_AtomicUpdates() {
|
|
user1 := s.mustCreateUser(&service.User{
|
|
Email: "a@example.com",
|
|
Username: "Alice",
|
|
Role: service.RoleUser,
|
|
Status: service.StatusActive,
|
|
Balance: 10,
|
|
})
|
|
user2 := s.mustCreateUser(&service.User{
|
|
Email: "b@example.com",
|
|
Username: "Bob",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusActive,
|
|
Balance: 1,
|
|
})
|
|
s.mustCreateUser(&service.User{
|
|
Email: "c@example.com",
|
|
Role: service.RoleAdmin,
|
|
Status: service.StatusDisabled,
|
|
})
|
|
|
|
got, err := s.repo.GetByID(s.ctx, user1.ID)
|
|
s.Require().NoError(err, "GetByID")
|
|
s.Require().Equal(user1.Email, got.Email, "GetByID email mismatch")
|
|
|
|
gotByEmail, err := s.repo.GetByEmail(s.ctx, user2.Email)
|
|
s.Require().NoError(err, "GetByEmail")
|
|
s.Require().Equal(user2.ID, gotByEmail.ID, "GetByEmail ID mismatch")
|
|
|
|
got.Username = "Alice2"
|
|
s.Require().NoError(s.repo.Update(s.ctx, got), "Update")
|
|
got2, err := s.repo.GetByID(s.ctx, user1.ID)
|
|
s.Require().NoError(err, "GetByID after update")
|
|
s.Require().Equal("Alice2", got2.Username, "Update did not persist")
|
|
|
|
s.Require().NoError(s.repo.UpdateBalance(s.ctx, user1.ID, 2.5), "UpdateBalance")
|
|
got3, err := s.repo.GetByID(s.ctx, user1.ID)
|
|
s.Require().NoError(err, "GetByID after UpdateBalance")
|
|
s.Require().InDelta(12.5, got3.Balance, 1e-6)
|
|
|
|
s.Require().NoError(s.repo.DeductBalance(s.ctx, user1.ID, 5), "DeductBalance")
|
|
got4, err := s.repo.GetByID(s.ctx, user1.ID)
|
|
s.Require().NoError(err, "GetByID after DeductBalance")
|
|
s.Require().InDelta(7.5, got4.Balance, 1e-6)
|
|
|
|
// 透支策略：允许扣除超过余额的金额
|
|
err = s.repo.DeductBalance(s.ctx, user1.ID, 999)
|
|
s.Require().NoError(err, "DeductBalance should allow overdraft")
|
|
gotOverdraft, err := s.repo.GetByID(s.ctx, user1.ID)
|
|
s.Require().NoError(err, "GetByID after overdraft")
|
|
s.Require().Less(gotOverdraft.Balance, 0.0, "Balance should be negative after overdraft")
|
|
|
|
s.Require().NoError(s.repo.UpdateConcurrency(s.ctx, user1.ID, 3), "UpdateConcurrency")
|
|
got5, err := s.repo.GetByID(s.ctx, user1.ID)
|
|
s.Require().NoError(err, "GetByID after UpdateConcurrency")
|
|
s.Require().Equal(user1.Concurrency+3, got5.Concurrency)
|
|
|
|
params := pagination.PaginationParams{Page: 1, PageSize: 10}
|
|
users, page, err := s.repo.ListWithFilters(s.ctx, params, service.UserListFilters{Status: service.StatusActive, Role: service.RoleAdmin, Search: "b@"})
|
|
s.Require().NoError(err, "ListWithFilters")
|
|
s.Require().Equal(int64(1), page.Total, "ListWithFilters total mismatch")
|
|
s.Require().Len(users, 1, "ListWithFilters len mismatch")
|
|
s.Require().Equal(user2.ID, users[0].ID, "ListWithFilters result mismatch")
|
|
}
|
|
|
|
// --- UpdateBalance/UpdateConcurrency 影响行数校验测试 ---
|
|
|
|
func (s *UserRepoSuite) TestUpdateBalance_NotFound() {
|
|
err := s.repo.UpdateBalance(s.ctx, 999999, 10.0)
|
|
s.Require().Error(err, "expected error for non-existent user")
|
|
s.Require().ErrorIs(err, service.ErrUserNotFound)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestUpdateConcurrency_NotFound() {
|
|
err := s.repo.UpdateConcurrency(s.ctx, 999999, 5)
|
|
s.Require().Error(err, "expected error for non-existent user")
|
|
s.Require().ErrorIs(err, service.ErrUserNotFound)
|
|
}
|
|
|
|
func (s *UserRepoSuite) TestDeductBalance_NotFound() {
|
|
err := s.repo.DeductBalance(s.ctx, 999999, 5)
|
|
s.Require().Error(err, "expected error for non-existent user")
|
|
// DeductBalance 在用户不存在时返回 ErrUserNotFound
|
|
s.Require().ErrorIs(err, service.ErrUserNotFound)
|
|
}
|