a8ca5c423f
* Return an error seperately from Response * Remove BuildErrorResponse * Drop Response.Error from model/client4.go * Migrate require.Nil checks * Migrate require.NotNil checks * More manual fixes * Move error check out of CheckOKStatus and CheckCreatedStatus * Move error check out of CheckForbiddenStatus * Move error check out of CheckUnauthorizedStatus * Move error check out of CheckNotFoundStatus * Move error check out of CheckBadRequestStatus * Move error check out of CheckNotImplementedStatus and CheckRequestEntityTooLargeStatus * Move error check out of CheckInternalErrorStatus * Move error check out of CheckServiceUnavailableStatus * Remove error check from checkHTTPStatus * Remove remaining references to Response.Error * Check previously unchecked errors * Manually fix compile and linter errors * Return error in CreateWebSocket methods * Return error instead of *AppError in DoApi methods * Manually fix bad replacments * Conistently return Response and error * Use err instead of seperate bool return value to indicate success * Reduce ussage of model.AppError in web/oauth_test.go * Remove client4.Must * Check error in buf.ReadFrom * Fix failing tests
115 строки
3.3 KiB
Go
115 строки
3.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package slashcommands
|
|
|
|
import (
|
|
"math/rand"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost-server/v6/app"
|
|
"github.com/mattermost/mattermost-server/v6/app/request"
|
|
"github.com/mattermost/mattermost-server/v6/model"
|
|
"github.com/mattermost/mattermost-server/v6/utils"
|
|
)
|
|
|
|
type TestEnvironment struct {
|
|
Teams []*model.Team
|
|
Environments []TeamEnvironment
|
|
}
|
|
|
|
func CreateTestEnvironmentWithTeams(a *app.App, c *request.Context, client *model.Client4, rangeTeams utils.Range, rangeChannels utils.Range, rangeUsers utils.Range, rangePosts utils.Range, fuzzy bool) (TestEnvironment, error) {
|
|
rand.Seed(time.Now().UTC().UnixNano())
|
|
|
|
teamCreator := NewAutoTeamCreator(client)
|
|
teamCreator.Fuzzy = fuzzy
|
|
teams, err := teamCreator.CreateTestTeams(rangeTeams)
|
|
if err != nil {
|
|
return TestEnvironment{}, err
|
|
}
|
|
|
|
environment := TestEnvironment{teams, make([]TeamEnvironment, len(teams))}
|
|
|
|
for i, team := range teams {
|
|
userCreator := NewAutoUserCreator(a, client, team)
|
|
userCreator.Fuzzy = fuzzy
|
|
randomUser, err := userCreator.createRandomUser(c)
|
|
if err != nil {
|
|
return TestEnvironment{}, err
|
|
}
|
|
client.LoginById(randomUser.Id, UserPassword)
|
|
teamEnvironment, err := CreateTestEnvironmentInTeam(a, c, client, team, rangeChannels, rangeUsers, rangePosts, fuzzy)
|
|
if err != nil {
|
|
return TestEnvironment{}, err
|
|
}
|
|
environment.Environments[i] = teamEnvironment
|
|
}
|
|
|
|
return environment, nil
|
|
}
|
|
|
|
func CreateTestEnvironmentInTeam(a *app.App, c *request.Context, client *model.Client4, team *model.Team, rangeChannels utils.Range, rangeUsers utils.Range, rangePosts utils.Range, fuzzy bool) (TeamEnvironment, error) {
|
|
rand.Seed(time.Now().UTC().UnixNano())
|
|
|
|
// We need to create at least one user
|
|
if rangeUsers.Begin <= 0 {
|
|
rangeUsers.Begin = 1
|
|
}
|
|
|
|
userCreator := NewAutoUserCreator(a, client, team)
|
|
userCreator.Fuzzy = fuzzy
|
|
users, err := userCreator.CreateTestUsers(c, rangeUsers)
|
|
if err != nil {
|
|
return TeamEnvironment{}, nil
|
|
}
|
|
usernames := make([]string, len(users))
|
|
for i, user := range users {
|
|
usernames[i] = user.Username
|
|
}
|
|
|
|
channelCreator := NewAutoChannelCreator(a, team, users[0].Id)
|
|
channelCreator.Fuzzy = fuzzy
|
|
channels, err := channelCreator.CreateTestChannels(c, rangeChannels)
|
|
if err != nil {
|
|
return TeamEnvironment{}, nil
|
|
}
|
|
|
|
// Have every user join every channel
|
|
for _, user := range users {
|
|
for _, channel := range channels {
|
|
_, _, err := client.LoginById(user.Id, UserPassword)
|
|
if err != nil {
|
|
return TeamEnvironment{}, err
|
|
}
|
|
|
|
_, _, err = client.AddChannelMember(channel.Id, user.Id)
|
|
if err != nil {
|
|
return TeamEnvironment{}, err
|
|
}
|
|
}
|
|
}
|
|
|
|
numPosts := utils.RandIntFromRange(rangePosts)
|
|
numImages := utils.RandIntFromRange(rangePosts) / 4
|
|
for j := 0; j < numPosts; j++ {
|
|
user := users[utils.RandIntFromRange(utils.Range{Begin: 0, End: len(users) - 1})]
|
|
_, _, err := client.LoginById(user.Id, UserPassword)
|
|
if err != nil {
|
|
return TeamEnvironment{}, err
|
|
}
|
|
|
|
for i, channel := range channels {
|
|
postCreator := NewAutoPostCreator(a, channel.Id, user.Id)
|
|
postCreator.HasImage = i < numImages
|
|
postCreator.Users = usernames
|
|
postCreator.Fuzzy = fuzzy
|
|
_, err := postCreator.CreateRandomPost(c)
|
|
if err != nil {
|
|
return TeamEnvironment{}, err
|
|
}
|
|
}
|
|
}
|
|
|
|
return TeamEnvironment{users, channels}, nil
|
|
}
|