snapshot of https://github.com/mattermost/mattermost.git (master, 2026-09-16; blobs > 131072B omitted)

server/channels/app/slashcommands/auto_environment.go

@@ -0,0 +1,118 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+package slashcommands
+
+import (
+	"context"
+	"math/rand"
+	"time"
+
+	"github.com/mattermost/mattermost/server/public/model"
+	"github.com/mattermost/mattermost/server/public/shared/request"
+	"github.com/mattermost/mattermost/server/v8/channels/app"
+	"github.com/mattermost/mattermost/server/v8/channels/utils"
+)
+
+type TestEnvironment struct {
+	Teams        []*model.Team
+	Environments []TeamEnvironment
+}
+
+func CreateTestEnvironmentWithTeams(a *app.App, rctx request.CTX, 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(rctx)
+		if err != nil {
+			return TestEnvironment{}, err
+		}
+		_, _, err = client.LoginById(context.Background(), randomUser.Id, UserPassword)
+		if err != nil {
+			return TestEnvironment{}, err
+		}
+		teamEnvironment, err := CreateTestEnvironmentInTeam(a, rctx, client, team, rangeChannels, rangeUsers, rangePosts, fuzzy)
+		if err != nil {
+			return TestEnvironment{}, err
+		}
+		environment.Environments[i] = teamEnvironment
+	}
+
+	return environment, nil
+}
+
+func CreateTestEnvironmentInTeam(a *app.App, rctx request.CTX, 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(rctx, rangeUsers)
+	if err != nil {
+		return TeamEnvironment{}, err
+	}
+	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(rctx, rangeChannels)
+	if err != nil {
+		return TeamEnvironment{}, err
+	}
+
+	// Have every user join every channel
+	for _, user := range users {
+		for _, channel := range channels {
+			_, _, err := client.LoginById(context.Background(), user.Id, UserPassword)
+			if err != nil {
+				return TeamEnvironment{}, err
+			}
+
+			_, _, err = client.AddChannelMember(context.Background(), channel.Id, user.Id)
+			if err != nil {
+				return TeamEnvironment{}, err
+			}
+		}
+	}
+
+	numPosts := utils.RandIntFromRange(rangePosts)
+	numImages := utils.RandIntFromRange(rangePosts) / 4
+	for range numPosts {
+		user := users[utils.RandIntFromRange(utils.Range{Begin: 0, End: len(users) - 1})]
+		_, _, err := client.LoginById(context.Background(), 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(rctx)
+			if err != nil {
+				return TeamEnvironment{}, err
+			}
+		}
+	}
+
+	return TeamEnvironment{users, channels}, nil
+}