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>
81 lines
2.3 KiB
Go
81 lines
2.3 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
|
|
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func (s *MmctlUnitTestSuite) TestSamlAuthDataReset() {
|
|
s.Run("Reset auth data without confirmation returns an error", func() {
|
|
cmd := &cobra.Command{}
|
|
err := samlAuthDataResetCmdF(s.client, cmd, nil)
|
|
s.Require().NotNil(err)
|
|
s.Require().EqualError(err, "could not proceed, either enable --confirm flag or use an interactive shell to complete operation: this is not an interactive shell")
|
|
})
|
|
|
|
s.Run("Reset auth data without errors", func() {
|
|
printer.Clean()
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("yes", true, "")
|
|
outputMessage := "1 user records were changed.\n"
|
|
|
|
s.client.
|
|
EXPECT().
|
|
ResetSamlAuthDataToEmail(context.TODO(), false, false, []string{}).
|
|
Return(int64(1), &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := samlAuthDataResetCmdF(s.client, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], outputMessage)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Reset auth data dry run", func() {
|
|
printer.Clean()
|
|
outputMessage := "1 user records would be affected.\n"
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("dry-run", true, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
ResetSamlAuthDataToEmail(context.TODO(), false, true, []string{}).
|
|
Return(int64(1), &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err := samlAuthDataResetCmdF(s.client, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], outputMessage)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("Reset auth data with specified users", func() {
|
|
printer.Clean()
|
|
users := []string{"user1"}
|
|
s.client.
|
|
EXPECT().
|
|
ResetSamlAuthDataToEmail(context.TODO(), false, false, users).
|
|
Return(int64(1), &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("yes", true, "")
|
|
cmd.Flags().StringSlice("users", users, "")
|
|
|
|
err := samlAuthDataResetCmdF(s.client, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|