mirror of https://github.com/muety/wakapi.git
95 lines
2.6 KiB
Go
95 lines
2.6 KiB
Go
package mocks
|
|
|
|
import (
|
|
"github.com/muety/wakapi/models"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
type UserServiceMock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *UserServiceMock) GetUserById(s string) (*models.User, error) {
|
|
args := m.Called(s)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GetUserByKey(s string) (*models.User, error) {
|
|
args := m.Called(s)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GetUserByEmail(s string) (*models.User, error) {
|
|
args := m.Called(s)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GetUserByResetToken(s string) (*models.User, error) {
|
|
args := m.Called(s)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GetAll() ([]*models.User, error) {
|
|
args := m.Called()
|
|
return args.Get(0).([]*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GetAllByReports(b bool) ([]*models.User, error) {
|
|
args := m.Called(b)
|
|
return args.Get(0).([]*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GetActive(b bool) ([]*models.User, error) {
|
|
args := m.Called(b)
|
|
return args.Get(0).([]*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) Count() (int64, error) {
|
|
args := m.Called()
|
|
return int64(args.Int(0)), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) CreateOrGet(signup *models.Signup, isAdmin bool) (*models.User, bool, error) {
|
|
args := m.Called(signup, isAdmin)
|
|
return args.Get(0).(*models.User), args.Bool(1), args.Error(2)
|
|
}
|
|
|
|
func (m *UserServiceMock) Update(user *models.User) (*models.User, error) {
|
|
args := m.Called(user)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) Delete(user *models.User) error {
|
|
args := m.Called(user)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *UserServiceMock) ResetApiKey(user *models.User) (*models.User, error) {
|
|
args := m.Called(user)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) ToggleBadges(user *models.User) (*models.User, error) {
|
|
args := m.Called(user)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) SetWakatimeApiCredentials(user *models.User, s1, s2 string) (*models.User, error) {
|
|
args := m.Called(user, s1, s2)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) MigrateMd5Password(user *models.User, login *models.Login) (*models.User, error) {
|
|
args := m.Called(user, login)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) GenerateResetToken(user *models.User) (*models.User, error) {
|
|
args := m.Called(user)
|
|
return args.Get(0).(*models.User), args.Error(1)
|
|
}
|
|
|
|
func (m *UserServiceMock) FlushCache() {
|
|
m.Called()
|
|
}
|