mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
ed555faa75
* MM-19137 - Migrate tests from "model/post_test.go" to use testify * revert go.sum file Co-authored-by: mattermod <mattermod@users.noreply.github.com>
505 lines
9.6 KiB
Go
505 lines
9.6 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package model
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPostToJson(t *testing.T) {
|
|
o := Post{Id: NewId(), Message: NewId()}
|
|
j := o.ToJson()
|
|
ro := PostFromJson(strings.NewReader(j))
|
|
|
|
assert.NotNil(t, ro)
|
|
assert.Equal(t, o, *ro)
|
|
}
|
|
|
|
func TestPostFromJsonError(t *testing.T) {
|
|
ro := PostFromJson(strings.NewReader(""))
|
|
assert.Nil(t, ro)
|
|
}
|
|
|
|
func TestPostIsValid(t *testing.T) {
|
|
o := Post{}
|
|
maxPostSize := 10000
|
|
|
|
err := o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.Id = NewId()
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.CreateAt = GetMillis()
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.UpdateAt = GetMillis()
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.UserId = NewId()
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.ChannelId = NewId()
|
|
o.RootId = "123"
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.RootId = ""
|
|
o.ParentId = "123"
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.ParentId = NewId()
|
|
o.RootId = ""
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.ParentId = ""
|
|
o.Message = strings.Repeat("0", maxPostSize+1)
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.Message = strings.Repeat("0", maxPostSize)
|
|
err = o.IsValid(maxPostSize)
|
|
require.Nil(t, err)
|
|
|
|
o.Message = "test"
|
|
err = o.IsValid(maxPostSize)
|
|
require.Nil(t, err)
|
|
o.Type = "junk"
|
|
err = o.IsValid(maxPostSize)
|
|
require.NotNil(t, err)
|
|
|
|
o.Type = POST_CUSTOM_TYPE_PREFIX + "type"
|
|
err = o.IsValid(maxPostSize)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestPostPreSave(t *testing.T) {
|
|
o := Post{Message: "test"}
|
|
o.PreSave()
|
|
|
|
require.NotEqual(t, 0, o.CreateAt)
|
|
|
|
past := GetMillis() - 1
|
|
o = Post{Message: "test", CreateAt: past}
|
|
o.PreSave()
|
|
|
|
require.LessOrEqual(t, o.CreateAt, past)
|
|
|
|
o.Etag()
|
|
}
|
|
|
|
func TestPostIsSystemMessage(t *testing.T) {
|
|
post1 := Post{Message: "test_1"}
|
|
post1.PreSave()
|
|
|
|
require.False(t, post1.IsSystemMessage())
|
|
|
|
post2 := Post{Message: "test_2", Type: POST_JOIN_LEAVE}
|
|
post2.PreSave()
|
|
|
|
require.True(t, post2.IsSystemMessage())
|
|
}
|
|
|
|
func TestPostChannelMentions(t *testing.T) {
|
|
post := Post{Message: "~a ~b ~b ~c/~d."}
|
|
assert.Equal(t, []string{"a", "b", "c", "d"}, post.ChannelMentions())
|
|
}
|
|
|
|
func TestPostSanitizeProps(t *testing.T) {
|
|
post1 := &Post{
|
|
Message: "test",
|
|
}
|
|
|
|
post1.SanitizeProps()
|
|
|
|
require.Nil(t, post1.Props[PROPS_ADD_CHANNEL_MEMBER])
|
|
|
|
post2 := &Post{
|
|
Message: "test",
|
|
Props: StringInterface{
|
|
PROPS_ADD_CHANNEL_MEMBER: "test",
|
|
},
|
|
}
|
|
|
|
post2.SanitizeProps()
|
|
|
|
require.Nil(t, post2.Props[PROPS_ADD_CHANNEL_MEMBER])
|
|
|
|
post3 := &Post{
|
|
Message: "test",
|
|
Props: StringInterface{
|
|
PROPS_ADD_CHANNEL_MEMBER: "no good",
|
|
"attachments": "good",
|
|
},
|
|
}
|
|
|
|
post3.SanitizeProps()
|
|
|
|
require.Nil(t, post3.Props[PROPS_ADD_CHANNEL_MEMBER])
|
|
|
|
require.NotNil(t, post3.Props["attachments"])
|
|
}
|
|
|
|
func TestPost_AttachmentsEqual(t *testing.T) {
|
|
post1 := &Post{}
|
|
post2 := &Post{}
|
|
for name, tc := range map[string]struct {
|
|
Attachments1 []*SlackAttachment
|
|
Attachments2 []*SlackAttachment
|
|
Expected bool
|
|
}{
|
|
"Empty": {
|
|
nil,
|
|
nil,
|
|
true,
|
|
},
|
|
"DifferentLength": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World",
|
|
},
|
|
},
|
|
nil,
|
|
false,
|
|
},
|
|
"EqualText": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World",
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World",
|
|
},
|
|
},
|
|
true,
|
|
},
|
|
"DifferentText": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World",
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World 2",
|
|
},
|
|
},
|
|
false,
|
|
},
|
|
"DifferentColor": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World",
|
|
Color: "#152313",
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Text: "Hello World 2",
|
|
},
|
|
},
|
|
false,
|
|
},
|
|
"EqualFields": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Fields: []*SlackAttachmentField{
|
|
{
|
|
Title: "Hello World",
|
|
Value: "FooBar",
|
|
},
|
|
{
|
|
Title: "Hello World2",
|
|
Value: "FooBar2",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Fields: []*SlackAttachmentField{
|
|
{
|
|
Title: "Hello World",
|
|
Value: "FooBar",
|
|
},
|
|
{
|
|
Title: "Hello World2",
|
|
Value: "FooBar2",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
true,
|
|
},
|
|
"DifferentFields": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Fields: []*SlackAttachmentField{
|
|
{
|
|
Title: "Hello World",
|
|
Value: "FooBar",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Fields: []*SlackAttachmentField{
|
|
{
|
|
Title: "Hello World",
|
|
Value: "FooBar",
|
|
Short: false,
|
|
},
|
|
{
|
|
Title: "Hello World2",
|
|
Value: "FooBar2",
|
|
Short: true,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
},
|
|
"EqualActions": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Actions: []*PostAction{
|
|
{
|
|
Name: "FooBar",
|
|
Options: []*PostActionOptions{
|
|
{
|
|
Text: "abcdef",
|
|
Value: "abcdef",
|
|
},
|
|
},
|
|
Integration: &PostActionIntegration{
|
|
URL: "http://localhost",
|
|
Context: map[string]interface{}{
|
|
"context": "foobar",
|
|
"test": 123,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Actions: []*PostAction{
|
|
{
|
|
Name: "FooBar",
|
|
Options: []*PostActionOptions{
|
|
{
|
|
Text: "abcdef",
|
|
Value: "abcdef",
|
|
},
|
|
},
|
|
Integration: &PostActionIntegration{
|
|
URL: "http://localhost",
|
|
Context: map[string]interface{}{
|
|
"context": "foobar",
|
|
"test": 123,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
true,
|
|
},
|
|
"DifferentActions": {
|
|
[]*SlackAttachment{
|
|
{
|
|
Actions: []*PostAction{
|
|
{
|
|
Name: "FooBar",
|
|
Options: []*PostActionOptions{
|
|
{
|
|
Text: "abcdef",
|
|
Value: "abcdef",
|
|
},
|
|
},
|
|
Integration: &PostActionIntegration{
|
|
URL: "http://localhost",
|
|
Context: map[string]interface{}{
|
|
"context": "foobar",
|
|
"test": "mattermost",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
[]*SlackAttachment{
|
|
{
|
|
Actions: []*PostAction{
|
|
{
|
|
Name: "FooBar",
|
|
Options: []*PostActionOptions{
|
|
{
|
|
Text: "abcdef",
|
|
Value: "abcdef",
|
|
},
|
|
},
|
|
Integration: &PostActionIntegration{
|
|
URL: "http://localhost",
|
|
Context: map[string]interface{}{
|
|
"context": "foobar",
|
|
"test": 123,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
false,
|
|
},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
post1.AddProp("attachments", tc.Attachments1)
|
|
post2.AddProp("attachments", tc.Attachments2)
|
|
assert.Equal(t, tc.Expected, post1.AttachmentsEqual(post2))
|
|
})
|
|
}
|
|
}
|
|
|
|
var markdownSample, markdownSampleWithRewrittenImageURLs string
|
|
|
|
func init() {
|
|
bytes, err := ioutil.ReadFile("testdata/markdown-sample.md")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
markdownSample = string(bytes)
|
|
|
|
bytes, err = ioutil.ReadFile("testdata/markdown-sample-with-rewritten-image-urls.md")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
markdownSampleWithRewrittenImageURLs = string(bytes)
|
|
}
|
|
|
|
func TestRewriteImageURLs(t *testing.T) {
|
|
for name, tc := range map[string]struct {
|
|
Markdown string
|
|
Expected string
|
|
}{
|
|
"Empty": {
|
|
Markdown: ``,
|
|
Expected: ``,
|
|
},
|
|
"NoImages": {
|
|
Markdown: `foo`,
|
|
Expected: `foo`,
|
|
},
|
|
"Link": {
|
|
Markdown: `[foo](/url)`,
|
|
Expected: `[foo](/url)`,
|
|
},
|
|
"Image": {
|
|
Markdown: ``,
|
|
Expected: ``,
|
|
},
|
|
"SpacedURL": {
|
|
Markdown: ``,
|
|
Expected: ``,
|
|
},
|
|
"Title": {
|
|
Markdown: ``,
|
|
Expected: ``,
|
|
},
|
|
"Parentheses": {
|
|
Markdown: ` "title")`,
|
|
Expected: ` "title")`,
|
|
},
|
|
"AngleBrackets": {
|
|
Markdown: ``,
|
|
Expected: ``,
|
|
},
|
|
"MultipleLines": {
|
|
Markdown: `![foo](
|
|
</url\<1\>\\>
|
|
"title"
|
|
)`,
|
|
Expected: `![foo](
|
|
<rewritten:/url\<1\>\\>
|
|
"title"
|
|
)`,
|
|
},
|
|
"ReferenceLink": {
|
|
Markdown: `[foo]: </url\<1\>\\> "title"
|
|
[foo]`,
|
|
Expected: `[foo]: </url\<1\>\\> "title"
|
|
[foo]`,
|
|
},
|
|
"ReferenceImage": {
|
|
Markdown: `[foo]: </url\<1\>\\> "title"
|
|
![foo]`,
|
|
Expected: `[foo]: <rewritten:/url\<1\>\\> "title"
|
|
![foo]`,
|
|
},
|
|
"MultipleReferenceImages": {
|
|
Markdown: `[foo]: </url1> "title"
|
|
[bar]: </url2>
|
|
[baz]: /url3 "title"
|
|
[qux]: /url4
|
|
![foo]![qux]`,
|
|
Expected: `[foo]: <rewritten:/url1> "title"
|
|
[bar]: </url2>
|
|
[baz]: /url3 "title"
|
|
[qux]: rewritten:/url4
|
|
![foo]![qux]`,
|
|
},
|
|
"DuplicateReferences": {
|
|
Markdown: `[foo]: </url1> "title"
|
|
[foo]: </url2>
|
|
[foo]: /url3 "title"
|
|
[foo]: /url4
|
|
![foo]![foo]![foo]`,
|
|
Expected: `[foo]: <rewritten:/url1> "title"
|
|
[foo]: </url2>
|
|
[foo]: /url3 "title"
|
|
[foo]: /url4
|
|
![foo]![foo]![foo]`,
|
|
},
|
|
"TrailingURL": {
|
|
Markdown: "![foo]\n\n[foo]: /url",
|
|
Expected: "![foo]\n\n[foo]: rewritten:/url",
|
|
},
|
|
"Sample": {
|
|
Markdown: markdownSample,
|
|
Expected: markdownSampleWithRewrittenImageURLs,
|
|
},
|
|
} {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert.Equal(t, tc.Expected, RewriteImageURLs(tc.Markdown, func(url string) string {
|
|
return "rewritten:" + url
|
|
}))
|
|
})
|
|
}
|
|
}
|
|
|
|
var rewriteImageURLsSink string
|
|
|
|
func BenchmarkRewriteImageURLs(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
rewriteImageURLsSink = RewriteImageURLs(markdownSample, func(url string) string {
|
|
return "rewritten:" + url
|
|
})
|
|
}
|
|
}
|