github.com/vnforks/kid@v5.11.1+incompatible/migrations/helper_test.go (about) 1 // Copyright (c) 2016-present Mattermost, Inc. All Rights Reserved. 2 // See License.txt for license information. 3 4 package migrations 5 6 import ( 7 "os" 8 "time" 9 10 "github.com/mattermost/mattermost-server/app" 11 "github.com/mattermost/mattermost-server/config" 12 "github.com/mattermost/mattermost-server/mlog" 13 "github.com/mattermost/mattermost-server/model" 14 "github.com/mattermost/mattermost-server/utils" 15 ) 16 17 type TestHelper struct { 18 App *app.App 19 Server *app.Server 20 BasicTeam *model.Team 21 BasicUser *model.User 22 BasicUser2 *model.User 23 BasicChannel *model.Channel 24 BasicPost *model.Post 25 26 SystemAdminUser *model.User 27 28 tempWorkspace string 29 } 30 31 func setupTestHelper(enterprise bool) *TestHelper { 32 store := mainHelper.GetStore() 33 store.DropAllTables() 34 35 memoryStore, err := config.NewMemoryStore() 36 if err != nil { 37 panic("failed to initialize memory store: " + err.Error()) 38 } 39 40 var options []app.Option 41 options = append(options, app.ConfigStore(memoryStore)) 42 options = append(options, app.StoreOverride(mainHelper.Store)) 43 44 s, err := app.NewServer(options...) 45 if err != nil { 46 panic(err) 47 } 48 49 th := &TestHelper{ 50 App: s.FakeApp(), 51 Server: s, 52 } 53 54 th.App.UpdateConfig(func(cfg *model.Config) { *cfg.TeamSettings.MaxUsersPerTeam = 50 }) 55 th.App.UpdateConfig(func(cfg *model.Config) { *cfg.RateLimitSettings.Enable = false }) 56 prevListenAddress := *th.App.Config().ServiceSettings.ListenAddress 57 th.App.UpdateConfig(func(cfg *model.Config) { *cfg.ServiceSettings.ListenAddress = ":0" }) 58 59 serverErr := th.Server.Start() 60 if serverErr != nil { 61 panic(serverErr) 62 } 63 64 th.App.UpdateConfig(func(cfg *model.Config) { *cfg.ServiceSettings.ListenAddress = prevListenAddress }) 65 66 th.App.DoAdvancedPermissionsMigration() 67 th.App.DoEmojisPermissionsMigration() 68 69 th.App.Srv.Store.MarkSystemRanUnitTests() 70 71 th.App.UpdateConfig(func(cfg *model.Config) { *cfg.TeamSettings.EnableOpenServer = true }) 72 73 if enterprise { 74 th.App.SetLicense(model.NewTestLicense()) 75 } else { 76 th.App.SetLicense(nil) 77 } 78 79 return th 80 } 81 82 func SetupEnterprise() *TestHelper { 83 return setupTestHelper(true) 84 } 85 86 func Setup() *TestHelper { 87 return setupTestHelper(false) 88 } 89 90 func (me *TestHelper) InitBasic() *TestHelper { 91 me.SystemAdminUser = me.CreateUser() 92 me.App.UpdateUserRoles(me.SystemAdminUser.Id, model.SYSTEM_USER_ROLE_ID+" "+model.SYSTEM_ADMIN_ROLE_ID, false) 93 me.SystemAdminUser, _ = me.App.GetUser(me.SystemAdminUser.Id) 94 95 me.BasicTeam = me.CreateTeam() 96 me.BasicUser = me.CreateUser() 97 me.LinkUserToTeam(me.BasicUser, me.BasicTeam) 98 me.BasicUser2 = me.CreateUser() 99 me.LinkUserToTeam(me.BasicUser2, me.BasicTeam) 100 me.BasicChannel = me.CreateChannel(me.BasicTeam) 101 me.BasicPost = me.CreatePost(me.BasicChannel) 102 103 return me 104 } 105 106 func (me *TestHelper) MakeEmail() string { 107 return "success_" + model.NewId() + "@simulator.amazonses.com" 108 } 109 110 func (me *TestHelper) CreateTeam() *model.Team { 111 id := model.NewId() 112 team := &model.Team{ 113 DisplayName: "dn_" + id, 114 Name: "name" + id, 115 Email: "success+" + id + "@simulator.amazonses.com", 116 Type: model.TEAM_OPEN, 117 } 118 119 utils.DisableDebugLogForTest() 120 var err *model.AppError 121 if team, err = me.App.CreateTeam(team); err != nil { 122 mlog.Error(err.Error()) 123 124 time.Sleep(time.Second) 125 panic(err) 126 } 127 utils.EnableDebugLogForTest() 128 return team 129 } 130 131 func (me *TestHelper) CreateUser() *model.User { 132 id := model.NewId() 133 134 user := &model.User{ 135 Email: "success+" + id + "@simulator.amazonses.com", 136 Username: "un_" + id, 137 Nickname: "nn_" + id, 138 Password: "Password1", 139 EmailVerified: true, 140 } 141 142 utils.DisableDebugLogForTest() 143 var err *model.AppError 144 if user, err = me.App.CreateUser(user); err != nil { 145 mlog.Error(err.Error()) 146 147 time.Sleep(time.Second) 148 panic(err) 149 } 150 utils.EnableDebugLogForTest() 151 return user 152 } 153 154 func (me *TestHelper) CreateChannel(team *model.Team) *model.Channel { 155 return me.createChannel(team, model.CHANNEL_OPEN) 156 } 157 158 func (me *TestHelper) createChannel(team *model.Team, channelType string) *model.Channel { 159 id := model.NewId() 160 161 channel := &model.Channel{ 162 DisplayName: "dn_" + id, 163 Name: "name_" + id, 164 Type: channelType, 165 TeamId: team.Id, 166 CreatorId: me.BasicUser.Id, 167 } 168 169 utils.DisableDebugLogForTest() 170 var err *model.AppError 171 if channel, err = me.App.CreateChannel(channel, true); err != nil { 172 mlog.Error(err.Error()) 173 174 time.Sleep(time.Second) 175 panic(err) 176 } 177 utils.EnableDebugLogForTest() 178 return channel 179 } 180 181 func (me *TestHelper) CreateDmChannel(user *model.User) *model.Channel { 182 utils.DisableDebugLogForTest() 183 var err *model.AppError 184 var channel *model.Channel 185 if channel, err = me.App.GetOrCreateDirectChannel(me.BasicUser.Id, user.Id); err != nil { 186 mlog.Error(err.Error()) 187 188 time.Sleep(time.Second) 189 panic(err) 190 } 191 utils.EnableDebugLogForTest() 192 return channel 193 } 194 195 func (me *TestHelper) CreatePost(channel *model.Channel) *model.Post { 196 id := model.NewId() 197 198 post := &model.Post{ 199 UserId: me.BasicUser.Id, 200 ChannelId: channel.Id, 201 Message: "message_" + id, 202 CreateAt: model.GetMillis() - 10000, 203 } 204 205 utils.DisableDebugLogForTest() 206 var err *model.AppError 207 if post, err = me.App.CreatePost(post, channel, false); err != nil { 208 mlog.Error(err.Error()) 209 210 time.Sleep(time.Second) 211 panic(err) 212 } 213 utils.EnableDebugLogForTest() 214 return post 215 } 216 217 func (me *TestHelper) LinkUserToTeam(user *model.User, team *model.Team) { 218 utils.DisableDebugLogForTest() 219 220 err := me.App.JoinUserToTeam(team, user, "") 221 if err != nil { 222 mlog.Error(err.Error()) 223 224 time.Sleep(time.Second) 225 panic(err) 226 } 227 228 utils.EnableDebugLogForTest() 229 } 230 231 func (me *TestHelper) AddUserToChannel(user *model.User, channel *model.Channel) *model.ChannelMember { 232 utils.DisableDebugLogForTest() 233 234 member, err := me.App.AddUserToChannel(user, channel) 235 if err != nil { 236 mlog.Error(err.Error()) 237 238 time.Sleep(time.Second) 239 panic(err) 240 } 241 242 utils.EnableDebugLogForTest() 243 244 return member 245 } 246 247 func (me *TestHelper) TearDown() { 248 me.Server.Shutdown() 249 if err := recover(); err != nil { 250 panic(err) 251 } 252 if me.tempWorkspace != "" { 253 os.RemoveAll(me.tempWorkspace) 254 } 255 } 256 257 func (me *TestHelper) ResetRoleMigration() { 258 sqlSupplier := mainHelper.GetSqlSupplier() 259 if _, err := sqlSupplier.GetMaster().Exec("DELETE from Roles"); err != nil { 260 panic(err) 261 } 262 263 mainHelper.GetClusterInterface().SendClearRoleCacheMessage() 264 265 if _, err := sqlSupplier.GetMaster().Exec("DELETE from Systems where Name = :Name", map[string]interface{}{"Name": app.ADVANCED_PERMISSIONS_MIGRATION_KEY}); err != nil { 266 panic(err) 267 } 268 } 269 270 func (me *TestHelper) DeleteAllJobsByTypeAndMigrationKey(jobType string, migrationKey string) { 271 if res := <-me.App.Srv.Store.Job().GetAllByType(model.JOB_TYPE_MIGRATIONS); res.Err != nil { 272 panic(res.Err) 273 } else { 274 jobs := res.Data.([]*model.Job) 275 276 for _, job := range jobs { 277 if key, ok := job.Data[JOB_DATA_KEY_MIGRATION]; ok && key == migrationKey { 278 if res := <-me.App.Srv.Store.Job().Delete(job.Id); res.Err != nil { 279 panic(res.Err) 280 } 281 } 282 } 283 } 284 }