ec1f89217a
Full Mattermost server source with integrated Community Enterprise features. Includes vendor directory for offline/air-gapped builds. Structure: - enterprise-impl/: Enterprise feature implementations - enterprise-community/: Init files that register implementations - enterprise/: Bridge imports (community_imports.go) - vendor/: All dependencies for offline builds Build (online): go build ./cmd/mattermost Build (offline/air-gapped): go build -mod=vendor ./cmd/mattermost 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
469 lines
12 KiB
Go
469 lines
12 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/public/model"
|
|
)
|
|
|
|
func TestGetGroup(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
group, err := th.App.GetGroup(group.Id, nil, nil)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, group)
|
|
|
|
nilGroup, err := th.App.GetGroup(model.NewId(), nil, nil)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, nilGroup)
|
|
|
|
group, err = th.App.GetGroup(group.Id, &model.GetGroupOpts{IncludeMemberCount: false}, nil)
|
|
require.Nil(t, err)
|
|
require.Nil(t, group.MemberCount)
|
|
|
|
group, err = th.App.GetGroup(group.Id, &model.GetGroupOpts{IncludeMemberCount: true}, nil)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, group.MemberCount)
|
|
}
|
|
|
|
func TestGetGroupByRemoteID(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
g, err := th.App.GetGroupByRemoteID(*group.RemoteId, model.GroupSourceLdap)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.GetGroupByRemoteID(model.NewId(), model.GroupSourceLdap)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, g)
|
|
}
|
|
|
|
func TestGetGroupsByType(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
th.CreateGroup()
|
|
th.CreateGroup()
|
|
th.CreateGroup()
|
|
|
|
groups, err := th.App.GetGroupsBySource(model.GroupSourceLdap)
|
|
require.Nil(t, err)
|
|
require.NotEmpty(t, groups)
|
|
|
|
groups, err = th.App.GetGroupsBySource(model.GroupSource("blah"))
|
|
require.Nil(t, err)
|
|
require.Empty(t, groups)
|
|
}
|
|
|
|
func TestCreateGroup(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
|
|
id := model.NewId()
|
|
group := &model.Group{
|
|
DisplayName: "dn_" + id,
|
|
Name: model.NewPointer("name" + id),
|
|
Source: model.GroupSourceLdap,
|
|
RemoteId: model.NewPointer(model.NewId()),
|
|
}
|
|
|
|
g, err := th.App.CreateGroup(group)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.CreateGroup(group)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, g)
|
|
|
|
t.Run("should check if the group mention is in use as a username", func(t *testing.T) {
|
|
user := th.CreateUser()
|
|
usernameGroup := &model.Group{
|
|
DisplayName: "dn_" + model.NewId(),
|
|
Name: &user.Username,
|
|
Source: model.GroupSourceLdap,
|
|
RemoteId: model.NewPointer(model.NewId()),
|
|
}
|
|
g, err = th.App.CreateGroup(usernameGroup)
|
|
require.NotNil(t, err)
|
|
require.Equal(t, "app.group.username_conflict", err.Id)
|
|
require.Nil(t, g)
|
|
})
|
|
}
|
|
|
|
func TestUpdateGroup(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
group.DisplayName = model.NewId()
|
|
|
|
g, err := th.App.UpdateGroup(group)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
user := th.CreateUser()
|
|
g.Name = &user.Username
|
|
g, err = th.App.UpdateGroup(g)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, g)
|
|
}
|
|
|
|
func TestDeleteGroup(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
g, err := th.App.DeleteGroup(group.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.DeleteGroup(group.Id)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, g)
|
|
}
|
|
|
|
func TestUndeleteGroup(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
g, err := th.App.DeleteGroup(group.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.RestoreGroup(group.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.RestoreGroup(group.Id)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, g)
|
|
}
|
|
|
|
func TestCreateOrRestoreGroupMember(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
g, err := th.App.UpsertGroupMember(group.Id, th.BasicUser.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.UpsertGroupMember(group.Id, th.BasicUser.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
}
|
|
|
|
func TestDeleteGroupMember(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
groupMember, err := th.App.UpsertGroupMember(group.Id, th.BasicUser.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, groupMember)
|
|
|
|
groupMember, err = th.App.DeleteGroupMember(groupMember.GroupId, groupMember.UserId)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, groupMember)
|
|
|
|
groupMember, err = th.App.DeleteGroupMember(groupMember.GroupId, groupMember.UserId)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, groupMember)
|
|
}
|
|
|
|
func TestUpsertGroupSyncable(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
groupSyncable := model.NewGroupTeam(group.Id, th.BasicTeam.Id, false)
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
// can update again without error
|
|
gs, err = th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
gs, err = th.App.DeleteGroupSyncable(gs.GroupId, gs.SyncableId, gs.Type)
|
|
require.Nil(t, err)
|
|
require.NotEqual(t, int64(0), gs.DeleteAt)
|
|
|
|
// Un-deleting works
|
|
gs.DeleteAt = 0
|
|
gs, err = th.App.UpsertGroupSyncable(gs)
|
|
require.Nil(t, err)
|
|
require.Equal(t, int64(0), gs.DeleteAt)
|
|
}
|
|
|
|
func TestUpsertGroupSyncableTeamGroupConstrained(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
|
|
group1 := th.CreateGroup()
|
|
group2 := th.CreateGroup()
|
|
|
|
team := th.CreateTeam()
|
|
team.GroupConstrained = model.NewPointer(true)
|
|
team, err := th.App.UpdateTeam(team)
|
|
require.Nil(t, err)
|
|
_, err = th.App.UpsertGroupSyncable(model.NewGroupTeam(group1.Id, team.Id, false))
|
|
require.Nil(t, err)
|
|
|
|
channel := th.CreateChannel(th.Context, team)
|
|
|
|
_, err = th.App.UpsertGroupSyncable(model.NewGroupChannel(group2.Id, channel.Id, false))
|
|
require.NotNil(t, err)
|
|
require.Equal(t, err.Id, "group_not_associated_to_synced_team")
|
|
|
|
gs, err := th.App.GetGroupSyncable(group2.Id, channel.Id, model.GroupSyncableTypeChannel)
|
|
require.Nil(t, gs)
|
|
require.NotNil(t, err)
|
|
|
|
_, err = th.App.UpsertGroupSyncable(model.NewGroupChannel(group1.Id, channel.Id, false))
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestGetGroupSyncable(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
groupSyncable := model.NewGroupTeam(group.Id, th.BasicTeam.Id, false)
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
gs, err = th.App.GetGroupSyncable(group.Id, th.BasicTeam.Id, model.GroupSyncableTypeTeam)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
}
|
|
|
|
func TestGetGroupSyncables(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
// Create a group team
|
|
groupSyncable := model.NewGroupTeam(group.Id, th.BasicTeam.Id, false)
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
groupTeams, err := th.App.GetGroupSyncables(group.Id, model.GroupSyncableTypeTeam)
|
|
require.Nil(t, err)
|
|
|
|
require.NotEmpty(t, groupTeams)
|
|
}
|
|
|
|
func TestDeleteGroupSyncable(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
groupChannel := model.NewGroupChannel(group.Id, th.BasicChannel.Id, false)
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupChannel)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
gs, err = th.App.DeleteGroupSyncable(group.Id, th.BasicChannel.Id, model.GroupSyncableTypeChannel)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
gs, err = th.App.DeleteGroupSyncable(group.Id, th.BasicChannel.Id, model.GroupSyncableTypeChannel)
|
|
require.NotNil(t, err)
|
|
require.Nil(t, gs)
|
|
}
|
|
|
|
func TestGetGroupsByChannel(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
// Create a group channel
|
|
groupSyncable := &model.GroupSyncable{
|
|
GroupId: group.Id,
|
|
AutoAdd: false,
|
|
SyncableId: th.BasicChannel.Id,
|
|
Type: model.GroupSyncableTypeChannel,
|
|
}
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
opts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 60,
|
|
},
|
|
}
|
|
|
|
groups, _, err := th.App.GetGroupsByChannel(th.BasicChannel.Id, opts)
|
|
require.Nil(t, err)
|
|
require.ElementsMatch(t, []*model.GroupWithSchemeAdmin{{Group: *group, SchemeAdmin: model.NewPointer(false)}}, groups)
|
|
require.NotNil(t, groups[0].SchemeAdmin)
|
|
|
|
groups, _, err = th.App.GetGroupsByChannel(model.NewId(), opts)
|
|
require.Nil(t, err)
|
|
require.Empty(t, groups)
|
|
}
|
|
|
|
func TestGetGroupsAssociatedToChannelsByTeam(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
// Create a group channel
|
|
groupSyncable := &model.GroupSyncable{
|
|
GroupId: group.Id,
|
|
AutoAdd: false,
|
|
SyncableId: th.BasicChannel.Id,
|
|
Type: model.GroupSyncableTypeChannel,
|
|
}
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
opts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 60,
|
|
},
|
|
}
|
|
|
|
groups, err := th.App.GetGroupsAssociatedToChannelsByTeam(th.BasicTeam.Id, opts)
|
|
require.Nil(t, err)
|
|
|
|
assert.Equal(t, map[string][]*model.GroupWithSchemeAdmin{
|
|
th.BasicChannel.Id: {
|
|
{Group: *group, SchemeAdmin: model.NewPointer(false)},
|
|
},
|
|
}, groups)
|
|
require.NotNil(t, groups[th.BasicChannel.Id][0].SchemeAdmin)
|
|
|
|
groups, err = th.App.GetGroupsAssociatedToChannelsByTeam(model.NewId(), opts)
|
|
require.Nil(t, err)
|
|
require.Empty(t, groups)
|
|
}
|
|
|
|
func TestGetGroupsByTeam(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
// Create a group team
|
|
groupSyncable := &model.GroupSyncable{
|
|
GroupId: group.Id,
|
|
AutoAdd: false,
|
|
SyncableId: th.BasicTeam.Id,
|
|
Type: model.GroupSyncableTypeTeam,
|
|
}
|
|
|
|
gs, err := th.App.UpsertGroupSyncable(groupSyncable)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, gs)
|
|
|
|
groups, _, err := th.App.GetGroupsByTeam(th.BasicTeam.Id, model.GroupSearchOpts{})
|
|
require.Nil(t, err)
|
|
require.ElementsMatch(t, []*model.GroupWithSchemeAdmin{{Group: *group, SchemeAdmin: model.NewPointer(false)}}, groups)
|
|
require.NotNil(t, groups[0].SchemeAdmin)
|
|
|
|
groups, _, err = th.App.GetGroupsByTeam(model.NewId(), model.GroupSearchOpts{})
|
|
require.Nil(t, err)
|
|
require.Empty(t, groups)
|
|
}
|
|
|
|
func TestGetGroups(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t)
|
|
defer th.TearDown()
|
|
group := th.CreateGroup()
|
|
|
|
groups, err := th.App.GetGroups(0, 60, model.GroupSearchOpts{}, nil)
|
|
require.Nil(t, err)
|
|
require.ElementsMatch(t, []*model.Group{group}, groups)
|
|
}
|
|
|
|
func TestUserIsInAdminRoleGroup(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic()
|
|
defer th.TearDown()
|
|
group1 := th.CreateGroup()
|
|
group2 := th.CreateGroup()
|
|
|
|
g, err := th.App.UpsertGroupMember(group1.Id, th.BasicUser.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
g, err = th.App.UpsertGroupMember(group2.Id, th.BasicUser.Id)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, g)
|
|
|
|
_, err = th.App.UpsertGroupSyncable(&model.GroupSyncable{
|
|
GroupId: group1.Id,
|
|
AutoAdd: false,
|
|
SyncableId: th.BasicTeam.Id,
|
|
Type: model.GroupSyncableTypeTeam,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
groupSyncable2, err := th.App.UpsertGroupSyncable(&model.GroupSyncable{
|
|
GroupId: group2.Id,
|
|
AutoAdd: false,
|
|
SyncableId: th.BasicTeam.Id,
|
|
Type: model.GroupSyncableTypeTeam,
|
|
})
|
|
require.Nil(t, err)
|
|
|
|
// no syncables are set to scheme admin true, so this returns false
|
|
actual, err := th.App.UserIsInAdminRoleGroup(th.BasicUser.Id, th.BasicTeam.Id, model.GroupSyncableTypeTeam)
|
|
require.Nil(t, err)
|
|
require.False(t, actual)
|
|
|
|
// set a syncable to be scheme admins
|
|
groupSyncable2.SchemeAdmin = true
|
|
_, err = th.App.UpdateGroupSyncable(groupSyncable2)
|
|
require.Nil(t, err)
|
|
|
|
// a syncable is set to scheme admin true, so this returns true
|
|
actual, err = th.App.UserIsInAdminRoleGroup(th.BasicUser.Id, th.BasicTeam.Id, model.GroupSyncableTypeTeam)
|
|
require.Nil(t, err)
|
|
require.True(t, actual)
|
|
|
|
// delete the syncable, should be false again
|
|
_, appErr := th.App.DeleteGroupSyncable(group2.Id, th.BasicTeam.Id, model.GroupSyncableTypeTeam)
|
|
require.Nil(t, appErr)
|
|
actual, err = th.App.UserIsInAdminRoleGroup(th.BasicUser.Id, th.BasicTeam.Id, model.GroupSyncableTypeTeam)
|
|
require.Nil(t, err)
|
|
require.False(t, actual)
|
|
}
|