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>
105 lines
3.5 KiB
Go
105 lines
3.5 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/client"
|
|
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func (s *MmctlE2ETestSuite) TestAssignUsersCmd() {
|
|
s.SetupEnterpriseTestHelper().InitBasic()
|
|
|
|
user, appErr := s.th.App.CreateUser(s.th.Context, &model.User{Email: s.th.GenerateTestEmail(), Username: model.NewUsername(), Password: model.NewId()})
|
|
s.Require().Nil(appErr)
|
|
|
|
s.Run("MM-T3721 Should not allow normal user to assign a role", func() {
|
|
printer.Clean()
|
|
|
|
err := assignUsersCmdF(s.th.Client, &cobra.Command{}, []string{model.SystemAdminRoleId, user.Email})
|
|
s.Require().Error(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("MM-T3722 Assigning a user to a non-existent role", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := assignUsersCmdF(c, &cobra.Command{}, []string{"not_a_role", user.Email})
|
|
s.Require().Error(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("Assigning non existen user to a role", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := assignUsersCmdF(c, &cobra.Command{}, []string{model.SystemManagerRoleId, "non_existent_user"})
|
|
s.Require().Error(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 1)
|
|
s.Require().Contains(printer.GetErrorLines()[0], "Couldn't find user 'non_existent_user'.")
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("MM-T3648 Assigning a user to a role", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := assignUsersCmdF(c, &cobra.Command{}, []string{model.SystemManagerRoleId, user.Email})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
|
|
roles := user.Roles
|
|
|
|
u, err2 := s.th.App.GetUser(user.Id)
|
|
s.Require().Nil(err2)
|
|
s.Require().True(u.IsInRole(model.SystemManagerRoleId))
|
|
|
|
_, err2 = s.th.App.UpdateUserRoles(s.th.Context, user.Id, roles, false)
|
|
s.Require().Nil(err2)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestUnassignUsersCmd() {
|
|
s.SetupEnterpriseTestHelper().InitBasic()
|
|
|
|
user, appErr := s.th.App.CreateUser(s.th.Context, &model.User{Email: s.th.GenerateTestEmail(), Username: model.NewUsername(), Password: model.NewId()})
|
|
s.Require().Nil(appErr)
|
|
|
|
s.Run("MM-T3965 Should not allow normal user to unassign a user from a role", func() {
|
|
printer.Clean()
|
|
|
|
err := unassignUsersCmdF(s.th.Client, &cobra.Command{}, []string{model.SystemAdminRoleId, s.th.SystemAdminUser.Email})
|
|
s.Require().Error(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("MM-T3964 Unassign a user from a role", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
user.Roles = user.Roles + "," + model.SystemManagerRoleId
|
|
_, appErr = s.th.App.UpdateUser(s.th.Context, user, false)
|
|
s.Require().Nil(appErr)
|
|
defer func() {
|
|
user.Roles = model.SystemUserRoleId
|
|
_, appErr := s.th.App.UpdateUser(s.th.Context, user, false)
|
|
s.Require().Nil(appErr)
|
|
}()
|
|
|
|
err := unassignUsersCmdF(c, &cobra.Command{}, []string{model.SystemManagerRoleId, user.Email})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
|
|
u, err2 := s.th.App.GetUser(user.Id)
|
|
s.Require().Nil(err2)
|
|
s.Require().False(u.IsInRole(model.SystemManagerRoleId))
|
|
})
|
|
}
|