diff --git a/internal/testutils/stubs.go b/internal/testutils/stubs.go index 89a022b..48cfae3 100644 --- a/internal/testutils/stubs.go +++ b/internal/testutils/stubs.go @@ -153,6 +153,7 @@ type UserRepositoryStub struct { UnlockFn func(uint) error GetPostsFn func(uint, int, int) ([]database.Post, error) GetDeletedUsersFn func() ([]database.User, error) + GetByUsernamePrefixFn func(string) (*database.User, error) HardDeleteAllFn func() (int64, error) CountFn func() (int64, error) WithTxFn func(*gorm.DB) repositories.UserRepository @@ -281,6 +282,13 @@ func (s *UserRepositoryStub) GetDeletedUsers() ([]database.User, error) { return nil, nil } +func (s *UserRepositoryStub) GetByUsernamePrefix(prefix string) (*database.User, error) { + if s != nil && s.GetByUsernamePrefixFn != nil { + return s.GetByUsernamePrefixFn(prefix) + } + return nil, gorm.ErrRecordNotFound +} + func (s *UserRepositoryStub) HardDeleteAll() (int64, error) { if s != nil && s.HardDeleteAllFn != nil { return s.HardDeleteAllFn()