2d1135ca46
New plugin APIs and hooks for accessing Shared Channels service via plugin. - RegisterPluginForSharedChannels(opts model.RegisterPluginOpts) (remoteID string, err error) - UnregisterPluginForSharedChannels(pluginID string) error - ShareChannel(sc *model.SharedChannel) (*model.SharedChannel, error) - UpdateSharedChannel(sc *model.SharedChannel) (*model.SharedChannel, error) - UnshareChannel(channelID string) (unshared bool, err error) - UpdateSharedChannelCursor(channelID, remoteID string, cusror model.GetPostsSinceForSyncCursor) error - SyncSharedChannel(channelID string) error - InviteRemoteToChannel(channelID string, remoteID string, userID string) error - UninviteRemoteFromChannel(channelID string, remoteID string) error Hooks - OnSharedChannelsSyncMsg(msg *model.SyncMsg, rc *model.RemoteCluster) (model.SyncResponse, error) - OnSharedChannelsPing(rc *model.RemoteCluster) bool
120 строки
3.2 KiB
Go
120 строки
3.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package remotecluster
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/plugin/plugintest/mock"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store/storetest/mocks"
|
|
"github.com/mattermost/mattermost/server/v8/einterfaces"
|
|
)
|
|
|
|
type mockServer struct {
|
|
remotes []*model.RemoteCluster
|
|
logger *mlog.Logger
|
|
user *model.User
|
|
}
|
|
|
|
func newMockServer(t *testing.T, remotes []*model.RemoteCluster) *mockServer {
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
return &mockServer{
|
|
remotes: remotes,
|
|
logger: logger,
|
|
}
|
|
}
|
|
|
|
func (ms *mockServer) SetUser(user *model.User) {
|
|
ms.user = user
|
|
}
|
|
|
|
func (ms *mockServer) Config() *model.Config { return nil }
|
|
func (ms *mockServer) GetMetrics() einterfaces.MetricsInterface { return nil }
|
|
func (ms *mockServer) IsLeader() bool { return true }
|
|
func (ms *mockServer) AddClusterLeaderChangedListener(listener func()) string { return model.NewId() }
|
|
func (ms *mockServer) RemoveClusterLeaderChangedListener(id string) {}
|
|
func (ms *mockServer) Log() *mlog.Logger {
|
|
return ms.logger
|
|
}
|
|
func (ms *mockServer) GetStore() store.Store {
|
|
anyQueryFilter := mock.MatchedBy(func(filter model.RemoteClusterQueryFilter) bool {
|
|
return true
|
|
})
|
|
anyUserId := mock.AnythingOfType("string")
|
|
anyId := mock.AnythingOfType("string")
|
|
|
|
remoteClusterStoreMock := &mocks.RemoteClusterStore{}
|
|
remoteClusterStoreMock.On("GetByTopic", "share").Return(ms.remotes, nil)
|
|
remoteClusterStoreMock.On("GetAll", anyQueryFilter).Return(ms.remotes, nil)
|
|
remoteClusterStoreMock.On("SetLastPingAt", anyId).Return(nil)
|
|
|
|
userStoreMock := &mocks.UserStore{}
|
|
userStoreMock.On("Get", context.Background(), anyUserId).Return(ms.user, nil)
|
|
|
|
storeMock := &mocks.Store{}
|
|
storeMock.On("RemoteCluster").Return(remoteClusterStoreMock)
|
|
storeMock.On("User").Return(userStoreMock)
|
|
return storeMock
|
|
}
|
|
|
|
type mockApp struct {
|
|
offlinePluginIDs []string
|
|
|
|
mux sync.Mutex
|
|
totalPingCount int
|
|
totalPingErrors int
|
|
pingCounts map[string]int
|
|
}
|
|
|
|
func newMockApp(t *testing.T, offlinePluginIDs []string) *mockApp {
|
|
return &mockApp{
|
|
offlinePluginIDs: offlinePluginIDs,
|
|
pingCounts: make(map[string]int),
|
|
}
|
|
}
|
|
|
|
func (ma *mockApp) OnSharedChannelsPing(rc *model.RemoteCluster) bool {
|
|
ma.mux.Lock()
|
|
defer ma.mux.Unlock()
|
|
|
|
for _, id := range ma.offlinePluginIDs {
|
|
if rc.PluginID == id {
|
|
ma.totalPingErrors++
|
|
return false
|
|
}
|
|
}
|
|
|
|
ma.totalPingCount++
|
|
|
|
count := ma.pingCounts[rc.PluginID]
|
|
ma.pingCounts[rc.PluginID] = count + 1
|
|
|
|
return true
|
|
}
|
|
|
|
func (ma *mockApp) GetTotalPingCount() int {
|
|
ma.mux.Lock()
|
|
defer ma.mux.Unlock()
|
|
return ma.totalPingCount
|
|
}
|
|
|
|
func (ma *mockApp) GetTotalPingErrorCount() int {
|
|
ma.mux.Lock()
|
|
defer ma.mux.Unlock()
|
|
return ma.totalPingErrors
|
|
}
|
|
|
|
func (ma *mockApp) GetPingCount(pluginID string) int {
|
|
ma.mux.Lock()
|
|
defer ma.mux.Unlock()
|
|
|
|
return ma.pingCounts[pluginID]
|
|
}
|