mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
94827d06a3
th.InitBasic initializes users, channels, and teams. But a lot of tests don't use it at all, causing unncessary delays. A single InitBasic call takes around 400ms. This PR removes 71 of them, approximately shaving off 28 seconds. Arguably, this would be more because some tests run in loops.
252 lines
6.6 KiB
Go
252 lines
6.6 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSetAutoResponderStatus(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
user := th.CreateUser()
|
|
defer th.App.PermanentDeleteUser(user)
|
|
|
|
th.App.SetStatusOnline(user.Id, true)
|
|
|
|
patch := &model.UserPatch{}
|
|
patch.NotifyProps = make(map[string]string)
|
|
patch.NotifyProps["auto_responder_active"] = "true"
|
|
patch.NotifyProps["auto_responder_message"] = "Hello, I'm unavailable today."
|
|
|
|
userUpdated1, _ := th.App.PatchUser(user.Id, patch, true)
|
|
|
|
// autoResponder is enabled, status should be OOO
|
|
th.App.SetAutoResponderStatus(userUpdated1, user.NotifyProps)
|
|
|
|
status, err := th.App.GetStatus(userUpdated1.Id)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, model.STATUS_OUT_OF_OFFICE, status.Status)
|
|
|
|
patch2 := &model.UserPatch{}
|
|
patch2.NotifyProps = make(map[string]string)
|
|
patch2.NotifyProps["auto_responder_active"] = "false"
|
|
patch2.NotifyProps["auto_responder_message"] = "Hello, I'm unavailable today."
|
|
|
|
userUpdated2, _ := th.App.PatchUser(user.Id, patch2, true)
|
|
|
|
// autoResponder is disabled, status should be ONLINE
|
|
th.App.SetAutoResponderStatus(userUpdated2, userUpdated1.NotifyProps)
|
|
|
|
status, err = th.App.GetStatus(userUpdated2.Id)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, model.STATUS_ONLINE, status.Status)
|
|
|
|
}
|
|
|
|
func TestDisableAutoResponder(t *testing.T) {
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
user := th.CreateUser()
|
|
defer th.App.PermanentDeleteUser(user)
|
|
|
|
th.App.SetStatusOnline(user.Id, true)
|
|
|
|
patch := &model.UserPatch{}
|
|
patch.NotifyProps = make(map[string]string)
|
|
patch.NotifyProps["auto_responder_active"] = "true"
|
|
patch.NotifyProps["auto_responder_message"] = "Hello, I'm unavailable today."
|
|
|
|
th.App.PatchUser(user.Id, patch, true)
|
|
|
|
th.App.DisableAutoResponder(user.Id, true)
|
|
|
|
userUpdated1, err := th.App.GetUser(user.Id)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, userUpdated1.NotifyProps["auto_responder_active"], "false")
|
|
|
|
th.App.DisableAutoResponder(user.Id, true)
|
|
|
|
userUpdated2, err := th.App.GetUser(user.Id)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, userUpdated2.NotifyProps["auto_responder_active"], "false")
|
|
}
|
|
|
|
func TestSendAutoResponseIfNecessary(t *testing.T) {
|
|
t.Run("should send auto response when enabled", func(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
receiver := th.CreateUser()
|
|
|
|
patch := &model.UserPatch{
|
|
NotifyProps: map[string]string{
|
|
"auto_responder_active": "true",
|
|
"auto_responder_message": "Hello, I'm unavailable today.",
|
|
},
|
|
}
|
|
receiver, err := th.App.PatchUser(receiver.Id, patch, true)
|
|
require.Nil(t, err)
|
|
|
|
channel := th.CreateDmChannel(receiver)
|
|
|
|
sent, err := th.App.SendAutoResponseIfNecessary(channel, th.BasicUser)
|
|
|
|
assert.Nil(t, err)
|
|
assert.True(t, sent)
|
|
})
|
|
|
|
t.Run("should not send auto response when disabled", func(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
receiver := th.CreateUser()
|
|
|
|
patch := &model.UserPatch{
|
|
NotifyProps: map[string]string{
|
|
"auto_responder_active": "false",
|
|
"auto_responder_message": "Hello, I'm unavailable today.",
|
|
},
|
|
}
|
|
receiver, err := th.App.PatchUser(receiver.Id, patch, true)
|
|
require.Nil(t, err)
|
|
|
|
channel := th.CreateDmChannel(receiver)
|
|
|
|
sent, err := th.App.SendAutoResponseIfNecessary(channel, th.BasicUser)
|
|
|
|
assert.Nil(t, err)
|
|
assert.False(t, sent)
|
|
})
|
|
|
|
t.Run("should not send auto response for non-DM channel", func(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
sent, err := th.App.SendAutoResponseIfNecessary(th.BasicChannel, th.BasicUser)
|
|
|
|
assert.Nil(t, err)
|
|
assert.False(t, sent)
|
|
})
|
|
|
|
t.Run("should not send auto response for bot", func(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
receiver := th.CreateUser()
|
|
|
|
patch := &model.UserPatch{
|
|
NotifyProps: map[string]string{
|
|
"auto_responder_active": "true",
|
|
"auto_responder_message": "Hello, I'm unavailable today.",
|
|
},
|
|
}
|
|
receiver, err := th.App.PatchUser(receiver.Id, patch, true)
|
|
require.Nil(t, err)
|
|
|
|
channel := th.CreateDmChannel(receiver)
|
|
|
|
bot, err := th.App.CreateBot(&model.Bot{
|
|
Username: "botusername",
|
|
Description: "bot",
|
|
OwnerId: th.BasicUser.Id,
|
|
})
|
|
assert.Nil(t, err)
|
|
|
|
botUser, err := th.App.GetUser(bot.UserId)
|
|
assert.Nil(t, err)
|
|
|
|
sent, err := th.App.SendAutoResponseIfNecessary(channel, botUser)
|
|
|
|
assert.Nil(t, err)
|
|
assert.False(t, sent)
|
|
})
|
|
}
|
|
|
|
func TestSendAutoResponseSuccess(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
user := th.CreateUser()
|
|
defer th.App.PermanentDeleteUser(user)
|
|
|
|
patch := &model.UserPatch{}
|
|
patch.NotifyProps = make(map[string]string)
|
|
patch.NotifyProps["auto_responder_active"] = "true"
|
|
patch.NotifyProps["auto_responder_message"] = "Hello, I'm unavailable today."
|
|
|
|
userUpdated1, err := th.App.PatchUser(user.Id, patch, true)
|
|
require.Nil(t, err)
|
|
|
|
th.App.CreatePost(&model.Post{
|
|
ChannelId: th.BasicChannel.Id,
|
|
Message: "zz" + model.NewId() + "a",
|
|
UserId: th.BasicUser.Id},
|
|
th.BasicChannel,
|
|
false, true)
|
|
|
|
sent, err := th.App.SendAutoResponse(th.BasicChannel, userUpdated1)
|
|
|
|
assert.Nil(t, err)
|
|
assert.True(t, sent)
|
|
|
|
if list, err := th.App.GetPosts(th.BasicChannel.Id, 0, 1); err != nil {
|
|
require.Nil(t, err)
|
|
} else {
|
|
autoResponderPostFound := false
|
|
for _, post := range list.Posts {
|
|
if post.Type == model.POST_AUTO_RESPONDER {
|
|
autoResponderPostFound = true
|
|
}
|
|
}
|
|
assert.True(t, autoResponderPostFound)
|
|
}
|
|
}
|
|
|
|
func TestSendAutoResponseFailure(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
user := th.CreateUser()
|
|
defer th.App.PermanentDeleteUser(user)
|
|
|
|
patch := &model.UserPatch{}
|
|
patch.NotifyProps = make(map[string]string)
|
|
patch.NotifyProps["auto_responder_active"] = "false"
|
|
patch.NotifyProps["auto_responder_message"] = "Hello, I'm unavailable today."
|
|
|
|
userUpdated1, err := th.App.PatchUser(user.Id, patch, true)
|
|
require.Nil(t, err)
|
|
|
|
th.App.CreatePost(&model.Post{
|
|
ChannelId: th.BasicChannel.Id,
|
|
Message: "zz" + model.NewId() + "a",
|
|
UserId: th.BasicUser.Id},
|
|
th.BasicChannel,
|
|
false, true)
|
|
|
|
sent, err := th.App.SendAutoResponse(th.BasicChannel, userUpdated1)
|
|
|
|
assert.Nil(t, err)
|
|
assert.False(t, sent)
|
|
|
|
if list, err := th.App.GetPosts(th.BasicChannel.Id, 0, 1); err != nil {
|
|
require.Nil(t, err)
|
|
} else {
|
|
autoResponderPostFound := false
|
|
for _, post := range list.Posts {
|
|
if post.Type == model.POST_AUTO_RESPONDER {
|
|
autoResponderPostFound = true
|
|
}
|
|
}
|
|
assert.False(t, autoResponderPostFound)
|
|
}
|
|
}
|