375c0632fa
* Created message export scheduled task * Added CLI command to immediately kick off an export job * Added email addresses for users joining and leaving the channel to the export * Added support for both MySQL and PostgreSQL * Fixing gofmt error * Added a new ChannelMemberHistory store and associated tests * Updating the ChannelMemberHistory channel as users create/join/leave channels * Added user email to the message export object so it can be included in the actiance export xml * Don't fail to log a leave event if a corresponding join event wasn't logged * Adding copyright notices * Adding message export settings to daily diagnostics report * Added System Console integration for message export * Cleaned up TODOs * Made batch size configurable * Added export from timestamp to CLI command * Made ChannelMemberHistory table updates best effort * Added a context-based timeout option to the message export CLI * Minor PR updates/improvements * Removed unnecessary fields from MessageExport object to reduce query overhead * Removed JSON functions from the message export query in an effort to optimize performance * Changed the way that channel member history queries and purges work to better account for edge cases * Fixing a test I missed with the last refactor * Added file copy functionality to file backend, improved config validation, added default config values * Fixed file copy tests * More concise use of the testing libraries * Fixed context leak error * Changed default export path to correctly place an 'export' directory under the 'data' directory * Can't delete records from a read replica * Fixed copy file tests * Start job workers when license is applied, if configured to do so * Suggestions from the PR * Moar unit tests * Fixed test imports
109 строки
4.8 KiB
Go
109 строки
4.8 KiB
Go
// Copyright (c) 2017-present Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package storetest
|
|
|
|
import (
|
|
"github.com/stretchr/testify/mock"
|
|
|
|
"github.com/mattermost/mattermost-server/store"
|
|
"github.com/mattermost/mattermost-server/store/storetest/mocks"
|
|
)
|
|
|
|
// NewStoreChannel returns a channel that will receive the given result.
|
|
func NewStoreChannel(result store.StoreResult) store.StoreChannel {
|
|
ch := make(store.StoreChannel, 1)
|
|
ch <- result
|
|
return ch
|
|
}
|
|
|
|
// Store can be used to provide mock stores for testing.
|
|
type Store struct {
|
|
TeamStore mocks.TeamStore
|
|
ChannelStore mocks.ChannelStore
|
|
PostStore mocks.PostStore
|
|
UserStore mocks.UserStore
|
|
AuditStore mocks.AuditStore
|
|
ClusterDiscoveryStore mocks.ClusterDiscoveryStore
|
|
ComplianceStore mocks.ComplianceStore
|
|
SessionStore mocks.SessionStore
|
|
OAuthStore mocks.OAuthStore
|
|
SystemStore mocks.SystemStore
|
|
WebhookStore mocks.WebhookStore
|
|
CommandStore mocks.CommandStore
|
|
CommandWebhookStore mocks.CommandWebhookStore
|
|
PreferenceStore mocks.PreferenceStore
|
|
LicenseStore mocks.LicenseStore
|
|
TokenStore mocks.TokenStore
|
|
EmojiStore mocks.EmojiStore
|
|
StatusStore mocks.StatusStore
|
|
FileInfoStore mocks.FileInfoStore
|
|
ReactionStore mocks.ReactionStore
|
|
JobStore mocks.JobStore
|
|
UserAccessTokenStore mocks.UserAccessTokenStore
|
|
PluginStore mocks.PluginStore
|
|
ChannelMemberHistoryStore mocks.ChannelMemberHistoryStore
|
|
}
|
|
|
|
func (s *Store) Team() store.TeamStore { return &s.TeamStore }
|
|
func (s *Store) Channel() store.ChannelStore { return &s.ChannelStore }
|
|
func (s *Store) Post() store.PostStore { return &s.PostStore }
|
|
func (s *Store) User() store.UserStore { return &s.UserStore }
|
|
func (s *Store) Audit() store.AuditStore { return &s.AuditStore }
|
|
func (s *Store) ClusterDiscovery() store.ClusterDiscoveryStore { return &s.ClusterDiscoveryStore }
|
|
func (s *Store) Compliance() store.ComplianceStore { return &s.ComplianceStore }
|
|
func (s *Store) Session() store.SessionStore { return &s.SessionStore }
|
|
func (s *Store) OAuth() store.OAuthStore { return &s.OAuthStore }
|
|
func (s *Store) System() store.SystemStore { return &s.SystemStore }
|
|
func (s *Store) Webhook() store.WebhookStore { return &s.WebhookStore }
|
|
func (s *Store) Command() store.CommandStore { return &s.CommandStore }
|
|
func (s *Store) CommandWebhook() store.CommandWebhookStore { return &s.CommandWebhookStore }
|
|
func (s *Store) Preference() store.PreferenceStore { return &s.PreferenceStore }
|
|
func (s *Store) License() store.LicenseStore { return &s.LicenseStore }
|
|
func (s *Store) Token() store.TokenStore { return &s.TokenStore }
|
|
func (s *Store) Emoji() store.EmojiStore { return &s.EmojiStore }
|
|
func (s *Store) Status() store.StatusStore { return &s.StatusStore }
|
|
func (s *Store) FileInfo() store.FileInfoStore { return &s.FileInfoStore }
|
|
func (s *Store) Reaction() store.ReactionStore { return &s.ReactionStore }
|
|
func (s *Store) Job() store.JobStore { return &s.JobStore }
|
|
func (s *Store) UserAccessToken() store.UserAccessTokenStore { return &s.UserAccessTokenStore }
|
|
func (s *Store) Plugin() store.PluginStore { return &s.PluginStore }
|
|
func (s *Store) ChannelMemberHistory() store.ChannelMemberHistoryStore {
|
|
return &s.ChannelMemberHistoryStore
|
|
}
|
|
func (s *Store) MarkSystemRanUnitTests() { /* do nothing */ }
|
|
func (s *Store) Close() { /* do nothing */ }
|
|
func (s *Store) DropAllTables() { /* do nothing */ }
|
|
func (s *Store) TotalMasterDbConnections() int { return 1 }
|
|
func (s *Store) TotalReadDbConnections() int { return 1 }
|
|
func (s *Store) TotalSearchDbConnections() int { return 1 }
|
|
|
|
func (s *Store) AssertExpectations(t mock.TestingT) bool {
|
|
return mock.AssertExpectationsForObjects(t,
|
|
&s.TeamStore,
|
|
&s.ChannelStore,
|
|
&s.PostStore,
|
|
&s.UserStore,
|
|
&s.AuditStore,
|
|
&s.ClusterDiscoveryStore,
|
|
&s.ComplianceStore,
|
|
&s.SessionStore,
|
|
&s.OAuthStore,
|
|
&s.SystemStore,
|
|
&s.WebhookStore,
|
|
&s.CommandStore,
|
|
&s.CommandWebhookStore,
|
|
&s.PreferenceStore,
|
|
&s.LicenseStore,
|
|
&s.TokenStore,
|
|
&s.EmojiStore,
|
|
&s.StatusStore,
|
|
&s.FileInfoStore,
|
|
&s.ReactionStore,
|
|
&s.JobStore,
|
|
&s.UserAccessTokenStore,
|
|
&s.ChannelMemberHistoryStore,
|
|
&s.PluginStore,
|
|
)
|
|
}
|