mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
8a0e649f98
* Implemented initial changes for files table * Removed *_benchmark_test.go files * Re-implemented GetPublicFile and added support for old path * Localization for files table * Moved file system code into utils package * Finished server-side changes and added initial upgrade script * Added getPostFiles api * Re-add Extension and HasPreviewImage fields to FileInfo * Removed unused translation * Fixed merge conflicts left over after permissions changes * Forced FileInfo.extension to be lower case * Changed FileUploadResponse to contain the FileInfos instead of FileIds * Fixed permissions on getFile* calls * Fixed notifications for file uploads * Added initial version of client code for files changes * Permanently added FileIds field to Post object and removed Post.HasFiles * Updated PostStore.Update to be usable in more circumstances * Re-added Filenames field and switched file migration to be entirely lazy-loaded * Increased max listener count for FileStore * Removed unused fileInfoCache * Moved file system code back into api * Removed duplicate test case * Fixed unit test running on ports other than 8065 * Renamed HasPermissionToPostContext to HasPermissionToChannelByPostContext * Refactored handleImages to make it more easily understandable * Renamed getPostFiles to getFileInfosForPost * Re-added pre-FileIds posts to analytics * Changed files to be saved as their ids as opposed to id/filename.ext * Renamed FileInfo.UserId to FileInfo.CreatorId * Fixed detection of language in CodePreview * Fixed switching between threads in the RHS not loading new files * Add serverside protection against a rare bug where the client sends the same file twice for a single post * Refactored the important parts of uploadFile api call into a function that can be called without a web context
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
// Copyright (c) 2015 Mattermost, Inc. All Rights Reserved.
|
|
// See License.txt for license information.
|
|
|
|
package api
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/mattermost/platform/model"
|
|
"github.com/mattermost/platform/utils"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
type AutoPostCreator struct {
|
|
client *model.Client
|
|
channelid 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(client *model.Client, channelid string) *AutoPostCreator {
|
|
return &AutoPostCreator{
|
|
client: client,
|
|
channelid: channelid,
|
|
Fuzzy: false,
|
|
TextLength: utils.Range{100, 200},
|
|
HasImage: false,
|
|
ImageFilenames: TEST_IMAGE_FILENAMES,
|
|
Users: []string{},
|
|
Mentions: utils.Range{0, 5},
|
|
Tags: utils.Range{0, 7},
|
|
}
|
|
}
|
|
|
|
func (cfg *AutoPostCreator) UploadTestFile() ([]string, bool) {
|
|
filename := cfg.ImageFilenames[utils.RandIntFromRange(utils.Range{0, len(cfg.ImageFilenames) - 1})]
|
|
|
|
path := utils.FindDir("web/static/images")
|
|
file, err := os.Open(path + "/" + filename)
|
|
defer file.Close()
|
|
|
|
data := &bytes.Buffer{}
|
|
_, err = io.Copy(data, file)
|
|
if err != nil {
|
|
return nil, false
|
|
}
|
|
|
|
resp, appErr := cfg.client.UploadPostAttachment(data.Bytes(), cfg.channelid, filename)
|
|
if appErr != nil {
|
|
return nil, false
|
|
}
|
|
|
|
return []string{resp.FileInfos[0].Id}, true
|
|
}
|
|
|
|
func (cfg *AutoPostCreator) CreateRandomPost() (*model.Post, bool) {
|
|
var fileIds []string
|
|
if cfg.HasImage {
|
|
var err1 bool
|
|
fileIds, err1 = cfg.UploadTestFile()
|
|
if err1 == false {
|
|
return nil, false
|
|
}
|
|
}
|
|
|
|
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,
|
|
Message: postText,
|
|
FileIds: fileIds}
|
|
result, err2 := cfg.client.CreatePost(post)
|
|
if err2 != nil {
|
|
return nil, false
|
|
}
|
|
return result.Data.(*model.Post), true
|
|
}
|
|
|
|
func (cfg *AutoPostCreator) CreateTestPosts(rangePosts utils.Range) ([]*model.Post, bool) {
|
|
numPosts := utils.RandIntFromRange(rangePosts)
|
|
posts := make([]*model.Post, numPosts)
|
|
|
|
for i := 0; i < numPosts; i++ {
|
|
var err bool
|
|
posts[i], err = cfg.CreateRandomPost()
|
|
if err != true {
|
|
return posts, false
|
|
}
|
|
}
|
|
|
|
return posts, true
|
|
}
|