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>
1580 lines
41 KiB
Go
1580 lines
41 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"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) TestListLdapGroupsCmd() {
|
|
s.Run("Failure getting Ldap Groups", func() {
|
|
printer.Clean()
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetLdapGroups(context.TODO()).
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := listLdapGroupsCmdF(s.client, &cobra.Command{}, []string{})
|
|
s.Require().Equal(mockError, err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("List several groups", func() {
|
|
printer.Clean()
|
|
mockList := []*model.Group{
|
|
{DisplayName: "Group1"},
|
|
{DisplayName: "Group2"},
|
|
{DisplayName: "Group3"},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetLdapGroups(context.TODO()).
|
|
Return(mockList, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := listLdapGroupsCmdF(s.client, &cobra.Command{}, []string{})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(printer.GetLines(), 3)
|
|
for i, v := range mockList {
|
|
s.Require().Equal(v, printer.GetLines()[i])
|
|
}
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestTeamGroupEnableCmd() {
|
|
s.Run("Enable unexisting team", func() {
|
|
printer.Clean()
|
|
|
|
arg := "teamID"
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), arg, "").
|
|
Return(nil, &model.Response{}, errors.New("")).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), arg, "").
|
|
Return(nil, &model.Response{}, errors.New("")).
|
|
Times(1)
|
|
|
|
err := teamGroupEnableCmdF(s.client, &cobra.Command{}, []string{arg})
|
|
s.Require().EqualError(err, "Unable to find team '"+arg+"'")
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Error while getting the team groups", func() {
|
|
printer.Clean()
|
|
|
|
arg := "teamID"
|
|
mockTeam := model.Team{Id: arg}
|
|
mockError := errors.New("mock error")
|
|
groupOpts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), arg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByTeam(context.TODO(), mockTeam.Id, groupOpts).
|
|
Return(nil, 0, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := teamGroupEnableCmdF(s.client, &cobra.Command{}, []string{arg})
|
|
s.Require().Equal(mockError, err)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("No groups on team", func() {
|
|
printer.Clean()
|
|
|
|
arg := "teamID"
|
|
mockTeam := model.Team{Id: arg}
|
|
groupOpts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), arg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByTeam(context.TODO(), mockTeam.Id, groupOpts).
|
|
Return([]*model.GroupWithSchemeAdmin{}, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupEnableCmdF(s.client, &cobra.Command{}, []string{arg})
|
|
s.Require().EqualError(err, "Team '"+arg+"' has no groups associated. It cannot be group-constrained")
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Error patching the team", func() {
|
|
printer.Clean()
|
|
|
|
arg := "teamID"
|
|
mockTeam := model.Team{Id: arg}
|
|
mockError := errors.New("mock error")
|
|
groupOpts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
teamPatch := model.TeamPatch{GroupConstrained: model.NewPointer(true)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), arg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByTeam(context.TODO(), mockTeam.Id, groupOpts).
|
|
Return([]*model.GroupWithSchemeAdmin{{}}, 1, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchTeam(context.TODO(), mockTeam.Id, &teamPatch).
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := teamGroupEnableCmdF(s.client, &cobra.Command{}, []string{arg})
|
|
s.Require().Equal(mockError, err)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Successfully enable group", func() {
|
|
printer.Clean()
|
|
|
|
arg := "teamID"
|
|
mockTeam := model.Team{Id: arg}
|
|
groupOpts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
teamPatch := model.TeamPatch{GroupConstrained: model.NewPointer(true)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), arg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByTeam(context.TODO(), mockTeam.Id, groupOpts).
|
|
Return([]*model.GroupWithSchemeAdmin{{}}, 1, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchTeam(context.TODO(), mockTeam.Id, &teamPatch).
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupEnableCmdF(s.client, &cobra.Command{}, []string{arg})
|
|
s.Require().NoError(err)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestTeamGroupDisableCmd() {
|
|
s.Run("Disable existing team", func() {
|
|
printer.Clean()
|
|
teamArg := "example-team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
teamPatch := model.TeamPatch{GroupConstrained: model.NewPointer(false)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchTeam(context.TODO(), teamArg, &teamPatch).
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupDisableCmdF(s.client, &cobra.Command{}, []string{teamArg})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("Disable nonexisting team", func() {
|
|
printer.Clean()
|
|
teamArg := "example-team-id"
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupDisableCmdF(s.client, &cobra.Command{}, []string{teamArg})
|
|
s.Require().NotNil(err)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find team '"+teamArg+"'")
|
|
})
|
|
|
|
s.Run("Error response from PatchTeam", func() {
|
|
printer.Clean()
|
|
teamArg := "example-team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
teamPatch := model.TeamPatch{GroupConstrained: model.NewPointer(false)}
|
|
mockError := errors.New("patchteam error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchTeam(context.TODO(), teamArg, &teamPatch).
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := teamGroupDisableCmdF(s.client, &cobra.Command{}, []string{teamArg})
|
|
s.Require().NotNil(err)
|
|
s.Len(printer.GetLines(), 0)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, mockError.Error())
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestChannelGroupListCmd() {
|
|
s.Run("List groups for existing channel and team, when a single group exists", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
groupName := "group-name"
|
|
|
|
mockTeam := model.Team{Id: teamID}
|
|
mockChannel := model.Channel{Id: channelID}
|
|
mockGroup := &model.GroupWithSchemeAdmin{Group: model.Group{Name: model.NewPointer(groupName)}}
|
|
mockGroups := []*model.GroupWithSchemeAdmin{mockGroup}
|
|
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 9999,
|
|
},
|
|
}
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelID, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], mockGroup)
|
|
})
|
|
|
|
s.Run("List groups for existing channel and team, when multiple groups exist", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
mockTeam := model.Team{Id: teamID}
|
|
mockChannel := model.Channel{Id: channelID}
|
|
mockGroups := []*model.GroupWithSchemeAdmin{
|
|
{Group: model.Group{Name: model.NewPointer("group1")}},
|
|
{Group: model.Group{Name: model.NewPointer("group2")}},
|
|
}
|
|
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 9999,
|
|
},
|
|
}
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelID, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 2)
|
|
s.Require().Equal(printer.GetLines()[0], mockGroups[0])
|
|
s.Require().Equal(printer.GetLines()[1], mockGroups[1])
|
|
})
|
|
|
|
s.Run("List groups for existing channel and team, when no groups exist", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
mockTeam := model.Team{Id: teamID}
|
|
mockChannel := model.Channel{Id: channelID}
|
|
mockGroups := []*model.GroupWithSchemeAdmin{}
|
|
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 9999,
|
|
},
|
|
}
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelID, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("List groups for a nonexistent channel", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
mockTeam := model.Team{Id: teamID}
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.Require().NotNil(err)
|
|
s.EqualError(err, "Unable to find channel '"+cmdArg+"'")
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("List groups for a nonexistent team", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
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 := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.Require().NotNil(err)
|
|
s.EqualError(err, "Unable to find channel '"+cmdArg+"'")
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("Return error when GetGroupsByChannel returns error", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
mockTeam := model.Team{Id: teamID}
|
|
mockChannel := model.Channel{Id: channelID}
|
|
mockError := errors.New("mock error")
|
|
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 9999,
|
|
},
|
|
}
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelID, *groupOpts).
|
|
Return(nil, 0, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.Require().Equal(err, mockError)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("Return error when GetChannelByNameIncludeDeleted returns error", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
mockTeam := model.Team{Id: teamID}
|
|
mockError := errors.New("mock error")
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelID, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.EqualError(err, "Unable to find channel '"+cmdArg+"'")
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
|
|
s.Run("Return error when GetTeam returns error", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "team-id"
|
|
channelID := "channel-id"
|
|
|
|
mockError := errors.New("mock error")
|
|
|
|
cmdArg := teamID + ":" + channelID
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamID, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupListCmdF(s.client, &cobra.Command{}, []string{cmdArg})
|
|
s.EqualError(err, "Unable to find channel '"+cmdArg+"'")
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestTeamGroupListCmd() {
|
|
s.Run("Team group list returns error when passing a nonexistent team", func() {
|
|
printer.Clean()
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), "team1", "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), "team1", "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
err := teamGroupListCmdF(s.client, cmd, []string{"team1"})
|
|
|
|
s.Require().NotNil(err)
|
|
s.Require().Equal(err.Error(), "Unable to find team 'team1'")
|
|
})
|
|
|
|
s.Run("Team group list return error when GetGroupsByTeam returns error", func() {
|
|
printer.Clean()
|
|
groupID := "group1"
|
|
groupID2 := "group2"
|
|
mockError := errors.New("get groups by team error")
|
|
|
|
group1 := model.GroupWithSchemeAdmin{Group: model.Group{Id: groupID, DisplayName: "DisplayName1"}}
|
|
group2 := model.GroupWithSchemeAdmin{Group: model.Group{Id: groupID2, DisplayName: "DisplayName2"}}
|
|
|
|
groups := []*model.GroupWithSchemeAdmin{
|
|
&group1,
|
|
&group2,
|
|
}
|
|
|
|
mockTeam := model.Team{Id: "team1"}
|
|
groupOpts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 9999,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), "team1", "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByTeam(context.TODO(), "team1", groupOpts).
|
|
Return(groups, 2, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
err := teamGroupListCmdF(s.client, cmd, []string{"team1"})
|
|
|
|
s.Require().NotNil(err)
|
|
s.Require().Equal(err, mockError)
|
|
})
|
|
|
|
s.Run("Team group list should print group in console on success", func() {
|
|
printer.Clean()
|
|
groupID := "group1"
|
|
groupID2 := "group2"
|
|
group1 := model.GroupWithSchemeAdmin{Group: model.Group{Id: groupID, DisplayName: "DisplayName1"}}
|
|
group2 := model.GroupWithSchemeAdmin{Group: model.Group{Id: groupID2, DisplayName: "DisplayName2"}}
|
|
|
|
groups := []*model.GroupWithSchemeAdmin{
|
|
&group1,
|
|
&group2,
|
|
}
|
|
|
|
mockTeam := model.Team{Id: "team1"}
|
|
groupOpts := model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 9999,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), "team1", "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByTeam(context.TODO(), "team1", groupOpts).
|
|
Return(groups, 2, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
err := teamGroupListCmdF(s.client, cmd, []string{"team1"})
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 2)
|
|
s.Require().Equal(printer.GetLines()[0], &group1)
|
|
s.Require().Equal(printer.GetLines()[1], &group2)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestTeamGroupStatusCmd() {
|
|
s.Run("Should fail when team is not found", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
arg := teamID
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
|
|
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 := teamGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().EqualError(err, "Unable to find team '"+args[0]+"'")
|
|
})
|
|
|
|
s.Run("Should show valid response when group constraints status for a team is not present", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
arg := teamID
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
team := &model.Team{Id: teamID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], "Disabled")
|
|
})
|
|
|
|
s.Run("Should show valid response when group constraints status for a team is enabled", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
arg := teamID
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
team := &model.Team{Id: teamID, GroupConstrained: model.NewPointer(true)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], "Enabled")
|
|
})
|
|
|
|
s.Run("Should show valid response when group constraints status for a team is disabled", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
arg := teamID
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
team := &model.Team{Id: teamID, GroupConstrained: model.NewPointer(false)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := teamGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], "Disabled")
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestChannelGroupStatusCmd() {
|
|
s.Run("Should fail to get group constrain status of a channel when team is not found", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
channelID := "channelID"
|
|
arg := strings.Join([]string{teamID, channelID}, ":")
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
|
|
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 := channelGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().EqualError(err, "Unable to find channel '"+args[0]+"'")
|
|
})
|
|
|
|
s.Run("Should fail to get group constrain status of a channel when channel is not found", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
channelID := "channelID"
|
|
arg := strings.Join([]string{teamID, channelID}, ":")
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
|
|
team := &model.Team{Id: teamID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelID, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().EqualError(err, "Unable to find channel '"+args[0]+"'")
|
|
})
|
|
|
|
s.Run("Should get valid response when channel's group constrain status is enabled", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
channelID := "channelID"
|
|
arg := strings.Join([]string{teamID, channelID}, ":")
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
|
|
team := &model.Team{Id: teamID}
|
|
channel := &model.Channel{Id: channelID, GroupConstrained: model.NewPointer(true)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(channel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], "Enabled")
|
|
})
|
|
|
|
s.Run("Should get valid response when channel's group constrain status is disabled", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
channelID := "channelID"
|
|
arg := strings.Join([]string{teamID, channelID}, ":")
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
|
|
team := &model.Team{Id: teamID}
|
|
channel := &model.Channel{Id: channelID, GroupConstrained: model.NewPointer(false)}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(channel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], "Disabled")
|
|
})
|
|
|
|
s.Run("Should get valid response when channel's group constrain status is not present", func() {
|
|
printer.Clean()
|
|
|
|
teamID := "teamID"
|
|
channelID := "channelID"
|
|
arg := strings.Join([]string{teamID, channelID}, ":")
|
|
args := []string{arg}
|
|
cmd := &cobra.Command{}
|
|
|
|
team := &model.Team{Id: teamID}
|
|
channel := &model.Channel{Id: channelID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamID, "").
|
|
Return(team, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelID, teamID, "").
|
|
Return(channel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupStatusCmdF(s.client, cmd, args)
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], "Disabled")
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestChannelGroupEnableCmdF() {
|
|
s.Run("Enable group constraints with existing team and channel", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
group := &model.GroupWithSchemeAdmin{Group: model.Group{Name: model.NewPointer("group-name")}}
|
|
mockGroups := []*model.GroupWithSchemeAdmin{group}
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelPart, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchChannel(context.TODO(), channelPart, &model.ChannelPatch{GroupConstrained: model.NewPointer(true)}).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Enable group constraints with GetTeam error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
channelPart := "channel-id"
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Enable group constraints with GetChannelByNameIncludeDeleted error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelPart, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Enable group constraints with GetGroupsByChannel error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelPart, *groupOpts).
|
|
Return(nil, 0, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, mockError.Error())
|
|
})
|
|
|
|
s.Run("Enable group constraints with PatchChannel error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
group := &model.GroupWithSchemeAdmin{Group: model.Group{Name: model.NewPointer("group-name")}}
|
|
mockGroups := []*model.GroupWithSchemeAdmin{group}
|
|
mockError := errors.New("mock error")
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelPart, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchChannel(context.TODO(), channelPart, &model.ChannelPatch{GroupConstrained: model.NewPointer(true)}).
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, mockError.Error())
|
|
})
|
|
|
|
s.Run("Enable group constraints with no associated groups", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockGroups := []*model.GroupWithSchemeAdmin{}
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelPart, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Channel '"+channelArg+"' has no groups associated. It cannot be group-constrained")
|
|
})
|
|
|
|
s.Run("Enable group constraints with nonexistent team", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
channelPart := "channel-id"
|
|
channelArg := teamArg + ":" + channelPart
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Enable group constraints with nonexistent channel", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
channelArg := teamArg + ":" + channelPart
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelPart, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Enable group constraints with GetChannelByNameIncludeDeleted error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
group := &model.GroupWithSchemeAdmin{Group: model.Group{Name: model.NewPointer("group-name")}}
|
|
mockGroups := []*model.GroupWithSchemeAdmin{group}
|
|
mockError := errors.New("mock error")
|
|
groupOpts := &model.GroupSearchOpts{
|
|
PageOpts: &model.PageOpts{
|
|
Page: 0,
|
|
PerPage: 10,
|
|
},
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelPart, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetGroupsByChannel(context.TODO(), channelPart, *groupOpts).
|
|
Return(mockGroups, 0, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchChannel(context.TODO(), channelPart, &model.ChannelPatch{GroupConstrained: model.NewPointer(true)}).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupEnableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestChannelGroupDisableCmdF() {
|
|
s.Run("Disable group constraints with existing team and channel", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := strings.Join([]string{teamArg, channelPart}, ":")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchChannel(context.TODO(), channelPart, &model.ChannelPatch{GroupConstrained: model.NewPointer(false)}).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Disable group constraints with nonexistent team", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
channelPart := "channel-id"
|
|
channelArg := strings.Join([]string{teamArg, channelPart}, ":")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Disable group constraints with nonexistent channel", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
channelArg := strings.Join([]string{teamArg, channelPart}, ":")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelPart, "").
|
|
Return(nil, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Disable group constraints with GetTeam error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchChannel(context.TODO(), channelPart, &model.ChannelPatch{GroupConstrained: model.NewPointer(false)}).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Disable group constraints with GetTeamByName error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
channelPart := "channel-id"
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeamByName(context.TODO(), teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Disable group constraints with GetChannelByNameIncludeDeleted error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelPart, "").
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, "Unable to find channel '"+channelArg+"'")
|
|
})
|
|
|
|
s.Run("Disable group constraints with PatchChannel error", func() {
|
|
printer.Clean()
|
|
|
|
teamArg := "team-id"
|
|
mockTeam := model.Team{Id: teamArg}
|
|
channelPart := "channel-id"
|
|
mockChannel := model.Channel{Id: channelPart}
|
|
channelArg := teamArg + ":" + channelPart
|
|
mockError := errors.New("mock error")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetTeam(context.TODO(), teamArg, "").
|
|
Return(&mockTeam, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannelByNameIncludeDeleted(context.TODO(), channelPart, teamArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PatchChannel(context.TODO(), channelPart, &model.ChannelPatch{GroupConstrained: model.NewPointer(false)}).
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
err := channelGroupDisableCmdF(s.client, &cobra.Command{}, []string{channelArg})
|
|
s.Require().NotNil(err)
|
|
s.Require().Len(printer.GetLines(), 0)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
s.EqualError(err, mockError.Error())
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestUserGroupRestoreCmd() {
|
|
s.Run("User group restore command restores deleted user group", func() {
|
|
printer.Clean()
|
|
|
|
s.client.
|
|
EXPECT().
|
|
RestoreGroup(context.TODO(), "groupId", "").
|
|
Return(nil, &model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
err := userGroupRestoreCmdF(s.client, cmd, []string{"groupId"})
|
|
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
})
|
|
|
|
s.Run("User group restore command restores deleted user group", func() {
|
|
printer.Clean()
|
|
|
|
mockError := errors.New("no group found")
|
|
s.client.
|
|
EXPECT().
|
|
RestoreGroup(context.TODO(), "groupId", "").
|
|
Return(nil, &model.Response{StatusCode: http.StatusNotFound}, mockError).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
err := userGroupRestoreCmdF(s.client, cmd, []string{"groupId"})
|
|
|
|
s.Require().NotNil(err)
|
|
s.Require().Equal(mockError, err)
|
|
})
|
|
}
|