Initial migration of the store to be sync (#10592)

* Migrating audit store

* Final migration example for the audit store

* async example

* Ending migration

* Removing Async helper

* Fixing tests

* Fixing govet problems with the StoreResult instanstiation
Этот коммит содержится в:
Jesús Espino
2019-04-15 22:53:52 +02:00
коммит произвёл GitHub
родитель 957ec1388b
Коммит 12c50eb830
30 изменённых файлов: 597 добавлений и 583 удалений

Просмотреть файл

@@ -733,8 +733,7 @@ func (me *TestHelper) cleanupTestFile(info *model.FileInfo) error {
func (me *TestHelper) MakeUserChannelAdmin(user *model.User, channel *model.Channel) { func (me *TestHelper) MakeUserChannelAdmin(user *model.User, channel *model.Channel) {
utils.DisableDebugLogForTest() utils.DisableDebugLogForTest()
if cmr := <-me.App.Srv.Store.Channel().GetMember(channel.Id, user.Id); cmr.Err == nil { if cm, err := me.App.Srv.Store.Channel().GetMember(channel.Id, user.Id); err == nil {
cm := cmr.Data.(*model.ChannelMember)
cm.SchemeAdmin = true cm.SchemeAdmin = true
if sr := <-me.App.Srv.Store.Channel().UpdateMember(cm); sr.Err != nil { if sr := <-me.App.Srv.Store.Channel().UpdateMember(cm); sr.Err != nil {
utils.EnableDebugLogForTest() utils.EnableDebugLogForTest()
@@ -742,7 +741,7 @@ func (me *TestHelper) MakeUserChannelAdmin(user *model.User, channel *model.Chan
} }
} else { } else {
utils.EnableDebugLogForTest() utils.EnableDebugLogForTest()
panic(cmr.Err) panic(err)
} }
utils.EnableDebugLogForTest() utils.EnableDebugLogForTest()

Просмотреть файл

@@ -8,17 +8,9 @@ import (
) )
func (a *App) GetAudits(userId string, limit int) (model.Audits, *model.AppError) { func (a *App) GetAudits(userId string, limit int) (model.Audits, *model.AppError) {
result := <-a.Srv.Store.Audit().Get(userId, 0, limit) return a.Srv.Store.Audit().Get(userId, 0, limit)
if result.Err != nil {
return nil, result.Err
}
return result.Data.(model.Audits), nil
} }
func (a *App) GetAuditsPage(userId string, page int, perPage int) (model.Audits, *model.AppError) { func (a *App) GetAuditsPage(userId string, page int, perPage int) (model.Audits, *model.AppError) {
result := <-a.Srv.Store.Audit().Get(userId, page*perPage, perPage) return a.Srv.Store.Audit().Get(userId, page*perPage, perPage)
if result.Err != nil {
return nil, result.Err
}
return result.Data.(model.Audits), nil
} }

Просмотреть файл

@@ -34,22 +34,21 @@ func (a *App) PatchBot(botUserId string, botPatch *model.BotPatch) (*model.Bot,
bot.Patch(botPatch) bot.Patch(botPatch)
result := <-a.Srv.Store.User().Get(botUserId) user, err := a.Srv.Store.User().Get(botUserId)
if result.Err != nil { if err != nil {
return nil, result.Err return nil, err
} }
user := result.Data.(*model.User)
patchedUser := model.UserFromBot(bot) patchedUser := model.UserFromBot(bot)
user.Id = patchedUser.Id user.Id = patchedUser.Id
user.Username = patchedUser.Username user.Username = patchedUser.Username
user.Email = patchedUser.Email user.Email = patchedUser.Email
user.FirstName = patchedUser.FirstName user.FirstName = patchedUser.FirstName
if result = <-a.Srv.Store.User().Update(user, true); result.Err != nil { if result := <-a.Srv.Store.User().Update(user, true); result.Err != nil {
return nil, result.Err return nil, result.Err
} }
result = <-a.Srv.Store.Bot().Update(bot) result := <-a.Srv.Store.Bot().Update(bot)
if result.Err != nil { if result.Err != nil {
return nil, result.Err return nil, result.Err
} }
@@ -79,17 +78,16 @@ func (a *App) GetBots(options *model.BotGetOptions) (model.BotList, *model.AppEr
// UpdateBotActive marks a bot as active or inactive, along with its corresponding user. // UpdateBotActive marks a bot as active or inactive, along with its corresponding user.
func (a *App) UpdateBotActive(botUserId string, active bool) (*model.Bot, *model.AppError) { func (a *App) UpdateBotActive(botUserId string, active bool) (*model.Bot, *model.AppError) {
result := <-a.Srv.Store.User().Get(botUserId) user, err := a.Srv.Store.User().Get(botUserId)
if result.Err != nil { if err != nil {
return nil, result.Err return nil, err
} }
user := result.Data.(*model.User)
if _, err := a.UpdateActive(user, active); err != nil { if _, err := a.UpdateActive(user, active); err != nil {
return nil, err return nil, err
} }
result = <-a.Srv.Store.Bot().Get(botUserId, true) result := <-a.Srv.Store.Bot().Get(botUserId, true)
if result.Err != nil { if result.Err != nil {
return nil, result.Err return nil, result.Err
} }

Просмотреть файл

@@ -36,11 +36,11 @@ func (a *App) CreateDefaultChannels(teamId string) ([]*model.Channel, *model.App
func (a *App) JoinDefaultChannels(teamId string, user *model.User, shouldBeAdmin bool, userRequestorId string) *model.AppError { func (a *App) JoinDefaultChannels(teamId string, user *model.User, shouldBeAdmin bool, userRequestorId string) *model.AppError {
var requestor *model.User var requestor *model.User
if userRequestorId != "" { if userRequestorId != "" {
u := <-a.Srv.Store.User().Get(userRequestorId) var err *model.AppError
if u.Err != nil { requestor, err = a.Srv.Store.User().Get(userRequestorId)
return u.Err if err != nil {
return err
} }
requestor = u.Data.(*model.User)
} }
defaultChannelList := []string{"town-square"} defaultChannelList := []string{"town-square"}
@@ -312,8 +312,18 @@ func (a *App) GetOrCreateDirectChannel(userId, otherUserId string) (*model.Chann
} }
func (a *App) createDirectChannel(userId string, otherUserId string) (*model.Channel, *model.AppError) { func (a *App) createDirectChannel(userId string, otherUserId string) (*model.Channel, *model.AppError) {
uc1 := a.Srv.Store.User().Get(userId) uc1 := make(chan store.StoreResult, 1)
uc2 := a.Srv.Store.User().Get(otherUserId) uc2 := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
uc1 <- store.StoreResult{Data: user, Err: err}
close(uc1)
}()
go func() {
user, err := a.Srv.Store.User().Get(otherUserId)
uc2 <- store.StoreResult{Data: user, Err: err}
close(uc2)
}()
if result := <-uc1; result.Err != nil { if result := <-uc1; result.Err != nil {
return nil, model.NewAppError("CreateDirectChannel", "api.channel.create_direct_channel.invalid_user.app_error", nil, userId, http.StatusBadRequest) return nil, model.NewAppError("CreateDirectChannel", "api.channel.create_direct_channel.invalid_user.app_error", nil, userId, http.StatusBadRequest)
@@ -354,15 +364,15 @@ func (a *App) WaitForChannelMembership(channelId string, userId string) {
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
result := <-a.Srv.Store.Channel().GetMember(channelId, userId) _, err := a.Srv.Store.Channel().GetMember(channelId, userId)
// If the membership was found then return // If the membership was found then return
if result.Err == nil { if err == nil {
return return
} }
// If we received a error but it wasn't a missing channel member then return // If we received a error but it wasn't a missing channel member then return
if result.Err.Id != store.MISSING_CHANNEL_MEMBER_ERROR { if err.Id != store.MISSING_CHANNEL_MEMBER_ERROR {
return return
} }
} }
@@ -761,12 +771,11 @@ func (a *App) DeleteChannel(channel *model.Channel, userId string) *model.AppErr
var user *model.User var user *model.User
if userId != "" { if userId != "" {
uc := a.Srv.Store.User().Get(userId) var err *model.AppError
uresult := <-uc user, err = a.Srv.Store.User().Get(userId)
if uresult.Err != nil { if err != nil {
return uresult.Err return err
} }
user = uresult.Data.(*model.User)
} }
ihcresult := <-ihc ihcresult := <-ihc
@@ -844,14 +853,12 @@ func (a *App) addUserToChannel(user *model.User, channel *model.Channel, teamMem
return nil, model.NewAppError("AddUserToChannel", "api.channel.add_user_to_channel.type.app_error", nil, "", http.StatusBadRequest) return nil, model.NewAppError("AddUserToChannel", "api.channel.add_user_to_channel.type.app_error", nil, "", http.StatusBadRequest)
} }
cmchan := a.Srv.Store.Channel().GetMember(channel.Id, user.Id) channelMember, err := a.Srv.Store.Channel().GetMember(channel.Id, user.Id)
if err != nil {
if result := <-cmchan; result.Err != nil { if err.Id != store.MISSING_CHANNEL_MEMBER_ERROR {
if result.Err.Id != store.MISSING_CHANNEL_MEMBER_ERROR { return nil, err
return nil, result.Err
} }
} else { } else {
channelMember := result.Data.(*model.ChannelMember)
return channelMember, nil return channelMember, nil
} }
@@ -904,12 +911,12 @@ func (a *App) AddUserToChannel(user *model.User, channel *model.Channel) (*model
} }
func (a *App) AddChannelMember(userId string, channel *model.Channel, userRequestorId string, postRootId string, currentSessionId string) (*model.ChannelMember, *model.AppError) { func (a *App) AddChannelMember(userId string, channel *model.Channel, userRequestorId string, postRootId string, currentSessionId string) (*model.ChannelMember, *model.AppError) {
if result := <-a.Srv.Store.Channel().GetMember(channel.Id, userId); result.Err != nil { if member, err := a.Srv.Store.Channel().GetMember(channel.Id, userId); err != nil {
if result.Err.Id != store.MISSING_CHANNEL_MEMBER_ERROR { if err.Id != store.MISSING_CHANNEL_MEMBER_ERROR {
return nil, result.Err return nil, err
} }
} else { } else {
return result.Data.(*model.ChannelMember), nil return member, nil
} }
var user *model.User var user *model.User
@@ -1003,15 +1010,11 @@ func (a *App) AddDirectChannels(teamId string, user *model.User) *model.AppError
} }
func (a *App) PostUpdateChannelHeaderMessage(userId string, channel *model.Channel, oldChannelHeader, newChannelHeader string) *model.AppError { func (a *App) PostUpdateChannelHeaderMessage(userId string, channel *model.Channel, oldChannelHeader, newChannelHeader string) *model.AppError {
uc := a.Srv.Store.User().Get(userId) user, err := a.Srv.Store.User().Get(userId)
if err != nil {
uresult := <-uc return model.NewAppError("PostUpdateChannelHeaderMessage", "api.channel.post_update_channel_header_message_and_forget.retrieve_user.error", nil, err.Error(), http.StatusBadRequest)
if uresult.Err != nil {
return model.NewAppError("PostUpdateChannelHeaderMessage", "api.channel.post_update_channel_header_message_and_forget.retrieve_user.error", nil, uresult.Err.Error(), http.StatusBadRequest)
} }
user := uresult.Data.(*model.User)
var message string var message string
if oldChannelHeader == "" { if oldChannelHeader == "" {
message = fmt.Sprintf(utils.T("api.channel.post_update_channel_header_message_and_forget.updated_to"), user.Username, newChannelHeader) message = fmt.Sprintf(utils.T("api.channel.post_update_channel_header_message_and_forget.updated_to"), user.Username, newChannelHeader)
@@ -1041,15 +1044,11 @@ func (a *App) PostUpdateChannelHeaderMessage(userId string, channel *model.Chann
} }
func (a *App) PostUpdateChannelPurposeMessage(userId string, channel *model.Channel, oldChannelPurpose string, newChannelPurpose string) *model.AppError { func (a *App) PostUpdateChannelPurposeMessage(userId string, channel *model.Channel, oldChannelPurpose string, newChannelPurpose string) *model.AppError {
uc := a.Srv.Store.User().Get(userId) user, err := a.Srv.Store.User().Get(userId)
if err != nil {
uresult := <-uc return model.NewAppError("PostUpdateChannelPurposeMessage", "app.channel.post_update_channel_purpose_message.retrieve_user.error", nil, err.Error(), http.StatusBadRequest)
if uresult.Err != nil {
return model.NewAppError("PostUpdateChannelPurposeMessage", "app.channel.post_update_channel_purpose_message.retrieve_user.error", nil, uresult.Err.Error(), http.StatusBadRequest)
} }
user := uresult.Data.(*model.User)
var message string var message string
if oldChannelPurpose == "" { if oldChannelPurpose == "" {
message = fmt.Sprintf(utils.T("app.channel.post_update_channel_purpose_message.updated_to"), user.Username, newChannelPurpose) message = fmt.Sprintf(utils.T("app.channel.post_update_channel_purpose_message.updated_to"), user.Username, newChannelPurpose)
@@ -1078,15 +1077,11 @@ func (a *App) PostUpdateChannelPurposeMessage(userId string, channel *model.Chan
} }
func (a *App) PostUpdateChannelDisplayNameMessage(userId string, channel *model.Channel, oldChannelDisplayName, newChannelDisplayName string) *model.AppError { func (a *App) PostUpdateChannelDisplayNameMessage(userId string, channel *model.Channel, oldChannelDisplayName, newChannelDisplayName string) *model.AppError {
uc := a.Srv.Store.User().Get(userId) user, err := a.Srv.Store.User().Get(userId)
if err != nil {
uresult := <-uc return model.NewAppError("PostUpdateChannelDisplayNameMessage", "api.channel.post_update_channel_displayname_message_and_forget.retrieve_user.error", nil, err.Error(), http.StatusBadRequest)
if uresult.Err != nil {
return model.NewAppError("PostUpdateChannelDisplayNameMessage", "api.channel.post_update_channel_displayname_message_and_forget.retrieve_user.error", nil, uresult.Err.Error(), http.StatusBadRequest)
} }
user := uresult.Data.(*model.User)
message := fmt.Sprintf(utils.T("api.channel.post_update_channel_displayname_message_and_forget.updated_from"), user.Username, oldChannelDisplayName, newChannelDisplayName) message := fmt.Sprintf(utils.T("api.channel.post_update_channel_displayname_message_and_forget.updated_from"), user.Username, oldChannelDisplayName, newChannelDisplayName)
post := &model.Post{ post := &model.Post{
@@ -1234,11 +1229,7 @@ func (a *App) GetPublicChannelsForTeam(teamId string, offset int, limit int) (*m
} }
func (a *App) GetChannelMember(channelId string, userId string) (*model.ChannelMember, *model.AppError) { func (a *App) GetChannelMember(channelId string, userId string) (*model.ChannelMember, *model.AppError) {
result := <-a.Srv.Store.Channel().GetMember(channelId, userId) return a.Srv.Store.Channel().GetMember(channelId, userId)
if result.Err != nil {
return nil, result.Err
}
return result.Data.(*model.ChannelMember), nil
} }
func (a *App) GetChannelMembersPage(channelId string, page, perPage int) (*model.ChannelMembers, *model.AppError) { func (a *App) GetChannelMembersPage(channelId string, page, perPage int) (*model.ChannelMembers, *model.AppError) {
@@ -1330,8 +1321,18 @@ func (a *App) GetChannelUnread(channelId, userId string) (*model.ChannelUnread,
} }
func (a *App) JoinChannel(channel *model.Channel, userId string) *model.AppError { func (a *App) JoinChannel(channel *model.Channel, userId string) *model.AppError {
userChan := a.Srv.Store.User().Get(userId) userChan := make(chan store.StoreResult, 1)
memberChan := a.Srv.Store.Channel().GetMember(channel.Id, userId) memberChan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
userChan <- store.StoreResult{Data: user, Err: err}
close(userChan)
}()
go func() {
member, err := a.Srv.Store.Channel().GetMember(channel.Id, userId)
memberChan <- store.StoreResult{Data: member, Err: err}
close(memberChan)
}()
uresult := <-userChan uresult := <-userChan
if uresult.Err != nil { if uresult.Err != nil {
@@ -1419,7 +1420,12 @@ func (a *App) postJoinTeamMessage(user *model.User, channel *model.Channel) *mod
func (a *App) LeaveChannel(channelId string, userId string) *model.AppError { func (a *App) LeaveChannel(channelId string, userId string) *model.AppError {
sc := a.Srv.Store.Channel().Get(channelId, true) sc := a.Srv.Store.Channel().Get(channelId, true)
uc := a.Srv.Store.User().Get(userId) uc := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
uc <- store.StoreResult{Data: user, Err: err}
close(uc)
}()
ccm := a.Srv.Store.Channel().GetMemberCount(channelId, false) ccm := a.Srv.Store.Channel().GetMemberCount(channelId, false)
cresult := <-sc cresult := <-sc
@@ -1759,12 +1765,11 @@ func (a *App) MarkChannelsAsViewed(channelIds []string, userId string, currentSe
} }
channel := chanResult.Data.(*model.Channel) channel := chanResult.Data.(*model.Channel)
result := <-a.Srv.Store.Channel().GetMember(channelId, userId) member, err := a.Srv.Store.Channel().GetMember(channelId, userId)
if result.Err != nil { if err != nil {
mlog.Warn(fmt.Sprintf("Failed to get membership %v", result.Err)) mlog.Warn(fmt.Sprintf("Failed to get membership %v", err))
continue continue
} }
member := result.Data.(*model.ChannelMember)
notify := member.NotifyProps[model.PUSH_NOTIFY_PROP] notify := member.NotifyProps[model.PUSH_NOTIFY_PROP]
if notify == model.CHANNEL_NOTIFY_DEFAULT { if notify == model.CHANNEL_NOTIFY_DEFAULT {
@@ -1951,14 +1956,11 @@ func (a *App) GetPinnedPosts(channelId string) (*model.PostList, *model.AppError
} }
func (a *App) ToggleMuteChannel(channelId string, userId string) *model.ChannelMember { func (a *App) ToggleMuteChannel(channelId string, userId string) *model.ChannelMember {
result := <-a.Srv.Store.Channel().GetMember(channelId, userId) member, err := a.Srv.Store.Channel().GetMember(channelId, userId)
if err != nil {
if result.Err != nil {
return nil return nil
} }
member := result.Data.(*model.ChannelMember)
if member.NotifyProps[model.MARK_UNREAD_NOTIFY_PROP] == model.CHANNEL_NOTIFY_MENTION { if member.NotifyProps[model.MARK_UNREAD_NOTIFY_PROP] == model.CHANNEL_NOTIFY_MENTION {
member.NotifyProps[model.MARK_UNREAD_NOTIFY_PROP] = model.CHANNEL_MARK_UNREAD_ALL member.NotifyProps[model.MARK_UNREAD_NOTIFY_PROP] = model.CHANNEL_MARK_UNREAD_ALL
} else { } else {

Просмотреть файл

@@ -13,6 +13,7 @@ import (
"github.com/mattermost/mattermost-server/mlog" "github.com/mattermost/mattermost-server/mlog"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/store"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
goi18n "github.com/nicksnyder/go-i18n/i18n" goi18n "github.com/nicksnyder/go-i18n/i18n"
) )
@@ -219,7 +220,12 @@ func (a *App) tryExecuteCustomCommand(args *model.CommandArgs, trigger string, m
chanChan := a.Srv.Store.Channel().Get(args.ChannelId, true) chanChan := a.Srv.Store.Channel().Get(args.ChannelId, true)
teamChan := a.Srv.Store.Team().Get(args.TeamId) teamChan := a.Srv.Store.Team().Get(args.TeamId)
userChan := a.Srv.Store.User().Get(args.UserId) userChan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(args.UserId)
userChan <- store.StoreResult{Data: user, Err: err}
close(userChan)
}()
result := <-a.Srv.Store.Command().GetByTeam(args.TeamId) result := <-a.Srv.Store.Command().GetByTeam(args.TeamId)
if result.Err != nil { if result.Err != nil {

Просмотреть файл

@@ -199,26 +199,24 @@ func (job *EmailBatchingJob) checkPendingNotifications(now time.Time, handler fu
} }
func (s *Server) sendBatchedEmailNotification(userId string, notifications []*batchedNotification) { func (s *Server) sendBatchedEmailNotification(userId string, notifications []*batchedNotification) {
result := <-s.Store.User().Get(userId) user, err := s.Store.User().Get(userId)
if result.Err != nil { if err != nil {
mlog.Warn("Unable to find recipient for batched email notification") mlog.Warn("Unable to find recipient for batched email notification")
return return
} }
user := result.Data.(*model.User)
translateFunc := utils.GetUserTranslations(user.Locale) translateFunc := utils.GetUserTranslations(user.Locale)
displayNameFormat := *s.Config().TeamSettings.TeammateNameDisplay displayNameFormat := *s.Config().TeamSettings.TeammateNameDisplay
var contents string var contents string
for _, notification := range notifications { for _, notification := range notifications {
result := <-s.Store.User().Get(notification.post.UserId) sender, err := s.Store.User().Get(notification.post.UserId)
if result.Err != nil { if err != nil {
mlog.Warn("Unable to find sender of post for batched email notification") mlog.Warn("Unable to find sender of post for batched email notification")
continue continue
} }
sender := result.Data.(*model.User)
result = <-s.Store.Channel().Get(notification.post.ChannelId, true) result := <-s.Store.Channel().Get(notification.post.ChannelId, true)
if result.Err != nil { if result.Err != nil {
mlog.Warn("Unable to find channel of post for batched email notification") mlog.Warn("Unable to find channel of post for batched email notification")
continue continue

Просмотреть файл

@@ -10,6 +10,7 @@ import (
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/store" "github.com/mattermost/mattermost-server/store"
"github.com/stretchr/testify/require"
) )
func TestHandleNewNotifications(t *testing.T) { func TestHandleNewNotifications(t *testing.T) {
@@ -109,7 +110,8 @@ func TestCheckPendingNotifications(t *testing.T) {
}, },
} }
channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember) channelMember, err := th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)
require.Nil(t, err)
channelMember.LastViewedAt = 9999999 channelMember.LastViewedAt = 9999999
store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember)) store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember))
@@ -128,7 +130,8 @@ func TestCheckPendingNotifications(t *testing.T) {
} }
// test that notifications are cleared if the user has acted // test that notifications are cleared if the user has acted
channelMember = store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember) channelMember, err = th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)
require.Nil(t, err)
channelMember.LastViewedAt = 10001000 channelMember.LastViewedAt = 10001000
store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember)) store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember))
@@ -208,7 +211,8 @@ func TestCheckPendingNotificationsDefaultInterval(t *testing.T) {
job := NewEmailBatchingJob(th.Server, 128) job := NewEmailBatchingJob(th.Server, 128)
// bypasses recent user activity check // bypasses recent user activity check
channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember) channelMember, err := th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)
require.Nil(t, err)
channelMember.LastViewedAt = 9999000 channelMember.LastViewedAt = 9999000
store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember)) store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember))
@@ -246,7 +250,8 @@ func TestCheckPendingNotificationsCantParseInterval(t *testing.T) {
job := NewEmailBatchingJob(th.Server, 128) job := NewEmailBatchingJob(th.Server, 128)
// bypasses recent user activity check // bypasses recent user activity check
channelMember := store.Must(th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)).(*model.ChannelMember) channelMember, err := th.App.Srv.Store.Channel().GetMember(th.BasicChannel.Id, th.BasicUser.Id)
require.Nil(t, err)
channelMember.LastViewedAt = 9999000 channelMember.LastViewedAt = 9999000
store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember)) store.Must(th.App.Srv.Store.Channel().UpdateMember(channelMember))

Просмотреть файл

@@ -415,11 +415,10 @@ func (a *App) BuildPostReactions(postId string) (*[]ReactionImportData, *model.A
reactions := result.Data.([]*model.Reaction) reactions := result.Data.([]*model.Reaction)
for _, reaction := range reactions { for _, reaction := range reactions {
result := <-a.Srv.Store.User().Get(reaction.UserId) user, err := a.Srv.Store.User().Get(reaction.UserId)
if result.Err != nil { if err != nil {
return nil, result.Err return nil, err
} }
user := result.Data.(*model.User)
reactionsOfPost = append(reactionsOfPost, *ImportReactionFromPost(user, reaction)) reactionsOfPost = append(reactionsOfPost, *ImportReactionFromPost(user, reaction))
} }

Просмотреть файл

@@ -259,11 +259,11 @@ func (a *App) GetOAuthAccessTokenForCodeFlow(clientId, grantType, redirectUri, c
return nil, model.NewAppError("GetOAuthAccessToken", "api.oauth.get_access_token.redirect_uri.app_error", nil, "", http.StatusBadRequest) return nil, model.NewAppError("GetOAuthAccessToken", "api.oauth.get_access_token.redirect_uri.app_error", nil, "", http.StatusBadRequest)
} }
result = <-a.Srv.Store.User().Get(authData.UserId) var err *model.AppError
if result.Err != nil { user, err = a.Srv.Store.User().Get(authData.UserId)
if err != nil {
return nil, model.NewAppError("GetOAuthAccessToken", "api.oauth.get_access_token.internal_user.app_error", nil, "", http.StatusNotFound) return nil, model.NewAppError("GetOAuthAccessToken", "api.oauth.get_access_token.internal_user.app_error", nil, "", http.StatusNotFound)
} }
user = result.Data.(*model.User)
result = <-a.Srv.Store.OAuth().GetPreviousAccessData(user.Id, clientId) result = <-a.Srv.Store.OAuth().GetPreviousAccessData(user.Id, clientId)
if result.Err != nil { if result.Err != nil {
@@ -318,11 +318,10 @@ func (a *App) GetOAuthAccessTokenForCodeFlow(clientId, grantType, redirectUri, c
} }
accessData = result.Data.(*model.AccessData) accessData = result.Data.(*model.AccessData)
result = <-a.Srv.Store.User().Get(accessData.UserId) user, err := a.Srv.Store.User().Get(accessData.UserId)
if result.Err != nil { if err != nil {
return nil, model.NewAppError("GetOAuthAccessToken", "api.oauth.get_access_token.internal_user.app_error", nil, "", http.StatusNotFound) return nil, model.NewAppError("GetOAuthAccessToken", "api.oauth.get_access_token.internal_user.app_error", nil, "", http.StatusNotFound)
} }
user = result.Data.(*model.User)
access, err := a.newSessionUpdateToken(oauthApp.Name, accessData, user) access, err := a.newSessionUpdateToken(oauthApp.Name, accessData, user)
if err != nil { if err != nil {

Просмотреть файл

@@ -50,11 +50,10 @@ func (a *App) CreatePostAsUser(post *model.Post, currentSessionId string) (*mode
} }
if err.Id == "api.post.create_post.town_square_read_only" { if err.Id == "api.post.create_post.town_square_read_only" {
result := <-a.Srv.Store.User().Get(post.UserId) user, userErr := a.Srv.Store.User().Get(post.UserId)
if result.Err != nil { if userErr != nil {
return nil, result.Err return nil, userErr
} }
user := result.Data.(*model.User)
T := utils.GetUserTranslations(user.Locale) T := utils.GetUserTranslations(user.Locale)
a.SendEphemeralPost( a.SendEphemeralPost(
@@ -164,11 +163,10 @@ func (a *App) CreatePost(post *model.Post, channel *model.Channel, triggerWebhoo
pchan = a.Srv.Store.Post().Get(post.RootId) pchan = a.Srv.Store.Post().Get(post.RootId)
} }
result := <-a.Srv.Store.User().Get(post.UserId) user, err := a.Srv.Store.User().Get(post.UserId)
if result.Err != nil { if err != nil {
return nil, result.Err return nil, err
} }
user := result.Data.(*model.User)
if a.License() != nil && *a.Config().TeamSettings.ExperimentalTownSquareIsReadOnly && if a.License() != nil && *a.Config().TeamSettings.ExperimentalTownSquareIsReadOnly &&
!post.IsSystemMessage() && !post.IsSystemMessage() &&
@@ -180,7 +178,7 @@ func (a *App) CreatePost(post *model.Post, channel *model.Channel, triggerWebhoo
// Verify the parent/child relationships are correct // Verify the parent/child relationships are correct
var parentPostList *model.PostList var parentPostList *model.PostList
if pchan != nil { if pchan != nil {
result = <-pchan result := <-pchan
if result.Err != nil { if result.Err != nil {
return nil, model.NewAppError("createPost", "api.post.create_post.root_id.app_error", nil, "", http.StatusBadRequest) return nil, model.NewAppError("createPost", "api.post.create_post.root_id.app_error", nil, "", http.StatusBadRequest)
} }
@@ -245,7 +243,7 @@ func (a *App) CreatePost(post *model.Post, channel *model.Channel, triggerWebhoo
} }
} }
result = <-a.Srv.Store.Post().Save(post) result := <-a.Srv.Store.Post().Save(post)
if result.Err != nil { if result.Err != nil {
return nil, result.Err return nil, result.Err
} }

Просмотреть файл

@@ -9,6 +9,7 @@ import (
"github.com/mattermost/mattermost-server/mlog" "github.com/mattermost/mattermost-server/mlog"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/store"
) )
func (a *App) CreateSession(session *model.Session) (*model.Session, *model.AppError) { func (a *App) CreateSession(session *model.Session) (*model.Session, *model.AppError) {
@@ -247,7 +248,12 @@ func (a *App) CreateUserAccessToken(token *model.UserAccessToken) (*model.UserAc
token.Token = model.NewId() token.Token = model.NewId()
uchan := a.Srv.Store.User().Get(token.UserId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(token.UserId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
result := <-a.Srv.Store.UserAccessToken().Save(token) result := <-a.Srv.Store.UserAccessToken().Save(token)
if result.Err != nil { if result.Err != nil {
@@ -284,12 +290,10 @@ func (a *App) createSessionForUserAccessToken(tokenString string) (*model.Sessio
return nil, model.NewAppError("createSessionForUserAccessToken", "app.user_access_token.invalid_or_missing", nil, "inactive_token", http.StatusUnauthorized) return nil, model.NewAppError("createSessionForUserAccessToken", "app.user_access_token.invalid_or_missing", nil, "inactive_token", http.StatusUnauthorized)
} }
var user *model.User user, err := a.Srv.Store.User().Get(token.UserId)
result = <-a.Srv.Store.User().Get(token.UserId) if err != nil {
if result.Err != nil { return nil, err
return nil, result.Err
} }
user = result.Data.(*model.User)
if user.DeleteAt != 0 { if user.DeleteAt != 0 {
return nil, model.NewAppError("createSessionForUserAccessToken", "app.user_access_token.invalid_or_missing", nil, "inactive_user_id="+user.Id, http.StatusUnauthorized) return nil, model.NewAppError("createSessionForUserAccessToken", "app.user_access_token.invalid_or_missing", nil, "inactive_user_id="+user.Id, http.StatusUnauthorized)

Просмотреть файл

@@ -18,6 +18,7 @@ import (
"github.com/mattermost/mattermost-server/mlog" "github.com/mattermost/mattermost-server/mlog"
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/plugin" "github.com/mattermost/mattermost-server/plugin"
"github.com/mattermost/mattermost-server/store"
"github.com/mattermost/mattermost-server/utils" "github.com/mattermost/mattermost-server/utils"
) )
@@ -327,7 +328,12 @@ func (a *App) sendUpdatedMemberRoleEvent(userId string, member *model.TeamMember
func (a *App) AddUserToTeam(teamId string, userId string, userRequestorId string) (*model.Team, *model.AppError) { func (a *App) AddUserToTeam(teamId string, userId string, userRequestorId string) (*model.Team, *model.AppError) {
tchan := a.Srv.Store.Team().Get(teamId) tchan := a.Srv.Store.Team().Get(teamId)
uchan := a.Srv.Store.User().Get(userId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
result := <-tchan result := <-tchan
if result.Err != nil { if result.Err != nil {
@@ -375,7 +381,12 @@ func (a *App) AddUserToTeamByToken(userId string, tokenId string) (*model.Team,
tokenData := model.MapFromJson(strings.NewReader(token.Extra)) tokenData := model.MapFromJson(strings.NewReader(token.Extra))
tchan := a.Srv.Store.Team().Get(tokenData["teamId"]) tchan := a.Srv.Store.Team().Get(tokenData["teamId"])
uchan := a.Srv.Store.User().Get(userId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
result = <-tchan result = <-tchan
if result.Err != nil { if result.Err != nil {
@@ -402,7 +413,12 @@ func (a *App) AddUserToTeamByToken(userId string, tokenId string) (*model.Team,
func (a *App) AddUserToTeamByInviteId(inviteId string, userId string) (*model.Team, *model.AppError) { func (a *App) AddUserToTeamByInviteId(inviteId string, userId string) (*model.Team, *model.AppError) {
tchan := a.Srv.Store.Team().GetByInviteId(inviteId) tchan := a.Srv.Store.Team().GetByInviteId(inviteId)
uchan := a.Srv.Store.User().Get(userId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
result := <-tchan result := <-tchan
if result.Err != nil { if result.Err != nil {
@@ -761,7 +777,12 @@ func (a *App) GetTeamUnread(teamId, userId string) (*model.TeamUnread, *model.Ap
func (a *App) RemoveUserFromTeam(teamId string, userId string, requestorId string) *model.AppError { func (a *App) RemoveUserFromTeam(teamId string, userId string, requestorId string) *model.AppError {
tchan := a.Srv.Store.Team().Get(teamId) tchan := a.Srv.Store.Team().Get(teamId)
uchan := a.Srv.Store.User().Get(userId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(userId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
result := <-tchan result := <-tchan
if result.Err != nil { if result.Err != nil {
@@ -926,7 +947,12 @@ func (a *App) InviteNewUsersToTeam(emailList []string, teamId, senderId string)
} }
tchan := a.Srv.Store.Team().Get(teamId) tchan := a.Srv.Store.Team().Get(teamId)
uchan := a.Srv.Store.User().Get(senderId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(senderId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
result := <-tchan result := <-tchan
if result.Err != nil { if result.Err != nil {

Просмотреть файл

@@ -398,11 +398,7 @@ func (a *App) IsUsernameTaken(name string) bool {
} }
func (a *App) GetUser(userId string) (*model.User, *model.AppError) { func (a *App) GetUser(userId string) (*model.User, *model.AppError) {
result := <-a.Srv.Store.User().Get(userId) return a.Srv.Store.User().Get(userId)
if result.Err != nil {
return nil, result.Err
}
return result.Data.(*model.User), nil
} }
func (a *App) GetUserByUsername(username string) (*model.User, *model.AppError) { func (a *App) GetUserByUsername(username string) (*model.User, *model.AppError) {
@@ -655,11 +651,10 @@ func (a *App) GenerateMfaSecret(userId string) (*model.MfaSecret, *model.AppErro
} }
func (a *App) ActivateMfa(userId, token string) *model.AppError { func (a *App) ActivateMfa(userId, token string) *model.AppError {
result := <-a.Srv.Store.User().Get(userId) user, err := a.Srv.Store.User().Get(userId)
if result.Err != nil { if err != nil {
return result.Err return err
} }
user := result.Data.(*model.User)
if len(user.AuthService) > 0 && user.AuthService != model.USER_AUTH_SERVICE_LDAP { if len(user.AuthService) > 0 && user.AuthService != model.USER_AUTH_SERVICE_LDAP {
return model.NewAppError("ActivateMfa", "api.user.activate_mfa.email_and_ldap_only.app_error", nil, "", http.StatusBadRequest) return model.NewAppError("ActivateMfa", "api.user.activate_mfa.email_and_ldap_only.app_error", nil, "", http.StatusBadRequest)
@@ -1086,11 +1081,10 @@ func (a *App) sendUpdatedUserEvent(user model.User) {
} }
func (a *App) UpdateUser(user *model.User, sendNotifications bool) (*model.User, *model.AppError) { func (a *App) UpdateUser(user *model.User, sendNotifications bool) (*model.User, *model.AppError) {
result := <-a.Srv.Store.User().Get(user.Id) prev, err := a.Srv.Store.User().Get(user.Id)
if result.Err != nil { if err != nil {
return nil, result.Err return nil, err
} }
prev := result.Data.(*model.User)
if !CheckUserDomain(user, *a.Config().TeamSettings.RestrictCreationToDomains) { if !CheckUserDomain(user, *a.Config().TeamSettings.RestrictCreationToDomains) {
if !prev.IsLDAPUser() && !prev.IsSAMLUser() && user.Email != prev.Email { if !prev.IsLDAPUser() && !prev.IsSAMLUser() && user.Email != prev.Email {
@@ -1112,7 +1106,7 @@ func (a *App) UpdateUser(user *model.User, sendNotifications bool) (*model.User,
user.Email = prev.Email user.Email = prev.Email
} }
result = <-a.Srv.Store.User().Update(user, false) result := <-a.Srv.Store.User().Update(user, false)
if result.Err != nil { if result.Err != nil {
return nil, result.Err return nil, result.Err
} }
@@ -1482,8 +1476,8 @@ func (a *App) PermanentDeleteUser(user *model.User) *model.AppError {
return result.Err return result.Err
} }
if result := <-a.Srv.Store.Audit().PermanentDeleteByUser(user.Id); result.Err != nil { if err := a.Srv.Store.Audit().PermanentDeleteByUser(user.Id); err != nil {
return result.Err return err
} }
if result := <-a.Srv.Store.Team().RemoveAllMembersByUser(user.Id); result.Err != nil { if result := <-a.Srv.Store.Team().RemoveAllMembersByUser(user.Id); result.Err != nil {

Просмотреть файл

@@ -610,7 +610,12 @@ func (a *App) HandleIncomingWebhook(hookId string, req *model.IncomingWebhookReq
hook = result.Data.(*model.IncomingWebhook) hook = result.Data.(*model.IncomingWebhook)
} }
uchan := a.Srv.Store.User().Get(hook.UserId) uchan := make(chan store.StoreResult, 1)
go func() {
user, err := a.Srv.Store.User().Get(hook.UserId)
uchan <- store.StoreResult{Data: user, Err: err}
close(uchan)
}()
if len(req.Props) == 0 { if len(req.Props) == 0 {
req.Props = make(model.StringInterface) req.Props = make(model.StringInterface)

Просмотреть файл

@@ -30,9 +30,7 @@ func getUserFromUserArg(a *app.App, userArg string) *model.User {
} }
if user == nil { if user == nil {
if result := <-a.Srv.Store.User().Get(userArg); result.Err == nil { user, _ = a.Srv.Store.User().Get(userArg)
user = result.Data.(*model.User)
}
} }
return user return user

Просмотреть файл

@@ -34,71 +34,60 @@ func (s SqlAuditStore) CreateIndexesIfNotExists() {
s.CreateIndexIfNotExists("idx_audits_user_id", "Audits", "UserId") s.CreateIndexIfNotExists("idx_audits_user_id", "Audits", "UserId")
} }
func (s SqlAuditStore) Save(audit *model.Audit) store.StoreChannel { func (s SqlAuditStore) Save(audit *model.Audit) *model.AppError {
return store.Do(func(result *store.StoreResult) { audit.Id = model.NewId()
audit.Id = model.NewId() audit.CreateAt = model.GetMillis()
audit.CreateAt = model.GetMillis()
if err := s.GetMaster().Insert(audit); err != nil { if err := s.GetMaster().Insert(audit); err != nil {
result.Err = model.NewAppError("SqlAuditStore.Save", "store.sql_audit.save.saving.app_error", nil, "user_id="+audit.UserId+" action="+audit.Action, http.StatusInternalServerError) return model.NewAppError("SqlAuditStore.Save", "store.sql_audit.save.saving.app_error", nil, "user_id="+audit.UserId+" action="+audit.Action, http.StatusInternalServerError)
} }
}) return nil
} }
func (s SqlAuditStore) Get(user_id string, offset int, limit int) store.StoreChannel { func (s SqlAuditStore) Get(user_id string, offset int, limit int) (model.Audits, *model.AppError) {
return store.Do(func(result *store.StoreResult) { if limit > 1000 {
if limit > 1000 { return nil, model.NewAppError("SqlAuditStore.Get", "store.sql_audit.get.limit.app_error", nil, "user_id="+user_id, http.StatusBadRequest)
limit = 1000 }
result.Err = model.NewAppError("SqlAuditStore.Get", "store.sql_audit.get.limit.app_error", nil, "user_id="+user_id, http.StatusBadRequest)
return
}
query := "SELECT * FROM Audits" query := "SELECT * FROM Audits"
if len(user_id) != 0 { if len(user_id) != 0 {
query += " WHERE UserId = :user_id" query += " WHERE UserId = :user_id"
} }
query += " ORDER BY CreateAt DESC LIMIT :limit OFFSET :offset" query += " ORDER BY CreateAt DESC LIMIT :limit OFFSET :offset"
var audits model.Audits var audits model.Audits
if _, err := s.GetReplica().Select(&audits, query, map[string]interface{}{"user_id": user_id, "limit": limit, "offset": offset}); err != nil { if _, err := s.GetReplica().Select(&audits, query, map[string]interface{}{"user_id": user_id, "limit": limit, "offset": offset}); err != nil {
result.Err = model.NewAppError("SqlAuditStore.Get", "store.sql_audit.get.finding.app_error", nil, "user_id="+user_id, http.StatusInternalServerError) return nil, model.NewAppError("SqlAuditStore.Get", "store.sql_audit.get.finding.app_error", nil, "user_id="+user_id, http.StatusInternalServerError)
} else { }
result.Data = audits return audits, nil
}
})
} }
func (s SqlAuditStore) PermanentDeleteByUser(userId string) store.StoreChannel { func (s SqlAuditStore) PermanentDeleteByUser(userId string) *model.AppError {
return store.Do(func(result *store.StoreResult) { if _, err := s.GetMaster().Exec("DELETE FROM Audits WHERE UserId = :userId",
if _, err := s.GetMaster().Exec("DELETE FROM Audits WHERE UserId = :userId", map[string]interface{}{"userId": userId}); err != nil {
map[string]interface{}{"userId": userId}); err != nil { return model.NewAppError("SqlAuditStore.Delete", "store.sql_audit.permanent_delete_by_user.app_error", nil, "user_id="+userId, http.StatusInternalServerError)
result.Err = model.NewAppError("SqlAuditStore.Delete", "store.sql_audit.permanent_delete_by_user.app_error", nil, "user_id="+userId, http.StatusInternalServerError) }
} return nil
})
} }
func (s SqlAuditStore) PermanentDeleteBatch(endTime int64, limit int64) store.StoreChannel { func (s SqlAuditStore) PermanentDeleteBatch(endTime int64, limit int64) (int64, *model.AppError) {
return store.Do(func(result *store.StoreResult) { var query string
var query string if s.DriverName() == "postgres" {
if s.DriverName() == "postgres" { query = "DELETE from Audits WHERE Id = any (array (SELECT Id FROM Audits WHERE CreateAt < :EndTime LIMIT :Limit))"
query = "DELETE from Audits WHERE Id = any (array (SELECT Id FROM Audits WHERE CreateAt < :EndTime LIMIT :Limit))" } else {
} else { query = "DELETE from Audits WHERE CreateAt < :EndTime LIMIT :Limit"
query = "DELETE from Audits WHERE CreateAt < :EndTime LIMIT :Limit" }
}
sqlResult, err := s.GetMaster().Exec(query, map[string]interface{}{"EndTime": endTime, "Limit": limit}) sqlResult, err := s.GetMaster().Exec(query, map[string]interface{}{"EndTime": endTime, "Limit": limit})
if err != nil { if err != nil {
result.Err = model.NewAppError("SqlAuditStore.PermanentDeleteBatch", "store.sql_audit.permanent_delete_batch.app_error", nil, ""+err.Error(), http.StatusInternalServerError) return 0, model.NewAppError("SqlAuditStore.PermanentDeleteBatch", "store.sql_audit.permanent_delete_batch.app_error", nil, ""+err.Error(), http.StatusInternalServerError)
} else { }
rowsAffected, err1 := sqlResult.RowsAffected()
if err1 != nil { rowsAffected, err1 := sqlResult.RowsAffected()
result.Err = model.NewAppError("SqlAuditStore.PermanentDeleteBatch", "store.sql_audit.permanent_delete_batch.app_error", nil, ""+err.Error(), http.StatusInternalServerError) if err1 != nil {
result.Data = int64(0) return 0, model.NewAppError("SqlAuditStore.PermanentDeleteBatch", "store.sql_audit.permanent_delete_batch.app_error", nil, ""+err.Error(), http.StatusInternalServerError)
} else { }
result.Data = rowsAffected return rowsAffected, nil
}
}
})
} }

Просмотреть файл

@@ -1410,21 +1410,17 @@ func (s SqlChannelStore) GetChannelMembersTimezones(channelId string) store.Stor
}) })
} }
func (s SqlChannelStore) GetMember(channelId string, userId string) store.StoreChannel { func (s SqlChannelStore) GetMember(channelId string, userId string) (*model.ChannelMember, *model.AppError) {
return store.Do(func(result *store.StoreResult) { var dbMember channelMemberWithSchemeRoles
var dbMember channelMemberWithSchemeRoles
if err := s.GetReplica().SelectOne(&dbMember, CHANNEL_MEMBERS_WITH_SCHEME_SELECT_QUERY+"WHERE ChannelMembers.ChannelId = :ChannelId AND ChannelMembers.UserId = :UserId", map[string]interface{}{"ChannelId": channelId, "UserId": userId}); err != nil { if err := s.GetReplica().SelectOne(&dbMember, CHANNEL_MEMBERS_WITH_SCHEME_SELECT_QUERY+"WHERE ChannelMembers.ChannelId = :ChannelId AND ChannelMembers.UserId = :UserId", map[string]interface{}{"ChannelId": channelId, "UserId": userId}); err != nil {
if err == sql.ErrNoRows { if err == sql.ErrNoRows {
result.Err = model.NewAppError("SqlChannelStore.GetMember", store.MISSING_CHANNEL_MEMBER_ERROR, nil, "channel_id="+channelId+"user_id="+userId+","+err.Error(), http.StatusNotFound) return nil, model.NewAppError("SqlChannelStore.GetMember", store.MISSING_CHANNEL_MEMBER_ERROR, nil, "channel_id="+channelId+"user_id="+userId+","+err.Error(), http.StatusNotFound)
return
}
result.Err = model.NewAppError("SqlChannelStore.GetMember", "store.sql_channel.get_member.app_error", nil, "channel_id="+channelId+"user_id="+userId+","+err.Error(), http.StatusInternalServerError)
return
} }
return nil, model.NewAppError("SqlChannelStore.GetMember", "store.sql_channel.get_member.app_error", nil, "channel_id="+channelId+"user_id="+userId+","+err.Error(), http.StatusInternalServerError)
}
result.Data = dbMember.ToModel() return dbMember.ToModel(), nil
})
} }
func (s SqlChannelStore) InvalidateAllChannelMembersForUser(userId string) { func (s SqlChannelStore) InvalidateAllChannelMembersForUser(userId string) {

Просмотреть файл

@@ -329,27 +329,22 @@ func (us SqlUserStore) UpdateMfaActive(userId string, active bool) store.StoreCh
}) })
} }
func (us SqlUserStore) Get(id string) store.StoreChannel { func (us SqlUserStore) Get(id string) (*model.User, *model.AppError) {
return store.Do(func(result *store.StoreResult) { query := us.usersQuery.Where("Id = ?", id)
query := us.usersQuery.Where("Id = ?", id)
queryString, args, err := query.ToSql() queryString, args, err := query.ToSql()
if err != nil { if err != nil {
result.Err = model.NewAppError("SqlUserStore.Get", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError) return nil, model.NewAppError("SqlUserStore.Get", "store.sql_user.app_error", nil, err.Error(), http.StatusInternalServerError)
return }
}
user := &model.User{} user := &model.User{}
if err := us.GetReplica().SelectOne(user, queryString, args...); err == sql.ErrNoRows { if err := us.GetReplica().SelectOne(user, queryString, args...); err == sql.ErrNoRows {
result.Err = model.NewAppError("SqlUserStore.Get", store.MISSING_ACCOUNT_ERROR, nil, "user_id="+id, http.StatusNotFound) return nil, model.NewAppError("SqlUserStore.Get", store.MISSING_ACCOUNT_ERROR, nil, "user_id="+id, http.StatusNotFound)
return } else if err != nil {
} else if err != nil { return nil, model.NewAppError("SqlUserStore.Get", "store.sql_user.get.app_error", nil, "user_id="+id+", "+err.Error(), http.StatusInternalServerError)
result.Err = model.NewAppError("SqlUserStore.Get", "store.sql_user.get.app_error", nil, "user_id="+id+", "+err.Error(), http.StatusInternalServerError) }
return
}
result.Data = user return user, nil
})
} }
func (us SqlUserStore) GetAll() store.StoreChannel { func (us SqlUserStore) GetAll() store.StoreChannel {

Просмотреть файл

@@ -156,7 +156,7 @@ type ChannelStore interface {
SaveMember(member *model.ChannelMember) StoreChannel SaveMember(member *model.ChannelMember) StoreChannel
UpdateMember(member *model.ChannelMember) StoreChannel UpdateMember(member *model.ChannelMember) StoreChannel
GetMembers(channelId string, offset, limit int) StoreChannel GetMembers(channelId string, offset, limit int) StoreChannel
GetMember(channelId string, userId string) StoreChannel GetMember(channelId string, userId string) (*model.ChannelMember, *model.AppError)
GetChannelMembersTimezones(channelId string) StoreChannel GetChannelMembersTimezones(channelId string) StoreChannel
GetAllChannelMembersForUser(userId string, allowFromCache bool, includeDeleted bool) StoreChannel GetAllChannelMembersForUser(userId string, allowFromCache bool, includeDeleted bool) StoreChannel
InvalidateAllChannelMembersForUser(userId string) InvalidateAllChannelMembersForUser(userId string)
@@ -248,7 +248,7 @@ type UserStore interface {
UpdateAuthData(userId string, service string, authData *string, email string, resetMfa bool) StoreChannel UpdateAuthData(userId string, service string, authData *string, email string, resetMfa bool) StoreChannel
UpdateMfaSecret(userId, secret string) StoreChannel UpdateMfaSecret(userId, secret string) StoreChannel
UpdateMfaActive(userId string, active bool) StoreChannel UpdateMfaActive(userId string, active bool) StoreChannel
Get(id string) StoreChannel Get(id string) (*model.User, *model.AppError)
GetAll() StoreChannel GetAll() StoreChannel
ClearCaches() ClearCaches()
InvalidateProfilesInChannelCacheByUser(userId string) InvalidateProfilesInChannelCacheByUser(userId string)
@@ -322,10 +322,10 @@ type SessionStore interface {
} }
type AuditStore interface { type AuditStore interface {
Save(audit *model.Audit) StoreChannel Save(audit *model.Audit) *model.AppError
Get(user_id string, offset int, limit int) StoreChannel Get(user_id string, offset int, limit int) (model.Audits, *model.AppError)
PermanentDeleteByUser(userId string) StoreChannel PermanentDeleteByUser(userId string) *model.AppError
PermanentDeleteBatch(endTime int64, limit int64) StoreChannel PermanentDeleteBatch(endTime int64, limit int64) (int64, *model.AppError)
} }
type ClusterDiscoveryStore interface { type ClusterDiscoveryStore interface {

Просмотреть файл

@@ -9,6 +9,8 @@ import (
"github.com/mattermost/mattermost-server/model" "github.com/mattermost/mattermost-server/model"
"github.com/mattermost/mattermost-server/store" "github.com/mattermost/mattermost-server/store"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
) )
func TestAuditStore(t *testing.T, ss store.Store) { func TestAuditStore(t *testing.T, ss store.Store) {
@@ -18,74 +20,58 @@ func TestAuditStore(t *testing.T, ss store.Store) {
func testAuditStore(t *testing.T, ss store.Store) { func testAuditStore(t *testing.T, ss store.Store) {
audit := &model.Audit{UserId: model.NewId(), IpAddress: "ipaddress", Action: "Action"} audit := &model.Audit{UserId: model.NewId(), IpAddress: "ipaddress", Action: "Action"}
store.Must(ss.Audit().Save(audit)) require.Nil(t, ss.Audit().Save(audit))
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
store.Must(ss.Audit().Save(audit)) require.Nil(t, ss.Audit().Save(audit))
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
store.Must(ss.Audit().Save(audit)) require.Nil(t, ss.Audit().Save(audit))
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
audit.ExtraInfo = "extra" audit.ExtraInfo = "extra"
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
store.Must(ss.Audit().Save(audit)) require.Nil(t, ss.Audit().Save(audit))
time.Sleep(100 * time.Millisecond) time.Sleep(100 * time.Millisecond)
c := ss.Audit().Get(audit.UserId, 0, 100) audits, err := ss.Audit().Get(audit.UserId, 0, 100)
result := <-c require.Nil(t, err)
audits := result.Data.(model.Audits)
if len(audits) != 4 { assert.Len(t, audits, 4)
t.Fatal("Failed to save and retrieve 4 audit logs")
}
if audits[0].ExtraInfo != "extra" { assert.Equal(t, "extra", audits[0].ExtraInfo)
t.Fatal("Failed to save property for extra info")
}
c = ss.Audit().Get("missing", 0, 100) audits, err = ss.Audit().Get("missing", 0, 100)
result = <-c
audits = result.Data.(model.Audits)
if len(audits) != 0 { assert.Len(t, audits, 0)
t.Fatal("Should have returned empty because user_id is missing")
}
c = ss.Audit().Get("", 0, 100) audits, err = ss.Audit().Get("", 0, 100)
result = <-c
audits = result.Data.(model.Audits)
if len(audits) < 4 { if len(audits) < 4 {
t.Fatal("Failed to save and retrieve 4 audit logs") t.Fatal("Failed to save and retrieve 4 audit logs")
} }
if r2 := <-ss.Audit().PermanentDeleteByUser(audit.UserId); r2.Err != nil { require.Nil(t, ss.Audit().PermanentDeleteByUser(audit.UserId))
t.Fatal(r2.Err)
}
} }
func testAuditStorePermanentDeleteBatch(t *testing.T, ss store.Store) { func testAuditStorePermanentDeleteBatch(t *testing.T, ss store.Store) {
a1 := &model.Audit{UserId: model.NewId(), IpAddress: "ipaddress", Action: "Action"} a1 := &model.Audit{UserId: model.NewId(), IpAddress: "ipaddress", Action: "Action"}
store.Must(ss.Audit().Save(a1)) require.Nil(t, ss.Audit().Save(a1))
time.Sleep(10 * time.Millisecond) time.Sleep(10 * time.Millisecond)
a2 := &model.Audit{UserId: a1.UserId, IpAddress: "ipaddress", Action: "Action"} a2 := &model.Audit{UserId: a1.UserId, IpAddress: "ipaddress", Action: "Action"}
store.Must(ss.Audit().Save(a2)) require.Nil(t, ss.Audit().Save(a2))
time.Sleep(10 * time.Millisecond) time.Sleep(10 * time.Millisecond)
cutoff := model.GetMillis() cutoff := model.GetMillis()
time.Sleep(10 * time.Millisecond) time.Sleep(10 * time.Millisecond)
a3 := &model.Audit{UserId: a1.UserId, IpAddress: "ipaddress", Action: "Action"} a3 := &model.Audit{UserId: a1.UserId, IpAddress: "ipaddress", Action: "Action"}
store.Must(ss.Audit().Save(a3)) require.Nil(t, ss.Audit().Save(a3))
if r := <-ss.Audit().Get(a1.UserId, 0, 100); len(r.Data.(model.Audits)) != 3 { audits, err := ss.Audit().Get(a1.UserId, 0, 100)
t.Fatal("Expected 3 audits. Got ", len(r.Data.(model.Audits))) assert.Len(t, audits, 3)
}
store.Must(ss.Audit().PermanentDeleteBatch(cutoff, 1000000)) _, err = ss.Audit().PermanentDeleteBatch(cutoff, 1000000)
require.Nil(t, err)
if r := <-ss.Audit().Get(a1.UserId, 0, 100); len(r.Data.(model.Audits)) != 1 { audits, err = ss.Audit().Get(a1.UserId, 0, 100)
t.Fatal("Expected 1 audit. Got ", len(r.Data.(model.Audits))) assert.Len(t, audits, 1)
}
if r2 := <-ss.Audit().PermanentDeleteByUser(a1.UserId); r2.Err != nil { require.Nil(t, ss.Audit().PermanentDeleteByUser(a1.UserId))
t.Fatal(r2.Err)
}
} }

Просмотреть файл

@@ -889,7 +889,7 @@ func testChannelMemberStore(t *testing.T, ss store.Store) {
c1t3 := (<-ss.Channel().Get(c1.Id, false)).Data.(*model.Channel) c1t3 := (<-ss.Channel().Get(c1.Id, false)).Data.(*model.Channel)
assert.EqualValues(t, 0, c1t3.ExtraUpdateAt, "ExtraUpdateAt should be 0") assert.EqualValues(t, 0, c1t3.ExtraUpdateAt, "ExtraUpdateAt should be 0")
member := (<-ss.Channel().GetMember(o1.ChannelId, o1.UserId)).Data.(*model.ChannelMember) member, _ := ss.Channel().GetMember(o1.ChannelId, o1.UserId)
if member.ChannelId != o1.ChannelId { if member.ChannelId != o1.ChannelId {
t.Fatal("should have go member") t.Fatal("should have go member")
} }
@@ -1587,12 +1587,14 @@ func testChannelStoreUpdateLastViewedAt(t *testing.T, ss store.Store) {
t.Fatal("last viewed at time incorrect") t.Fatal("last viewed at time incorrect")
} }
rm1 := store.Must(ss.Channel().GetMember(m1.ChannelId, m1.UserId)).(*model.ChannelMember) rm1, err := ss.Channel().GetMember(m1.ChannelId, m1.UserId)
assert.Nil(t, err)
assert.Equal(t, rm1.LastViewedAt, o1.LastPostAt) assert.Equal(t, rm1.LastViewedAt, o1.LastPostAt)
assert.Equal(t, rm1.LastUpdateAt, o1.LastPostAt) assert.Equal(t, rm1.LastUpdateAt, o1.LastPostAt)
assert.Equal(t, rm1.MsgCount, o1.TotalMsgCount) assert.Equal(t, rm1.MsgCount, o1.TotalMsgCount)
rm2 := store.Must(ss.Channel().GetMember(m2.ChannelId, m2.UserId)).(*model.ChannelMember) rm2, err := ss.Channel().GetMember(m2.ChannelId, m2.UserId)
assert.Nil(t, err)
assert.Equal(t, rm2.LastViewedAt, o2.LastPostAt) assert.Equal(t, rm2.LastViewedAt, o2.LastPostAt)
assert.Equal(t, rm2.LastUpdateAt, o2.LastPostAt) assert.Equal(t, rm2.LastUpdateAt, o2.LastPostAt)
assert.Equal(t, rm2.MsgCount, o2.TotalMsgCount) assert.Equal(t, rm2.MsgCount, o2.TotalMsgCount)
@@ -1700,25 +1702,25 @@ func testGetMember(t *testing.T, ss store.Store) {
} }
store.Must(ss.Channel().SaveMember(m2)) store.Must(ss.Channel().SaveMember(m2))
if result := <-ss.Channel().GetMember(model.NewId(), userId); result.Err == nil { if _, err := ss.Channel().GetMember(model.NewId(), userId); err == nil {
t.Fatal("should've failed to get member for non-existent channel") t.Fatal("should've failed to get member for non-existent channel")
} }
if result := <-ss.Channel().GetMember(c1.Id, model.NewId()); result.Err == nil { if _, err := ss.Channel().GetMember(c1.Id, model.NewId()); err == nil {
t.Fatal("should've failed to get member for non-existent user") t.Fatal("should've failed to get member for non-existent user")
} }
if result := <-ss.Channel().GetMember(c1.Id, userId); result.Err != nil { if member, err := ss.Channel().GetMember(c1.Id, userId); err != nil {
t.Fatal("shouldn't have errored when getting member", result.Err) t.Fatal("shouldn't have errored when getting member", err)
} else if member := result.Data.(*model.ChannelMember); member.ChannelId != c1.Id { } else if member.ChannelId != c1.Id {
t.Fatal("should've gotten member of channel 1") t.Fatal("should've gotten member of channel 1")
} else if member.UserId != userId { } else if member.UserId != userId {
t.Fatal("should've gotten member for user") t.Fatal("should've gotten member for user")
} }
if result := <-ss.Channel().GetMember(c2.Id, userId); result.Err != nil { if member, err := ss.Channel().GetMember(c2.Id, userId); err != nil {
t.Fatal("shouldn't have errored when getting member", result.Err) t.Fatal("shouldn't have errored when getting member", err)
} else if member := result.Data.(*model.ChannelMember); member.ChannelId != c2.Id { } else if member.ChannelId != c2.Id {
t.Fatal("should've gotten member of channel 2") t.Fatal("should've gotten member of channel 2")
} else if member.UserId != userId { } else if member.UserId != userId {
t.Fatal("should've gotten member for user") t.Fatal("should've gotten member for user")
@@ -2815,23 +2817,20 @@ func testChannelStoreMigrateChannelMembers(t *testing.T, ss store.Store) {
ss.Channel().ClearCaches() ss.Channel().ClearCaches()
res1 := <-ss.Channel().GetMember(cm1.ChannelId, cm1.UserId) cm1b, err := ss.Channel().GetMember(cm1.ChannelId, cm1.UserId)
assert.Nil(t, res1.Err) assert.Nil(t, err)
cm1b := res1.Data.(*model.ChannelMember)
assert.Equal(t, "", cm1b.ExplicitRoles) assert.Equal(t, "", cm1b.ExplicitRoles)
assert.True(t, cm1b.SchemeUser) assert.True(t, cm1b.SchemeUser)
assert.True(t, cm1b.SchemeAdmin) assert.True(t, cm1b.SchemeAdmin)
res2 := <-ss.Channel().GetMember(cm2.ChannelId, cm2.UserId) cm2b, err := ss.Channel().GetMember(cm2.ChannelId, cm2.UserId)
assert.Nil(t, res2.Err) assert.Nil(t, err)
cm2b := res2.Data.(*model.ChannelMember)
assert.Equal(t, "", cm2b.ExplicitRoles) assert.Equal(t, "", cm2b.ExplicitRoles)
assert.True(t, cm2b.SchemeUser) assert.True(t, cm2b.SchemeUser)
assert.False(t, cm2b.SchemeAdmin) assert.False(t, cm2b.SchemeAdmin)
res3 := <-ss.Channel().GetMember(cm3.ChannelId, cm3.UserId) cm3b, err := ss.Channel().GetMember(cm3.ChannelId, cm3.UserId)
assert.Nil(t, res3.Err) assert.Nil(t, err)
cm3b := res3.Data.(*model.ChannelMember)
assert.Equal(t, "something_else", cm3b.ExplicitRoles) assert.Equal(t, "something_else", cm3b.ExplicitRoles)
assert.False(t, cm3b.SchemeUser) assert.False(t, cm3b.SchemeUser)
assert.False(t, cm3b.SchemeAdmin) assert.False(t, cm3b.SchemeAdmin)
@@ -2920,21 +2919,21 @@ func testChannelStoreClearAllCustomRoleAssignments(t *testing.T, ss store.Store)
require.Nil(t, (<-ss.Channel().ClearAllCustomRoleAssignments()).Err) require.Nil(t, (<-ss.Channel().ClearAllCustomRoleAssignments()).Err)
r1 := <-ss.Channel().GetMember(m1.ChannelId, m1.UserId) member, err := ss.Channel().GetMember(m1.ChannelId, m1.UserId)
require.Nil(t, r1.Err) require.Nil(t, err)
assert.Equal(t, m1.ExplicitRoles, r1.Data.(*model.ChannelMember).Roles) assert.Equal(t, m1.ExplicitRoles, member.Roles)
r2 := <-ss.Channel().GetMember(m2.ChannelId, m2.UserId) member, err = ss.Channel().GetMember(m2.ChannelId, m2.UserId)
require.Nil(t, r2.Err) require.Nil(t, err)
assert.Equal(t, "channel_user channel_admin", r2.Data.(*model.ChannelMember).Roles) assert.Equal(t, "channel_user channel_admin", member.Roles)
r3 := <-ss.Channel().GetMember(m3.ChannelId, m3.UserId) member, err = ss.Channel().GetMember(m3.ChannelId, m3.UserId)
require.Nil(t, r3.Err) require.Nil(t, err)
assert.Equal(t, m3.ExplicitRoles, r3.Data.(*model.ChannelMember).Roles) assert.Equal(t, m3.ExplicitRoles, member.Roles)
r4 := <-ss.Channel().GetMember(m4.ChannelId, m4.UserId) member, err = ss.Channel().GetMember(m4.ChannelId, m4.UserId)
require.Nil(t, r4.Err) require.Nil(t, err)
assert.Equal(t, "", r4.Data.(*model.ChannelMember).Roles) assert.Equal(t, "", member.Roles)
} }
// testMaterializedPublicChannels tests edge cases involving the triggers and stored procedures // testMaterializedPublicChannels tests edge cases involving the triggers and stored procedures

Просмотреть файл

@@ -6,7 +6,6 @@ package mocks
import mock "github.com/stretchr/testify/mock" import mock "github.com/stretchr/testify/mock"
import model "github.com/mattermost/mattermost-server/model" import model "github.com/mattermost/mattermost-server/model"
import store "github.com/mattermost/mattermost-server/store"
// AuditStore is an autogenerated mock type for the AuditStore type // AuditStore is an autogenerated mock type for the AuditStore type
type AuditStore struct { type AuditStore struct {
@@ -14,47 +13,63 @@ type AuditStore struct {
} }
// Get provides a mock function with given fields: user_id, offset, limit // Get provides a mock function with given fields: user_id, offset, limit
func (_m *AuditStore) Get(user_id string, offset int, limit int) store.StoreChannel { func (_m *AuditStore) Get(user_id string, offset int, limit int) (model.Audits, *model.AppError) {
ret := _m.Called(user_id, offset, limit) ret := _m.Called(user_id, offset, limit)
var r0 store.StoreChannel var r0 model.Audits
if rf, ok := ret.Get(0).(func(string, int, int) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(string, int, int) model.Audits); ok {
r0 = rf(user_id, offset, limit) r0 = rf(user_id, offset, limit)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(model.Audits)
} }
} }
return r0 var r1 *model.AppError
if rf, ok := ret.Get(1).(func(string, int, int) *model.AppError); ok {
r1 = rf(user_id, offset, limit)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
}
}
return r0, r1
} }
// PermanentDeleteBatch provides a mock function with given fields: endTime, limit // PermanentDeleteBatch provides a mock function with given fields: endTime, limit
func (_m *AuditStore) PermanentDeleteBatch(endTime int64, limit int64) store.StoreChannel { func (_m *AuditStore) PermanentDeleteBatch(endTime int64, limit int64) (int64, *model.AppError) {
ret := _m.Called(endTime, limit) ret := _m.Called(endTime, limit)
var r0 store.StoreChannel var r0 int64
if rf, ok := ret.Get(0).(func(int64, int64) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(int64, int64) int64); ok {
r0 = rf(endTime, limit) r0 = rf(endTime, limit)
} else { } else {
if ret.Get(0) != nil { r0 = ret.Get(0).(int64)
r0 = ret.Get(0).(store.StoreChannel) }
var r1 *model.AppError
if rf, ok := ret.Get(1).(func(int64, int64) *model.AppError); ok {
r1 = rf(endTime, limit)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
} }
} }
return r0 return r0, r1
} }
// PermanentDeleteByUser provides a mock function with given fields: userId // PermanentDeleteByUser provides a mock function with given fields: userId
func (_m *AuditStore) PermanentDeleteByUser(userId string) store.StoreChannel { func (_m *AuditStore) PermanentDeleteByUser(userId string) *model.AppError {
ret := _m.Called(userId) ret := _m.Called(userId)
var r0 store.StoreChannel var r0 *model.AppError
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(string) *model.AppError); ok {
r0 = rf(userId) r0 = rf(userId)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.AppError)
} }
} }
@@ -62,15 +77,15 @@ func (_m *AuditStore) PermanentDeleteByUser(userId string) store.StoreChannel {
} }
// Save provides a mock function with given fields: audit // Save provides a mock function with given fields: audit
func (_m *AuditStore) Save(audit *model.Audit) store.StoreChannel { func (_m *AuditStore) Save(audit *model.Audit) *model.AppError {
ret := _m.Called(audit) ret := _m.Called(audit)
var r0 store.StoreChannel var r0 *model.AppError
if rf, ok := ret.Get(0).(func(*model.Audit) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(*model.Audit) *model.AppError); ok {
r0 = rf(audit) r0 = rf(audit)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.AppError)
} }
} }

Просмотреть файл

@@ -483,19 +483,28 @@ func (_m *ChannelStore) GetFromMaster(id string) store.StoreChannel {
} }
// GetMember provides a mock function with given fields: channelId, userId // GetMember provides a mock function with given fields: channelId, userId
func (_m *ChannelStore) GetMember(channelId string, userId string) store.StoreChannel { func (_m *ChannelStore) GetMember(channelId string, userId string) (*model.ChannelMember, *model.AppError) {
ret := _m.Called(channelId, userId) ret := _m.Called(channelId, userId)
var r0 store.StoreChannel var r0 *model.ChannelMember
if rf, ok := ret.Get(0).(func(string, string) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(string, string) *model.ChannelMember); ok {
r0 = rf(channelId, userId) r0 = rf(channelId, userId)
} else { } else {
if ret.Get(0) != nil { if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel) r0 = ret.Get(0).(*model.ChannelMember)
} }
} }
return r0 var r1 *model.AppError
if rf, ok := ret.Get(1).(func(string, string) *model.AppError); ok {
r1 = rf(channelId, userId)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
}
}
return r0, r1
} }
// GetMemberCount provides a mock function with given fields: channelId, allowFromCache // GetMemberCount provides a mock function with given fields: channelId, allowFromCache

Просмотреть файл

@@ -13,6 +13,38 @@ type GroupStore struct {
mock.Mock mock.Mock
} }
// ChannelMembersToAdd provides a mock function with given fields: since
func (_m *GroupStore) ChannelMembersToAdd(since int64) store.StoreChannel {
ret := _m.Called(since)
var r0 store.StoreChannel
if rf, ok := ret.Get(0).(func(int64) store.StoreChannel); ok {
r0 = rf(since)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel)
}
}
return r0
}
// ChannelMembersToRemove provides a mock function with given fields:
func (_m *GroupStore) ChannelMembersToRemove() store.StoreChannel {
ret := _m.Called()
var r0 store.StoreChannel
if rf, ok := ret.Get(0).(func() store.StoreChannel); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel)
}
}
return r0
}
// Create provides a mock function with given fields: group // Create provides a mock function with given fields: group
func (_m *GroupStore) Create(group *model.Group) store.StoreChannel { func (_m *GroupStore) Create(group *model.Group) store.StoreChannel {
ret := _m.Called(group) ret := _m.Called(group)
@@ -269,22 +301,6 @@ func (_m *GroupStore) GetMemberUsersPage(groupID string, offset int, limit int)
return r0 return r0
} }
// ChannelMembersToAdd provides a mock function with given fields: since
func (_m *GroupStore) ChannelMembersToAdd(since int64) store.StoreChannel {
ret := _m.Called(since)
var r0 store.StoreChannel
if rf, ok := ret.Get(0).(func(int64) store.StoreChannel); ok {
r0 = rf(since)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel)
}
}
return r0
}
// TeamMembersToAdd provides a mock function with given fields: since // TeamMembersToAdd provides a mock function with given fields: since
func (_m *GroupStore) TeamMembersToAdd(since int64) store.StoreChannel { func (_m *GroupStore) TeamMembersToAdd(since int64) store.StoreChannel {
ret := _m.Called(since) ret := _m.Called(since)
@@ -301,22 +317,6 @@ func (_m *GroupStore) TeamMembersToAdd(since int64) store.StoreChannel {
return r0 return r0
} }
// ChannelMembersToRemove provides a mock function with given fields:
func (_m *GroupStore) ChannelMembersToRemove() store.StoreChannel {
ret := _m.Called()
var r0 store.StoreChannel
if rf, ok := ret.Get(0).(func() store.StoreChannel); ok {
r0 = rf()
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(store.StoreChannel)
}
}
return r0
}
// TeamMembersToRemove provides a mock function with given fields: // TeamMembersToRemove provides a mock function with given fields:
func (_m *GroupStore) TeamMembersToRemove() store.StoreChannel { func (_m *GroupStore) TeamMembersToRemove() store.StoreChannel {
ret := _m.Called() ret := _m.Called()

Просмотреть файл

@@ -78,6 +78,52 @@ func (_m *LayeredStoreDatabaseLayer) ChannelMemberHistory() store.ChannelMemberH
return r0 return r0
} }
// ChannelMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreDatabaseLayer) ChannelMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// ChannelMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreDatabaseLayer) ChannelMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// Close provides a mock function with given fields: // Close provides a mock function with given fields:
func (_m *LayeredStoreDatabaseLayer) Close() { func (_m *LayeredStoreDatabaseLayer) Close() {
_m.Called() _m.Called()
@@ -704,98 +750,6 @@ func (_m *LayeredStoreDatabaseLayer) OAuth() store.OAuthStore {
return r0 return r0
} }
// ChannelMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreDatabaseLayer) ChannelMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TeamMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreDatabaseLayer) TeamMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// ChannelMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreDatabaseLayer) ChannelMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TeamMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreDatabaseLayer) TeamMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// Plugin provides a mock function with given fields: // Plugin provides a mock function with given fields:
func (_m *LayeredStoreDatabaseLayer) Plugin() store.PluginStore { func (_m *LayeredStoreDatabaseLayer) Plugin() store.PluginStore {
ret := _m.Called() ret := _m.Called()
@@ -1398,6 +1352,52 @@ func (_m *LayeredStoreDatabaseLayer) Team() store.TeamStore {
return r0 return r0
} }
// TeamMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreDatabaseLayer) TeamMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TeamMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreDatabaseLayer) TeamMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TermsOfService provides a mock function with given fields: // TermsOfService provides a mock function with given fields:
func (_m *LayeredStoreDatabaseLayer) TermsOfService() store.TermsOfServiceStore { func (_m *LayeredStoreDatabaseLayer) TermsOfService() store.TermsOfServiceStore {
ret := _m.Called() ret := _m.Called()

Просмотреть файл

@@ -14,6 +14,52 @@ type LayeredStoreSupplier struct {
mock.Mock mock.Mock
} }
// ChannelMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreSupplier) ChannelMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// ChannelMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreSupplier) ChannelMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// GetGroupsByChannel provides a mock function with given fields: ctx, channelId, page, perPage, hints // GetGroupsByChannel provides a mock function with given fields: ctx, channelId, page, perPage, hints
func (_m *LayeredStoreSupplier) GetGroupsByChannel(ctx context.Context, channelId string, page int, perPage int, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult { func (_m *LayeredStoreSupplier) GetGroupsByChannel(ctx context.Context, channelId string, page int, perPage int, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints)) _va := make([]interface{}, len(hints))
@@ -444,98 +490,6 @@ func (_m *LayeredStoreSupplier) Next() store.LayeredStoreSupplier {
return r0 return r0
} }
// ChannelMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreSupplier) ChannelMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TeamMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreSupplier) TeamMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// ChannelMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreSupplier) ChannelMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TeamMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreSupplier) TeamMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// ReactionDelete provides a mock function with given fields: ctx, reaction, hints // ReactionDelete provides a mock function with given fields: ctx, reaction, hints
func (_m *LayeredStoreSupplier) ReactionDelete(ctx context.Context, reaction *model.Reaction, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult { func (_m *LayeredStoreSupplier) ReactionDelete(ctx context.Context, reaction *model.Reaction, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints)) _va := make([]interface{}, len(hints))
@@ -977,3 +931,49 @@ func (_m *LayeredStoreSupplier) SchemeSave(ctx context.Context, scheme *model.Sc
func (_m *LayeredStoreSupplier) SetChainNext(_a0 store.LayeredStoreSupplier) { func (_m *LayeredStoreSupplier) SetChainNext(_a0 store.LayeredStoreSupplier) {
_m.Called(_a0) _m.Called(_a0)
} }
// TeamMembersToAdd provides a mock function with given fields: ctx, since, hints
func (_m *LayeredStoreSupplier) TeamMembersToAdd(ctx context.Context, since int64, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx, since)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, int64, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, since, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}
// TeamMembersToRemove provides a mock function with given fields: ctx, hints
func (_m *LayeredStoreSupplier) TeamMembersToRemove(ctx context.Context, hints ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult {
_va := make([]interface{}, len(hints))
for _i := range hints {
_va[_i] = hints[_i]
}
var _ca []interface{}
_ca = append(_ca, ctx)
_ca = append(_ca, _va...)
ret := _m.Called(_ca...)
var r0 *store.LayeredStoreSupplierResult
if rf, ok := ret.Get(0).(func(context.Context, ...store.LayeredStoreHint) *store.LayeredStoreSupplierResult); ok {
r0 = rf(ctx, hints...)
} else {
if ret.Get(0) != nil {
r0 = ret.Get(0).(*store.LayeredStoreSupplierResult)
}
}
return r0
}

Просмотреть файл

@@ -99,19 +99,28 @@ func (_m *UserStore) Count(options model.UserCountOptions) store.StoreChannel {
} }
// Get provides a mock function with given fields: id // Get provides a mock function with given fields: id
func (_m *UserStore) Get(id string) store.StoreChannel { func (_m *UserStore) Get(id string) (*model.User, *model.AppError) {
ret := _m.Called(id) ret := _m.Called(id)
var r0 store.StoreChannel var r0 *model.User
if rf, ok := ret.Get(0).(func(string) store.StoreChannel); ok { if rf, ok := ret.Get(0).(func(string) *model.User); ok {
r0 = rf(id) r0 = rf(id)
} 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(string) *model.AppError); ok {
r1 = rf(id)
} else {
if ret.Get(1) != nil {
r1 = ret.Get(1).(*model.AppError)
}
}
return r0, r1
} }
// GetAll provides a mock function with given fields: // GetAll provides a mock function with given fields:

Просмотреть файл

@@ -1012,7 +1012,8 @@ func testSaveTeamMemberMaxMembers(t *testing.T, ss store.Store) {
} }
// Deactivating a user should make them stop counting against max members // Deactivating a user should make them stop counting against max members
user2 := store.Must(ss.User().Get(userIds[1])).(*model.User) user2, err := ss.User().Get(userIds[1])
require.Nil(t, err)
user2.DeleteAt = 1234 user2.DeleteAt = 1234
store.Must(ss.User().Update(user2, true)) store.Must(ss.User().Update(user2, true))

Просмотреть файл

@@ -220,12 +220,10 @@ func testUserStoreUpdateUpdateAt(t *testing.T, ss store.Store) {
t.Fatal(err) t.Fatal(err)
} }
if r1 := <-ss.User().Get(u1.Id); r1.Err != nil { user, err := ss.User().Get(u1.Id)
t.Fatal(r1.Err) require.Nil(t, err)
} else { if user.UpdateAt <= u1.UpdateAt {
if r1.Data.(*model.User).UpdateAt <= u1.UpdateAt { t.Fatal("UpdateAt not updated correctly")
t.Fatal("UpdateAt not updated correctly")
}
} }
} }
@@ -241,14 +239,11 @@ func testUserStoreUpdateFailedPasswordAttempts(t *testing.T, ss store.Store) {
t.Fatal(err) t.Fatal(err)
} }
if r1 := <-ss.User().Get(u1.Id); r1.Err != nil { user, err := ss.User().Get(u1.Id)
t.Fatal(r1.Err) require.Nil(t, err)
} else { if user.FailedAttempts != 3 {
if r1.Data.(*model.User).FailedAttempts != 3 { t.Fatal("FailedAttempts not updated correctly")
t.Fatal("FailedAttempts not updated correctly")
}
} }
} }
func testUserStoreGet(t *testing.T, ss store.Store) { func testUserStoreGet(t *testing.T, ss store.Store) {
@@ -274,23 +269,20 @@ func testUserStoreGet(t *testing.T, ss store.Store) {
store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}, -1)) store.Must(ss.Team().SaveMember(&model.TeamMember{TeamId: model.NewId(), UserId: u1.Id}, -1))
t.Run("fetch empty id", func(t *testing.T) { t.Run("fetch empty id", func(t *testing.T) {
require.NotNil(t, (<-ss.User().Get("")).Err) _, err := ss.User().Get("")
require.NotNil(t, err)
}) })
t.Run("fetch user 1", func(t *testing.T) { t.Run("fetch user 1", func(t *testing.T) {
result := <-ss.User().Get(u1.Id) actual, err := ss.User().Get(u1.Id)
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.(*model.User)
require.Equal(t, u1, actual) require.Equal(t, u1, actual)
require.False(t, actual.IsBot) require.False(t, actual.IsBot)
}) })
t.Run("fetch user 2, also a bot", func(t *testing.T) { t.Run("fetch user 2, also a bot", func(t *testing.T) {
result := <-ss.User().Get(u2.Id) actual, err := ss.User().Get(u2.Id)
require.Nil(t, result.Err) require.Nil(t, err)
actual := result.Data.(*model.User)
require.Equal(t, u2, actual) require.Equal(t, u2, actual)
require.True(t, actual.IsBot) require.True(t, actual.IsBot)
}) })

Просмотреть файл

@@ -25,8 +25,8 @@ type Context struct {
func (c *Context) LogAudit(extraInfo string) { func (c *Context) LogAudit(extraInfo string) {
audit := &model.Audit{UserId: c.App.Session.UserId, IpAddress: c.App.IpAddress, Action: c.App.Path, ExtraInfo: extraInfo, SessionId: c.App.Session.Id} audit := &model.Audit{UserId: c.App.Session.UserId, IpAddress: c.App.IpAddress, Action: c.App.Path, ExtraInfo: extraInfo, SessionId: c.App.Session.Id}
if r := <-c.App.Srv.Store.Audit().Save(audit); r.Err != nil { if err := c.App.Srv.Store.Audit().Save(audit); err != nil {
c.LogError(r.Err) c.LogError(err)
} }
} }
@@ -37,8 +37,8 @@ func (c *Context) LogAuditWithUserId(userId, extraInfo string) {
} }
audit := &model.Audit{UserId: userId, IpAddress: c.App.IpAddress, Action: c.App.Path, ExtraInfo: extraInfo, SessionId: c.App.Session.Id} audit := &model.Audit{UserId: userId, IpAddress: c.App.IpAddress, Action: c.App.Path, ExtraInfo: extraInfo, SessionId: c.App.Session.Id}
if r := <-c.App.Srv.Store.Audit().Save(audit); r.Err != nil { if err := c.App.Srv.Store.Audit().Save(audit); err != nil {
c.LogError(r.Err) c.LogError(err)
} }
} }