diff options
author | George Goldberg <george@gberg.me> | 2018-06-21 12:02:35 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2018-06-21 12:02:35 +0100 |
commit | 1f65f0e3d6d142b859f8dda52da99cb3d4a01c9c (patch) | |
tree | 57c552d90a67dc0e5a97efeb6e9f265ddecdfa4e /app/apptestlib.go | |
parent | 8e24dd78fb868cb01dd626bca3e3d756b67fddc8 (diff) | |
download | chat-1f65f0e3d6d142b859f8dda52da99cb3d4a01c9c.tar.gz chat-1f65f0e3d6d142b859f8dda52da99cb3d4a01c9c.tar.bz2 chat-1f65f0e3d6d142b859f8dda52da99cb3d4a01c9c.zip |
MM-10233: Add scheme importing to bulk importer. (#8928)
Diffstat (limited to 'app/apptestlib.go')
-rw-r--r-- | app/apptestlib.go | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/app/apptestlib.go b/app/apptestlib.go index 39e05a16e..818b21183 100644 --- a/app/apptestlib.go +++ b/app/apptestlib.go @@ -20,6 +20,7 @@ import ( "github.com/mattermost/mattermost-server/store/sqlstore" "github.com/mattermost/mattermost-server/store/storetest" "github.com/mattermost/mattermost-server/utils" + "testing" ) type TestHelper struct { @@ -446,6 +447,54 @@ func (me *TestHelper) ResetEmojisMigration() { } } +func (me *TestHelper) CheckTeamCount(t *testing.T, expected int64) { + if r := <-me.App.Srv.Store.Team().AnalyticsTeamCount(); r.Err == nil { + if r.Data.(int64) != expected { + t.Fatalf("Unexpected number of teams. Expected: %v, found: %v", expected, r.Data.(int64)) + } + } else { + t.Fatalf("Failed to get team count.") + } +} + +func (me *TestHelper) CheckChannelsCount(t *testing.T, expected int64) { + if r := <-me.App.Srv.Store.Channel().AnalyticsTypeCount("", model.CHANNEL_OPEN); r.Err == nil { + if r.Data.(int64) != expected { + t.Fatalf("Unexpected number of channels. Expected: %v, found: %v", expected, r.Data.(int64)) + } + } else { + t.Fatalf("Failed to get channel count.") + } +} + +func (me *TestHelper) SetupTeamScheme() *model.Scheme { + scheme := model.Scheme{ + Name: model.NewId(), + DisplayName: model.NewId(), + Scope: model.SCHEME_SCOPE_TEAM, + } + + if scheme, err := me.App.CreateScheme(&scheme); err == nil { + return scheme + } else { + panic(err) + } +} + +func (me *TestHelper) SetupChannelScheme() *model.Scheme { + scheme := model.Scheme{ + Name: model.NewId(), + DisplayName: model.NewId(), + Scope: model.SCHEME_SCOPE_CHANNEL, + } + + if scheme, err := me.App.CreateScheme(&scheme); err == nil { + return scheme + } else { + panic(err) + } +} + type FakeClusterInterface struct { clusterMessageHandler einterfaces.ClusterMessageHandler } |