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>
473 lines
13 KiB
Go
473 lines
13 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func (s *MmctlUnitTestSuite) TestChannelUsersAddCmdF() {
|
|
channelArg := teamID + ":" + channelName
|
|
mockTeam := model.Team{Id: teamID}
|
|
mockChannel := model.Channel{Id: channelID, Name: channelName}
|
|
mockUser := model.User{Id: userID, Email: userEmail}
|
|
|
|
s.Run("Not enough command line parameters", func() {
|
|
printer.Clean()
|
|
cmd := &cobra.Command{}
|
|
|
|
// One argument provided.
|
|
err := channelUsersAddCmdF(s.client, cmd, []string{channelArg})
|
|
s.EqualError(err, "not enough arguments")
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
|
|
// No arguments provided.
|
|
err = channelUsersAddCmdF(s.client, cmd, []string{})
|
|
s.EqualError(err, "not enough arguments")
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
s.Run("Add existing user to existing channel", func() {
|
|
printer.Clean()
|
|
cmd := &cobra.Command{}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
GetUserByEmail(context.TODO(), userEmail, "").
|
|
Return(&mockUser, &model.Response{}, nil).
|
|
Times(3)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
AddChannelMember(context.TODO(), channelID, userID).
|
|
Return(&model.ChannelMember{}, &model.Response{}, nil).
|
|
Times(2)
|
|
err := channelUsersAddCmdF(s.client, cmd, []string{channelArg, userEmail})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
s.Run("Add existing user to nonexistent channel", func() {
|
|
printer.Clean()
|
|
cmd := &cobra.Command{}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
// No channel is returned by client.
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, teamID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelUsersAddCmdF(s.client, cmd, []string{channelArg, userEmail})
|
|
s.EqualError(err, fmt.Sprintf("unable to find channel %q", channelArg))
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
s.Run("Add existing user to channel owned by nonexistent team", func() {
|
|
printer.Clean()
|
|
cmd := &cobra.Command{}
|
|
|
|
// No team is returned by client.
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelUsersAddCmdF(s.client, cmd, []string{channelArg, userEmail})
|
|
s.EqualError(err, fmt.Sprintf("unable to find channel %q", channelArg))
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
s.Run("Add multiple users, some nonexistent to existing channel", func() {
|
|
printer.Clean()
|
|
nilUserArg := "nonexistent-user"
|
|
cmd := &cobra.Command{}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
GetUserByUsername(context.TODO(), nilUserArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
GetUser(context.TODO(), nilUserArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
err := channelUsersAddCmdF(s.client, cmd, []string{channelArg, nilUserArg, userEmail})
|
|
s.Require().ErrorContains(err, "unable to find user")
|
|
s.Require().ErrorContains(err, nilUserArg)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 1)
|
|
})
|
|
s.Run("Error adding existing user to existing channel", func() {
|
|
printer.Clean()
|
|
cmd := &cobra.Command{}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
AddChannelMember(context.TODO(), channelID, userID).
|
|
Return(nil, &model.Response{}, errors.New("mock error")).
|
|
Times(1)
|
|
err := channelUsersAddCmdF(s.client, cmd, []string{channelArg, userEmail})
|
|
s.Require().ErrorContains(err, "unable to add")
|
|
s.Require().ErrorContains(err, userEmail)
|
|
s.Require().ErrorContains(err, channelName)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 1)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestChannelUsersRemoveCmd() {
|
|
mockUser := model.User{Id: userID, Email: userEmail}
|
|
mockUser2 := model.User{Id: userID + "2", Email: userID + "2@example.com"}
|
|
mockUser3 := model.User{Id: userID + "3", Email: userID + "3@example.com"}
|
|
argsTeamChannel := teamName + ":" + channelName
|
|
|
|
s.Run("should remove user from channel", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
args := []string{argsTeamChannel, userEmail}
|
|
|
|
foundTeam := &model.Team{
|
|
Id: teamID,
|
|
DisplayName: teamDisplayName,
|
|
Name: teamName,
|
|
}
|
|
|
|
foundChannel := &model.Channel{
|
|
Id: channelID,
|
|
Name: channelName,
|
|
DisplayName: channelDisplayName,
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamName, "").
|
|
Return(foundTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, foundTeam.Id, "").
|
|
Return(foundChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetUserByEmail(context.TODO(), userEmail, "").
|
|
Return(&mockUser, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser.Id).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
err := channelUsersRemoveCmdF(s.client, cmd, args)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("should throw error if both --all-users flag and user email are passed", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("all-users", true, "Remove all users from the indicated channel.")
|
|
args := []string{argsTeamChannel, userEmail}
|
|
|
|
err := channelUsersRemoveCmdF(s.client, cmd, args)
|
|
s.Require().EqualError(err, "individual users must not be specified in conjunction with the --all-users flag")
|
|
})
|
|
|
|
s.Run("should remove all users from channel", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("all-users", true, "Remove all users from the indicated channel.")
|
|
args := []string{argsTeamChannel}
|
|
|
|
foundTeam := &model.Team{
|
|
Id: teamID,
|
|
DisplayName: teamDisplayName,
|
|
Name: teamName,
|
|
}
|
|
|
|
foundChannel := &model.Channel{
|
|
Id: channelID,
|
|
Name: channelName,
|
|
DisplayName: channelDisplayName,
|
|
}
|
|
|
|
mockMember1 := model.ChannelMember{ChannelId: channelID, UserId: mockUser.Id}
|
|
mockMember2 := model.ChannelMember{ChannelId: channelID, UserId: mockUser2.Id}
|
|
mockMember3 := model.ChannelMember{ChannelId: channelID, UserId: mockUser3.Id}
|
|
mockChannelMembers := model.ChannelMembers{mockMember1, mockMember2, mockMember3}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamName, "").
|
|
Return(foundTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetUserByEmail(context.TODO(), userEmail, "").
|
|
Return(&mockUser, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, foundTeam.Id, "").
|
|
Return(foundChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelMembers(context.TODO(), foundChannel.Id, 0, 10000, "").
|
|
Return(mockChannelMembers, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser.Id).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser2.Id).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser3.Id).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
err := channelUsersRemoveCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("should remove multiple users from channel", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
args := []string{argsTeamChannel, userEmail, mockUser2.Email}
|
|
|
|
foundTeam := &model.Team{
|
|
Id: teamID,
|
|
DisplayName: teamDisplayName,
|
|
Name: teamName,
|
|
}
|
|
|
|
foundChannel := &model.Channel{
|
|
Id: channelID,
|
|
Name: channelName,
|
|
DisplayName: channelDisplayName,
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamName, "").
|
|
Return(foundTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, foundTeam.Id, "").
|
|
Return(foundChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetUserByEmail(context.TODO(), userEmail, "").
|
|
Return(&mockUser, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetUserByUsername(context.TODO(), mockUser2.Email, "").
|
|
Return(&mockUser2, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser.Id).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser2.Id).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
err := channelUsersRemoveCmdF(s.client, cmd, args)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("should remove all users from channel throws error", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("all-users", true, "Remove all users from the indicated channel.")
|
|
args := []string{argsTeamChannel}
|
|
|
|
foundTeam := &model.Team{
|
|
Id: teamID,
|
|
DisplayName: teamDisplayName,
|
|
Name: teamName,
|
|
}
|
|
|
|
foundChannel := &model.Channel{
|
|
Id: channelID,
|
|
Name: channelName,
|
|
DisplayName: channelDisplayName,
|
|
}
|
|
|
|
mockMember1 := model.ChannelMember{ChannelId: channelID, UserId: mockUser.Id}
|
|
mockChannelMembers := model.ChannelMembers{mockMember1}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamName, "").
|
|
Return(foundTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, foundTeam.Id, "").
|
|
Return(foundChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelMembers(context.TODO(), foundChannel.Id, 0, 10000, "").
|
|
Return(mockChannelMembers, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser.Id).
|
|
Return(&model.Response{StatusCode: http.StatusNotFound}, errors.New("mock error")).
|
|
Times(1)
|
|
|
|
err := channelUsersRemoveCmdF(s.client, cmd, args)
|
|
s.Require().ErrorContains(err, "unable to remove")
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 1)
|
|
})
|
|
|
|
s.Run("should remove user from channel throws error", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
args := []string{argsTeamChannel, userEmail}
|
|
|
|
foundTeam := &model.Team{
|
|
Id: teamID,
|
|
DisplayName: teamDisplayName,
|
|
Name: teamName,
|
|
}
|
|
|
|
foundChannel := &model.Channel{
|
|
Id: channelID,
|
|
Name: channelName,
|
|
DisplayName: channelDisplayName,
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamName, "").
|
|
Return(foundTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelName, foundTeam.Id, "").
|
|
Return(foundChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RemoveUserFromChannel(context.TODO(), foundChannel.Id, mockUser.Id).
|
|
Return(&model.Response{StatusCode: http.StatusNotFound}, errors.New("mock error")).
|
|
Times(1)
|
|
|
|
err := channelUsersRemoveCmdF(s.client, cmd, args)
|
|
s.Require().ErrorContains(err, "unable to remove")
|
|
s.Require().ErrorContains(err, userEmail)
|
|
s.Require().ErrorContains(err, channelName)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 1)
|
|
})
|
|
}
|