2019-11-29 12:59:40 +01:00
|
|
|
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
|
|
|
// See LICENSE.txt for license information.
|
2018-12-06 13:19:32 -05:00
|
|
|
|
|
|
|
|
package testlib
|
|
|
|
|
|
|
|
|
|
import (
|
2020-03-02 17:13:39 +01:00
|
|
|
"net/http"
|
|
|
|
|
"strconv"
|
|
|
|
|
|
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
|
|
|
"github.com/mattermost/mattermost-server/v5/plugin/plugintest/mock"
|
2019-11-28 14:39:38 +01:00
|
|
|
"github.com/mattermost/mattermost-server/v5/store"
|
2020-03-02 17:13:39 +01:00
|
|
|
"github.com/mattermost/mattermost-server/v5/store/storetest/mocks"
|
2018-12-06 13:19:32 -05:00
|
|
|
)
|
|
|
|
|
|
|
|
|
|
type TestStore struct {
|
|
|
|
|
store.Store
|
|
|
|
|
}
|
|
|
|
|
|
2018-12-10 11:58:09 -05:00
|
|
|
func (s *TestStore) Close() {
|
|
|
|
|
// Don't propagate to the underlying store, since this instance is persistent.
|
2018-12-06 13:19:32 -05:00
|
|
|
}
|
2020-03-02 17:13:39 +01:00
|
|
|
|
|
|
|
|
func GetMockStoreForSetupFunctions() *mocks.Store {
|
|
|
|
|
mockStore := mocks.Store{}
|
|
|
|
|
systemStore := mocks.SystemStore{}
|
2020-08-21 20:23:04 +02:00
|
|
|
systemStore.On("GetByName", "UpgradedFromTE").Return(nil, model.NewAppError("FakeError", "app.system.get_by_name.app_error", nil, "", http.StatusInternalServerError))
|
2021-02-26 07:41:05 +01:00
|
|
|
systemStore.On("GetByName", "ContentExtractionConfigMigrationComplete").Return(&model.System{Name: "ContentExtractionConfigMigrationComplete", Value: "true"}, nil)
|
2020-08-13 11:02:57 -04:00
|
|
|
systemStore.On("GetByName", "AsymmetricSigningKey").Return(nil, model.NewAppError("FakeError", "app.system.get_by_name.app_error", nil, "", http.StatusInternalServerError))
|
|
|
|
|
systemStore.On("GetByName", "PostActionCookieSecret").Return(nil, model.NewAppError("FakeError", "app.system.get_by_name.app_error", nil, "", http.StatusInternalServerError))
|
2020-03-02 17:13:39 +01:00
|
|
|
systemStore.On("GetByName", "InstallationDate").Return(&model.System{Name: "InstallationDate", Value: strconv.FormatInt(model.GetMillis(), 10)}, nil)
|
2020-04-30 18:18:12 +03:00
|
|
|
systemStore.On("GetByName", "FirstServerRunTimestamp").Return(&model.System{Name: "FirstServerRunTimestamp", Value: "10"}, nil)
|
2020-03-02 17:13:39 +01:00
|
|
|
systemStore.On("GetByName", "AdvancedPermissionsMigrationComplete").Return(&model.System{Name: "AdvancedPermissionsMigrationComplete", Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", "EmojisPermissionsMigrationComplete").Return(&model.System{Name: "EmojisPermissionsMigrationComplete", Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", "GuestRolesCreationMigrationComplete").Return(&model.System{Name: "GuestRolesCreationMigrationComplete", Value: "true"}, nil)
|
2020-08-21 16:49:31 -04:00
|
|
|
systemStore.On("GetByName", "SystemConsoleRolesCreationMigrationComplete").Return(&model.System{Name: "SystemConsoleRolesCreationMigrationComplete", Value: "true"}, nil)
|
2020-03-02 17:13:39 +01:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_EMOJI_PERMISSIONS_SPLIT).Return(&model.System{Name: model.MIGRATION_KEY_EMOJI_PERMISSIONS_SPLIT, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_WEBHOOK_PERMISSIONS_SPLIT).Return(&model.System{Name: model.MIGRATION_KEY_WEBHOOK_PERMISSIONS_SPLIT, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_LIST_JOIN_PUBLIC_PRIVATE_TEAMS).Return(&model.System{Name: model.MIGRATION_KEY_LIST_JOIN_PUBLIC_PRIVATE_TEAMS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_REMOVE_PERMANENT_DELETE_USER).Return(&model.System{Name: model.MIGRATION_KEY_REMOVE_PERMANENT_DELETE_USER, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_BOT_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_BOT_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_APPLY_CHANNEL_MANAGE_DELETE_TO_CHANNEL_USER).Return(&model.System{Name: model.MIGRATION_KEY_APPLY_CHANNEL_MANAGE_DELETE_TO_CHANNEL_USER, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_REMOVE_CHANNEL_MANAGE_DELETE_FROM_TEAM_USER).Return(&model.System{Name: model.MIGRATION_KEY_REMOVE_CHANNEL_MANAGE_DELETE_FROM_TEAM_USER, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_VIEW_MEMBERS_NEW_PERMISSION).Return(&model.System{Name: model.MIGRATION_KEY_VIEW_MEMBERS_NEW_PERMISSION, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_MANAGE_GUESTS_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_MANAGE_GUESTS_PERMISSIONS, Value: "true"}, nil)
|
2020-03-23 13:44:20 -04:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_CHANNEL_MODERATIONS_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_CHANNEL_MODERATIONS_PERMISSIONS, Value: "true"}, nil)
|
2020-03-30 23:08:48 -04:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_USE_GROUP_MENTIONS_PERMISSION).Return(&model.System{Name: model.MIGRATION_KEY_ADD_USE_GROUP_MENTIONS_PERMISSION, Value: "true"}, nil)
|
2020-08-21 16:49:31 -04:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_SYSTEM_CONSOLE_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_SYSTEM_CONSOLE_PERMISSIONS, Value: "true"}, nil)
|
2020-09-03 13:41:07 -04:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_CONVERT_CHANNEL_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_CONVERT_CHANNEL_PERMISSIONS, Value: "true"}, nil)
|
2020-11-13 10:57:57 -05:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_SYSTEM_ROLES_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_SYSTEM_ROLES_PERMISSIONS, Value: "true"}, nil)
|
2020-12-08 12:41:28 -05:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_BILLING_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_BILLING_PERMISSIONS, Value: "true"}, nil)
|
2021-01-12 14:45:17 -05:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_DOWNLOAD_COMPLIANCE_EXPORT_RESULTS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_DOWNLOAD_COMPLIANCE_EXPORT_RESULTS, Value: "true"}, nil)
|
2021-04-06 10:39:48 -04:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_SITE_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_SITE_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_EXPERIMENTAL_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_EXPERIMENTAL_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_AUTHENTICATION_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_AUTHENTICATION_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_COMPLIANCE_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_EXPERIMENTAL_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_ENVIRONMENT_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_ENVIRONMENT_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_REPORTING_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_REPORTING_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_ABOUT_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_ABOUT_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
|
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_INTEGRATIONS_SUBSECTION_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_INTEGRATIONS_SUBSECTION_PERMISSIONS, Value: "true"}, nil)
|
2020-10-26 15:54:50 +05:30
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_MANAGE_SHARED_CHANNEL_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_MANAGE_SHARED_CHANNEL_PERMISSIONS, Value: "true"}, nil)
|
2021-04-30 14:59:29 -04:00
|
|
|
systemStore.On("GetByName", model.MIGRATION_KEY_ADD_MANAGE_SECURE_CONNECTIONS_PERMISSIONS).Return(&model.System{Name: model.MIGRATION_KEY_ADD_MANAGE_SECURE_CONNECTIONS_PERMISSIONS, Value: "true"}, nil)
|
2020-03-02 17:13:39 +01:00
|
|
|
systemStore.On("Get").Return(make(model.StringMap), nil)
|
|
|
|
|
systemStore.On("Save", mock.AnythingOfType("*model.System")).Return(nil)
|
|
|
|
|
|
|
|
|
|
userStore := mocks.UserStore{}
|
|
|
|
|
userStore.On("Count", mock.AnythingOfType("model.UserCountOptions")).Return(int64(1), nil)
|
|
|
|
|
userStore.On("DeactivateGuests").Return(nil, nil)
|
|
|
|
|
userStore.On("ClearCaches").Return(nil)
|
|
|
|
|
|
|
|
|
|
postStore := mocks.PostStore{}
|
|
|
|
|
postStore.On("GetMaxPostSize").Return(4000)
|
|
|
|
|
|
|
|
|
|
statusStore := mocks.StatusStore{}
|
|
|
|
|
statusStore.On("ResetAll").Return(nil)
|
|
|
|
|
|
|
|
|
|
channelStore := mocks.ChannelStore{}
|
|
|
|
|
channelStore.On("ClearCaches").Return(nil)
|
|
|
|
|
|
2020-03-23 13:44:20 -04:00
|
|
|
schemeStore := mocks.SchemeStore{}
|
|
|
|
|
schemeStore.On("GetAllPage", model.SCHEME_SCOPE_TEAM, mock.Anything, 100).Return([]*model.Scheme{}, nil)
|
|
|
|
|
|
2020-03-02 17:13:39 +01:00
|
|
|
teamStore := mocks.TeamStore{}
|
|
|
|
|
|
2020-11-16 18:25:32 +05:30
|
|
|
roleStore := mocks.RoleStore{}
|
|
|
|
|
roleStore.On("GetAll").Return([]*model.Role{}, nil)
|
|
|
|
|
|
2020-03-02 17:13:39 +01:00
|
|
|
mockStore.On("System").Return(&systemStore)
|
|
|
|
|
mockStore.On("User").Return(&userStore)
|
|
|
|
|
mockStore.On("Post").Return(&postStore)
|
|
|
|
|
mockStore.On("Status").Return(&statusStore)
|
|
|
|
|
mockStore.On("Channel").Return(&channelStore)
|
|
|
|
|
mockStore.On("Team").Return(&teamStore)
|
2020-11-16 18:25:32 +05:30
|
|
|
mockStore.On("Role").Return(&roleStore)
|
2020-03-23 13:44:20 -04:00
|
|
|
mockStore.On("Scheme").Return(&schemeStore)
|
2020-03-02 17:13:39 +01:00
|
|
|
mockStore.On("Close").Return(nil)
|
|
|
|
|
mockStore.On("DropAllTables").Return(nil)
|
|
|
|
|
mockStore.On("MarkSystemRanUnitTests").Return(nil)
|
|
|
|
|
return &mockStore
|
|
|
|
|
}
|
