a22f02c867
* Fixing /test command. * Add error handling to channels command. * Switch to using app instead of client in some places. * Fix lint being confused. * Join channels automatically. Co-authored-by: Mattermod <mattermod@users.noreply.github.com>
105 строки
2.4 KiB
Go
105 строки
2.4 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
"github.com/mattermost/mattermost-server/v5/utils"
|
|
"github.com/mattermost/mattermost-server/v5/utils/fileutils"
|
|
)
|
|
|
|
type AutoPostCreator struct {
|
|
a *App
|
|
channelid string
|
|
userid string
|
|
Fuzzy bool
|
|
TextLength utils.Range
|
|
HasImage bool
|
|
ImageFilenames []string
|
|
Users []string
|
|
Mentions utils.Range
|
|
Tags utils.Range
|
|
}
|
|
|
|
// Automatic poster used for testing
|
|
func NewAutoPostCreator(a *App, channelid, userid string) *AutoPostCreator {
|
|
return &AutoPostCreator{
|
|
a: a,
|
|
channelid: channelid,
|
|
userid: userid,
|
|
Fuzzy: false,
|
|
TextLength: utils.Range{Begin: 100, End: 200},
|
|
HasImage: false,
|
|
ImageFilenames: TEST_IMAGE_FILENAMES,
|
|
Users: []string{},
|
|
Mentions: utils.Range{Begin: 0, End: 5},
|
|
Tags: utils.Range{Begin: 0, End: 7},
|
|
}
|
|
}
|
|
|
|
func (cfg *AutoPostCreator) UploadTestFile() ([]string, error) {
|
|
filename := cfg.ImageFilenames[utils.RandIntFromRange(utils.Range{Begin: 0, End: len(cfg.ImageFilenames) - 1})]
|
|
|
|
path, _ := fileutils.FindDir("tests")
|
|
file, err := os.Open(filepath.Join(path, filename))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer file.Close()
|
|
|
|
data := &bytes.Buffer{}
|
|
_, err = io.Copy(data, file)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
fileResp, err2 := cfg.a.UploadFile(data.Bytes(), cfg.channelid, filename)
|
|
if err2 != nil {
|
|
return nil, err2
|
|
}
|
|
|
|
return []string{fileResp.Id}, nil
|
|
}
|
|
|
|
func (cfg *AutoPostCreator) CreateRandomPost() (*model.Post, error) {
|
|
return cfg.CreateRandomPostNested("", "")
|
|
}
|
|
|
|
func (cfg *AutoPostCreator) CreateRandomPostNested(parentId, rootId string) (*model.Post, error) {
|
|
var fileIds []string
|
|
if cfg.HasImage {
|
|
var err error
|
|
fileIds, err = cfg.UploadTestFile()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
var postText string
|
|
if cfg.Fuzzy {
|
|
postText = utils.FuzzPost()
|
|
} else {
|
|
postText = utils.RandomText(cfg.TextLength, cfg.Tags, cfg.Mentions, cfg.Users)
|
|
}
|
|
|
|
post := &model.Post{
|
|
ChannelId: cfg.channelid,
|
|
UserId: cfg.userid,
|
|
ParentId: parentId,
|
|
RootId: rootId,
|
|
Message: postText,
|
|
FileIds: fileIds,
|
|
}
|
|
rpost, err := cfg.a.CreatePostMissingChannel(post, true)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return rpost, nil
|
|
}
|