mirror of
https://github.com/mattermost/mattermost.git
synced 2025-02-25 18:55:24 -06:00
232 lines
7.2 KiB
Go
232 lines
7.2 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost-server/v5/model"
|
|
)
|
|
|
|
func TestInviteProvider(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
channel := th.createChannel(th.BasicTeam, model.CHANNEL_OPEN)
|
|
privateChannel := th.createChannel(th.BasicTeam, model.CHANNEL_PRIVATE)
|
|
dmChannel := th.CreateDmChannel(th.BasicUser2)
|
|
privateChannel2 := th.createChannelWithAnotherUser(th.BasicTeam, model.CHANNEL_PRIVATE, th.BasicUser2.Id)
|
|
|
|
basicUser3 := th.CreateUser()
|
|
th.LinkUserToTeam(basicUser3, th.BasicTeam)
|
|
basicUser4 := th.CreateUser()
|
|
deactivatedUser := th.CreateUser()
|
|
th.App.UpdateActive(deactivatedUser, false)
|
|
|
|
var err *model.AppError
|
|
_, err = th.App.CreateBot(&model.Bot{
|
|
Username: "bot1",
|
|
OwnerId: basicUser3.Id,
|
|
Description: "a test bot",
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
bot2, err := th.App.CreateBot(&model.Bot{
|
|
Username: "bot2",
|
|
OwnerId: basicUser3.Id,
|
|
Description: "a test bot",
|
|
})
|
|
require.Nil(t, err)
|
|
_, err = th.App.AddUserToTeam(th.BasicTeam.Id, bot2.UserId, basicUser3.Id)
|
|
require.Nil(t, err)
|
|
|
|
bot3, err := th.App.CreateBot(&model.Bot{
|
|
Username: "bot3",
|
|
OwnerId: basicUser3.Id,
|
|
Description: "a test bot",
|
|
})
|
|
require.Nil(t, err)
|
|
_, err = th.App.AddUserToTeam(th.BasicTeam.Id, bot3.UserId, basicUser3.Id)
|
|
require.Nil(t, err)
|
|
err = th.App.RemoveUserFromTeam(th.BasicTeam.Id, bot3.UserId, basicUser3.Id)
|
|
require.Nil(t, err)
|
|
|
|
InviteP := InviteProvider{}
|
|
args := &model.CommandArgs{
|
|
T: func(s string, args ...interface{}) string { return s },
|
|
ChannelId: th.BasicChannel.Id,
|
|
TeamId: th.BasicTeam.Id,
|
|
Session: model.Session{UserId: th.BasicUser.Id, TeamMembers: []*model.TeamMember{{TeamId: th.BasicTeam.Id, Roles: model.TEAM_USER_ROLE_ID}}},
|
|
}
|
|
|
|
userAndWrongChannel := "@" + th.BasicUser2.Username + " wrongchannel1"
|
|
userAndChannel := "@" + th.BasicUser2.Username + " ~" + channel.Name + " "
|
|
userAndDisplayChannel := "@" + th.BasicUser2.Username + " ~" + channel.DisplayName + " "
|
|
userAndPrivateChannel := "@" + th.BasicUser2.Username + " ~" + privateChannel.Name
|
|
userAndDMChannel := "@" + basicUser3.Username + " ~" + dmChannel.Name
|
|
userAndInvalidPrivate := "@" + basicUser3.Username + " ~" + privateChannel2.Name
|
|
deactivatedUserPublicChannel := "@" + deactivatedUser.Username + " ~" + channel.Name
|
|
|
|
groupChannel := th.createChannel(th.BasicTeam, model.CHANNEL_PRIVATE)
|
|
_, err = th.App.AddChannelMember(th.BasicUser.Id, groupChannel, "", "")
|
|
require.Nil(t, err)
|
|
groupChannel.GroupConstrained = model.NewBool(true)
|
|
groupChannel, _ = th.App.UpdateChannel(groupChannel)
|
|
|
|
groupChannelNonUser := "@" + th.BasicUser2.Username + " ~" + groupChannel.Name
|
|
|
|
tests := []struct {
|
|
desc string
|
|
expected string
|
|
msg string
|
|
}{
|
|
{
|
|
desc: "Missing user and channel in the command",
|
|
expected: "api.command_invite.missing_message.app_error",
|
|
msg: "",
|
|
},
|
|
{
|
|
desc: "User added in the current channel",
|
|
expected: "",
|
|
msg: th.BasicUser2.Username,
|
|
},
|
|
{
|
|
desc: "Add user to another channel not the current",
|
|
expected: "api.command_invite.success",
|
|
msg: userAndChannel,
|
|
},
|
|
{
|
|
desc: "try to add a user to a direct channel",
|
|
expected: "api.command_invite.directchannel.app_error",
|
|
msg: userAndDMChannel,
|
|
},
|
|
{
|
|
desc: "Try to add a user to a invalid channel",
|
|
expected: "api.command_invite.channel.error",
|
|
msg: userAndWrongChannel,
|
|
},
|
|
{
|
|
desc: "Try to add a user to an private channel",
|
|
expected: "api.command_invite.success",
|
|
msg: userAndPrivateChannel,
|
|
},
|
|
{
|
|
desc: "Using display channel name which is different form Channel name",
|
|
expected: "api.command_invite.channel.error",
|
|
msg: userAndDisplayChannel,
|
|
},
|
|
{
|
|
desc: "Invalid user to current channel",
|
|
expected: "api.command_invite.missing_user.app_error",
|
|
msg: "@invalidUser123",
|
|
},
|
|
{
|
|
desc: "Invalid user to current channel without @",
|
|
expected: "api.command_invite.missing_user.app_error",
|
|
msg: "invalidUser321",
|
|
},
|
|
{
|
|
desc: "try to add a user which is not part of the team",
|
|
expected: "api.command_invite.user_not_in_team.app_error",
|
|
msg: basicUser4.Username,
|
|
},
|
|
{
|
|
desc: "try to add a user not part of the group to a group channel",
|
|
expected: "api.command_invite.group_constrained_user_denied",
|
|
msg: groupChannelNonUser,
|
|
},
|
|
{
|
|
desc: "try to add a user to a private channel with no permission",
|
|
expected: "api.command_invite.private_channel.app_error",
|
|
msg: userAndInvalidPrivate,
|
|
},
|
|
{
|
|
desc: "try to add a deleted user to a public channel",
|
|
expected: "api.command_invite.missing_user.app_error",
|
|
msg: deactivatedUserPublicChannel,
|
|
},
|
|
{
|
|
desc: "try to add bot to a public channel",
|
|
expected: "api.command_invite.user_not_in_team.app_error",
|
|
msg: "@bot1",
|
|
},
|
|
{
|
|
desc: "add bot to a public channel",
|
|
expected: "",
|
|
msg: "@bot2",
|
|
},
|
|
{
|
|
desc: "try to add bot removed from a team to a public channel",
|
|
expected: "api.command_invite.user_not_in_team.app_error",
|
|
msg: "@bot3",
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
actual := InviteP.DoCommand(th.App, args, test.msg).Text
|
|
assert.Equal(t, test.expected, actual)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestInviteGroup(t *testing.T) {
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
th.BasicTeam.GroupConstrained = model.NewBool(true)
|
|
var err *model.AppError
|
|
_, _ = th.App.AddTeamMember(th.BasicTeam.Id, th.BasicUser.Id)
|
|
_, err = th.App.AddTeamMember(th.BasicTeam.Id, th.BasicUser2.Id)
|
|
require.Nil(t, err)
|
|
th.BasicTeam, _ = th.App.UpdateTeam(th.BasicTeam)
|
|
|
|
privateChannel := th.createChannel(th.BasicTeam, model.CHANNEL_PRIVATE)
|
|
|
|
groupChannelUser1 := "@" + th.BasicUser.Username + " ~" + privateChannel.Name
|
|
groupChannelUser2 := "@" + th.BasicUser2.Username + " ~" + privateChannel.Name
|
|
basicUser3 := th.CreateUser()
|
|
groupChannelUser3 := "@" + basicUser3.Username + " ~" + privateChannel.Name
|
|
|
|
InviteP := InviteProvider{}
|
|
args := &model.CommandArgs{
|
|
T: func(s string, args ...interface{}) string { return s },
|
|
ChannelId: th.BasicChannel.Id,
|
|
TeamId: th.BasicTeam.Id,
|
|
Session: model.Session{UserId: th.BasicUser.Id, TeamMembers: []*model.TeamMember{{TeamId: th.BasicTeam.Id, Roles: model.TEAM_USER_ROLE_ID}}},
|
|
}
|
|
|
|
tests := []struct {
|
|
desc string
|
|
expected string
|
|
msg string
|
|
}{
|
|
{
|
|
desc: "try to add an existing user part of the group to a group channel",
|
|
expected: "api.command_invite.user_already_in_channel.app_error",
|
|
msg: groupChannelUser1,
|
|
},
|
|
{
|
|
desc: "try to add a user part of the group to a group channel",
|
|
expected: "api.command_invite.success",
|
|
msg: groupChannelUser2,
|
|
},
|
|
{
|
|
desc: "try to add a user NOT part of the group to a group channel",
|
|
expected: "api.command_invite.user_not_in_team.app_error",
|
|
msg: groupChannelUser3,
|
|
},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
t.Run(test.desc, func(t *testing.T) {
|
|
actual := InviteP.DoCommand(th.App, args, test.msg).Text
|
|
assert.Equal(t, test.expected, actual)
|
|
})
|
|
}
|
|
}
|