mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
63 lines
2.1 KiB
Go
63 lines
2.1 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package remotecluster
|
|
|
|
import (
|
|
"context"
|
|
"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")
|
|
|
|
remoteClusterStoreMock := &mocks.RemoteClusterStore{}
|
|
remoteClusterStoreMock.On("GetByTopic", "share").Return(ms.remotes, nil)
|
|
remoteClusterStoreMock.On("GetAll", anyQueryFilter).Return(ms.remotes, 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
|
|
}
|