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
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
957ec1388b
Коммит
12c50eb830
@@ -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()
|
||||||
|
|||||||
12
app/audit.go
12
app/audit.go
@@ -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
|
|
||||||
}
|
}
|
||||||
|
|||||||
20
app/bot.go
20
app/bot.go
@@ -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
|
||||||
}
|
}
|
||||||
|
|||||||
126
app/channel.go
126
app/channel.go
@@ -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))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
11
app/oauth.go
11
app/oauth.go
@@ -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 {
|
||||||
|
|||||||
18
app/post.go
18
app/post.go
@@ -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)
|
||||||
|
|||||||
36
app/team.go
36
app/team.go
@@ -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 {
|
||||||
|
|||||||
26
app/user.go
26
app/user.go
@@ -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)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Ссылка в новой задаче
Block a user