feat: add GetByUsernamePrefixFn field and method to UserRepositoryStub
This commit is contained in:
@@ -153,6 +153,7 @@ type UserRepositoryStub struct {
|
|||||||
UnlockFn func(uint) error
|
UnlockFn func(uint) error
|
||||||
GetPostsFn func(uint, int, int) ([]database.Post, error)
|
GetPostsFn func(uint, int, int) ([]database.Post, error)
|
||||||
GetDeletedUsersFn func() ([]database.User, error)
|
GetDeletedUsersFn func() ([]database.User, error)
|
||||||
|
GetByUsernamePrefixFn func(string) (*database.User, error)
|
||||||
HardDeleteAllFn func() (int64, error)
|
HardDeleteAllFn func() (int64, error)
|
||||||
CountFn func() (int64, error)
|
CountFn func() (int64, error)
|
||||||
WithTxFn func(*gorm.DB) repositories.UserRepository
|
WithTxFn func(*gorm.DB) repositories.UserRepository
|
||||||
@@ -281,6 +282,13 @@ func (s *UserRepositoryStub) GetDeletedUsers() ([]database.User, error) {
|
|||||||
return nil, nil
|
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) {
|
func (s *UserRepositoryStub) HardDeleteAll() (int64, error) {
|
||||||
if s != nil && s.HardDeleteAllFn != nil {
|
if s != nil && s.HardDeleteAllFn != nil {
|
||||||
return s.HardDeleteAllFn()
|
return s.HardDeleteAllFn()
|
||||||
|
|||||||
Reference in New Issue
Block a user