make user.Get all return []*model.User and *model.AppError (#11516)
run mock generate
Этот коммит содержится в:
коммит произвёл
Christopher Poile
родитель
d844c52f06
Коммит
8dfd3bab20
@@ -1519,11 +1519,10 @@ func (a *App) PermanentDeleteUser(user *model.User) *model.AppError {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (a *App) PermanentDeleteAllUsers() *model.AppError {
|
func (a *App) PermanentDeleteAllUsers() *model.AppError {
|
||||||
result := <-a.Srv.Store.User().GetAll()
|
users, err := a.Srv.Store.User().GetAll()
|
||||||
if result.Err != nil {
|
if err != nil {
|
||||||
return result.Err
|
return err
|
||||||
}
|
}
|
||||||
users := result.Data.([]*model.User)
|
|
||||||
for _, user := range users {
|
for _, user := range users {
|
||||||
a.PermanentDeleteUser(user)
|
a.PermanentDeleteUser(user)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -340,24 +340,19 @@ func (us SqlUserStore) Get(id string) (*model.User, *model.AppError) {
|
|||||||
return user, nil
|
return user, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (us SqlUserStore) GetAll() store.StoreChannel {
|
func (us SqlUserStore) GetAll() ([]*model.User, *model.AppError) {
|
||||||
return store.Do(func(result *store.StoreResult) {
|
query := us.usersQuery.OrderBy("Username ASC")
|
||||||
query := us.usersQuery.OrderBy("Username ASC")
|
|
||||||
|
|
||||||
queryString, args, err := query.ToSql()
|
queryString, args, err := query.ToSql()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
result.Err = model.NewAppError("SqlUserStore.GetAll", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError)
|
return nil, model.NewAppError("SqlUserStore.GetAll", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError)
|
||||||
return
|
}
|
||||||
}
|
|
||||||
|
|
||||||
var data []*model.User
|
var data []*model.User
|
||||||
if _, err := us.GetReplica().Select(&data, queryString, args...); err != nil {
|
if _, err := us.GetReplica().Select(&data, queryString, args...); err != nil {
|
||||||
result.Err = model.NewAppError("SqlUserStore.GetAll", "store.sql_user.get.app_error", nil, err.Error(), http.StatusInternalServerError)
|
return nil, model.NewAppError("SqlUserStore.GetAll", "store.sql_user.get.app_error", nil, err.Error(), http.StatusInternalServerError)
|
||||||
return
|
}
|
||||||
}
|
return data, nil
|
||||||
|
|
||||||
result.Data = data
|
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (us SqlUserStore) GetAllAfter(limit int, afterId string) ([]*model.User, *model.AppError) {
|
func (us SqlUserStore) GetAllAfter(limit int, afterId string) ([]*model.User, *model.AppError) {
|
||||||
|
|||||||
@@ -259,7 +259,7 @@ type UserStore interface {
|
|||||||
UpdateMfaSecret(userId, secret string) *model.AppError
|
UpdateMfaSecret(userId, secret string) *model.AppError
|
||||||
UpdateMfaActive(userId string, active bool) StoreChannel
|
UpdateMfaActive(userId string, active bool) StoreChannel
|
||||||
Get(id string) (*model.User, *model.AppError)
|
Get(id string) (*model.User, *model.AppError)
|
||||||
GetAll() StoreChannel
|
GetAll() ([]*model.User, *model.AppError)
|
||||||
ClearCaches()
|
ClearCaches()
|
||||||
InvalidateProfilesInChannelCacheByUser(userId string)
|
InvalidateProfilesInChannelCacheByUser(userId string)
|
||||||
InvalidateProfilesInChannelCache(channelId string)
|
InvalidateProfilesInChannelCache(channelId string)
|
||||||
|
|||||||
@@ -152,19 +152,28 @@ func (_m *UserStore) Get(id string) (*model.User, *model.AppError) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// GetAll provides a mock function with given fields:
|
// GetAll provides a mock function with given fields:
|
||||||
func (_m *UserStore) GetAll() store.StoreChannel {
|
func (_m *UserStore) GetAll() ([]*model.User, *model.AppError) {
|
||||||
ret := _m.Called()
|
ret := _m.Called()
|
||||||
|
|
||||||
var r0 store.StoreChannel
|
var r0 []*model.User
|
||||||
if rf, ok := ret.Get(0).(func() store.StoreChannel); ok {
|
if rf, ok := ret.Get(0).(func() []*model.User); ok {
|
||||||
r0 = rf()
|
r0 = rf()
|
||||||
} else {
|
} else {
|
||||||
if ret.Get(0) != nil {
|
if ret.Get(0) != nil {
|
||||||
r0 = ret.Get(0).(store.StoreChannel)
|
r0 = ret.Get(0).([]*model.User)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return r0
|
var r1 *model.AppError
|
||||||
|
if rf, ok := ret.Get(1).(func() *model.AppError); ok {
|
||||||
|
r1 = rf()
|
||||||
|
} else {
|
||||||
|
if ret.Get(1) != nil {
|
||||||
|
r1 = ret.Get(1).(*model.AppError)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return r0, r1
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetAllAfter provides a mock function with given fields: limit, afterId
|
// GetAllAfter provides a mock function with given fields: limit, afterId
|
||||||
|
|||||||
@@ -16,9 +16,8 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func TestUserStore(t *testing.T, ss store.Store) {
|
func TestUserStore(t *testing.T, ss store.Store) {
|
||||||
result := <-ss.User().GetAll()
|
users, err := ss.User().GetAll()
|
||||||
require.Nil(t, result.Err, "failed cleaning up test users")
|
require.Nil(t, err, "failed cleaning up test users")
|
||||||
users := result.Data.([]*model.User)
|
|
||||||
|
|
||||||
for _, u := range users {
|
for _, u := range users {
|
||||||
err := ss.User().PermanentDelete(u.Id)
|
err := ss.User().PermanentDelete(u.Id)
|
||||||
@@ -441,10 +440,9 @@ func testUserStoreGetAllProfiles(t *testing.T, ss store.Store) {
|
|||||||
})
|
})
|
||||||
|
|
||||||
t.Run("get all", func(t *testing.T) {
|
t.Run("get all", func(t *testing.T) {
|
||||||
result := <-ss.User().GetAll()
|
actual, err := ss.User().GetAll()
|
||||||
require.Nil(t, result.Err)
|
require.Nil(t, err)
|
||||||
|
|
||||||
actual := result.Data.([]*model.User)
|
|
||||||
require.Equal(t, []*model.User{
|
require.Equal(t, []*model.User{
|
||||||
u1,
|
u1,
|
||||||
u2,
|
u2,
|
||||||
|
|||||||
Ссылка в новой задаче
Block a user