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>
485 lines
15 KiB
Go
485 lines
15 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost/server/v8"
|
|
"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/mattermost/mattermost/server/public/utils"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportUploadCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
importName := "import_test.zip"
|
|
importFilePath := filepath.Join(server.GetPackagePath(), "tests", importName)
|
|
info, err := os.Stat(importFilePath)
|
|
s.Require().NoError(err)
|
|
|
|
s.Run("no permissions", func() {
|
|
printer.Clean()
|
|
|
|
err := importUploadCmdF(s.th.Client, &cobra.Command{}, []string{importFilePath})
|
|
s.Require().NotNil(err)
|
|
s.Require().Equal("failed to create upload session: You do not have the appropriate permissions.", err.Error())
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("invalid file", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := importUploadCmdF(s.th.Client, &cobra.Command{}, []string{"invalid_file"})
|
|
s.Require().NotNil(err)
|
|
s.Require().Equal("failed to open import file: open invalid_file: no such file or directory", err.Error())
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("full upload", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
if c == s.th.LocalClient {
|
|
cmd.Flags().Bool("local", true, "")
|
|
}
|
|
|
|
err := importUploadCmdF(c, cmd, []string{importFilePath})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 2)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Equal(importName, printer.GetLines()[0].(*model.UploadSession).Filename)
|
|
s.Require().Equal(importName, printer.GetLines()[1].(*model.FileInfo).Name)
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("resume upload", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
userID := "me"
|
|
cmd := &cobra.Command{}
|
|
if c == s.th.LocalClient {
|
|
cmd.Flags().Bool("local", true, "")
|
|
userID = "nouser"
|
|
}
|
|
|
|
us, _, err := c.CreateUpload(context.TODO(), &model.UploadSession{
|
|
Filename: importName,
|
|
FileSize: info.Size(),
|
|
Type: model.UploadTypeImport,
|
|
UserId: userID,
|
|
})
|
|
s.Require().NoError(err)
|
|
|
|
cmd.Flags().Bool("resume", true, "")
|
|
cmd.Flags().String("upload", us.Id, "")
|
|
|
|
err = importUploadCmdF(c, cmd, []string{importFilePath})
|
|
s.Require().NoError(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Equal(importName, printer.GetLines()[0].(*model.FileInfo).Name)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportProcessCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
importName := "import_test.zip"
|
|
importFilePath := filepath.Join(server.GetPackagePath(), "tests", "import_test.zip")
|
|
|
|
s.Run("no permissions", func() {
|
|
printer.Clean()
|
|
|
|
err := importProcessCmdF(s.th.Client, &cobra.Command{}, []string{"importName"})
|
|
s.Require().NotNil(err)
|
|
s.Require().Equal("failed to create import process job: You do not have the appropriate permissions.", err.Error())
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("process file", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
if c == s.th.LocalClient {
|
|
cmd.Flags().Bool("local", true, "")
|
|
}
|
|
|
|
err := importUploadCmdF(c, cmd, []string{importFilePath})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 2)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
|
|
us := printer.GetLines()[0].(*model.UploadSession)
|
|
printer.Clean()
|
|
|
|
err = importProcessCmdF(c, cmd, []string{us.Id + "_" + importName})
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Equal(us.Id+"_"+importName, printer.GetLines()[0].(*model.Job).Data["import_file"])
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportListAvailableCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
importName := "import_test.zip"
|
|
importFilePath := filepath.Join(server.GetPackagePath(), "tests", importName)
|
|
|
|
s.Run("no permissions", func() {
|
|
printer.Clean()
|
|
|
|
err := importListAvailableCmdF(s.th.Client, &cobra.Command{}, nil)
|
|
s.Require().NotNil(err)
|
|
s.Require().ErrorContains(err, "failed to list imports: You do not have the appropriate permissions.")
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("no imports", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := importListAvailableCmdF(c, &cobra.Command{}, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Equal("No import files found", printer.GetLines()[0])
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("some imports", func(c client.Client) {
|
|
cmd := &cobra.Command{}
|
|
if c == s.th.LocalClient {
|
|
cmd.Flags().Bool("local", true, "")
|
|
}
|
|
|
|
numImports := 3
|
|
for range numImports {
|
|
err := importUploadCmdF(c, cmd, []string{importFilePath})
|
|
s.Require().Nil(err)
|
|
}
|
|
printer.Clean()
|
|
|
|
imports, appErr := s.th.App.ListImports()
|
|
s.Require().Nil(appErr)
|
|
|
|
err := importListAvailableCmdF(c, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), len(imports))
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
for i, name := range printer.GetLines() {
|
|
s.Require().Equal(imports[i], name.(string))
|
|
}
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportListIncompleteCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
|
|
s.RunForAllClients("no incomplete import uploads", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := importListIncompleteCmdF(c, &cobra.Command{}, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Equal("No incomplete import uploads found", printer.GetLines()[0])
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("some incomplete import uploads", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
userID := "nouser"
|
|
if c == s.th.SystemAdminClient {
|
|
user, _, err := s.th.SystemAdminClient.GetMe(context.Background(), "")
|
|
s.Require().NoError(err)
|
|
userID = user.Id
|
|
} else {
|
|
cmd.Flags().Bool("local", true, "")
|
|
}
|
|
|
|
us1, appErr := s.th.App.CreateUploadSession(s.th.Context, &model.UploadSession{
|
|
Id: model.NewId(),
|
|
UserId: userID,
|
|
Type: model.UploadTypeImport,
|
|
Filename: "import1.zip",
|
|
FileSize: 1024 * 1024,
|
|
})
|
|
s.Require().Nil(appErr)
|
|
us1.Path = ""
|
|
|
|
time.Sleep(time.Millisecond)
|
|
|
|
_, appErr = s.th.App.CreateUploadSession(s.th.Context, &model.UploadSession{
|
|
Id: model.NewId(),
|
|
UserId: userID,
|
|
ChannelId: s.th.BasicChannel.Id,
|
|
Type: model.UploadTypeAttachment,
|
|
Filename: "somefile",
|
|
FileSize: 1024 * 1024,
|
|
})
|
|
s.Require().Nil(appErr)
|
|
|
|
time.Sleep(time.Millisecond)
|
|
|
|
us3, appErr := s.th.App.CreateUploadSession(s.th.Context, &model.UploadSession{
|
|
Id: model.NewId(),
|
|
UserId: userID,
|
|
Type: model.UploadTypeImport,
|
|
Filename: "import2.zip",
|
|
FileSize: 1024 * 1024,
|
|
})
|
|
s.Require().Nil(appErr)
|
|
us3.Path = ""
|
|
|
|
err := importListIncompleteCmdF(c, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 2)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Equal(us1, printer.GetLines()[0].(*model.UploadSession))
|
|
s.Require().Equal(us3, printer.GetLines()[1].(*model.UploadSession))
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportJobShowCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
|
|
job, appErr := s.th.App.CreateJob(s.th.Context, &model.Job{
|
|
Type: model.JobTypeImportProcess,
|
|
Data: map[string]string{"import_file": "import1.zip"},
|
|
})
|
|
s.Require().Nil(appErr)
|
|
|
|
s.Run("no permissions", func() {
|
|
printer.Clean()
|
|
|
|
job1, appErr := s.th.App.CreateJob(s.th.Context, &model.Job{
|
|
Type: model.JobTypeImportProcess,
|
|
Data: map[string]string{"import_file": "import1.zip"},
|
|
})
|
|
s.Require().Nil(appErr)
|
|
|
|
err := importJobShowCmdF(s.th.Client, &cobra.Command{}, []string{job1.Id})
|
|
s.Require().NotNil(err)
|
|
s.Require().ErrorContains(err, "failed to get import job: You do not have the appropriate permissions.")
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("not found", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := importJobShowCmdF(c, &cobra.Command{}, []string{model.NewId()})
|
|
s.Require().NotNil(err)
|
|
s.Require().ErrorContains(err, "failed to get import job: Unable to get the job.")
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("found", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
err := importJobShowCmdF(c, &cobra.Command{}, []string{job.Id})
|
|
s.Require().Nil(err)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Equal(job, printer.GetLines()[0].(*model.Job))
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportJobListCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
|
|
s.Run("no permissions", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Int("page", 0, "")
|
|
cmd.Flags().Int("per-page", 200, "")
|
|
cmd.Flags().Bool("all", false, "")
|
|
|
|
err := importJobListCmdF(s.th.Client, cmd, nil)
|
|
s.Require().NotNil(err)
|
|
s.Require().ErrorContains(err, "failed to get jobs: You do not have the appropriate permissions.")
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("no import jobs", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Int("page", 0, "")
|
|
cmd.Flags().Int("per-page", 200, "")
|
|
cmd.Flags().Bool("all", false, "")
|
|
|
|
err := importJobListCmdF(c, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Equal("No jobs found", printer.GetLines()[0])
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("some import jobs", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
perPage := 2
|
|
cmd.Flags().Int("page", 0, "")
|
|
cmd.Flags().Int("per-page", perPage, "")
|
|
cmd.Flags().Bool("all", false, "")
|
|
|
|
_, appErr := s.th.App.CreateJob(s.th.Context, &model.Job{
|
|
Type: model.JobTypeImportProcess,
|
|
Data: map[string]string{"import_file": "import1.zip"},
|
|
})
|
|
s.Require().Nil(appErr)
|
|
|
|
time.Sleep(time.Millisecond)
|
|
|
|
job2, appErr := s.th.App.CreateJob(s.th.Context, &model.Job{
|
|
Type: model.JobTypeImportProcess,
|
|
Data: map[string]string{"import_file": "import2.zip"},
|
|
})
|
|
s.Require().Nil(appErr)
|
|
|
|
time.Sleep(time.Millisecond)
|
|
|
|
job3, appErr := s.th.App.CreateJob(s.th.Context, &model.Job{
|
|
Type: model.JobTypeImportProcess,
|
|
Data: map[string]string{"import_file": "import3.zip"},
|
|
})
|
|
s.Require().Nil(appErr)
|
|
|
|
err := importJobListCmdF(c, cmd, nil)
|
|
s.Require().Nil(err)
|
|
s.Require().Len(printer.GetLines(), perPage)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Equal(job3, printer.GetLines()[0].(*model.Job))
|
|
s.Require().Equal(job2, printer.GetLines()[1].(*model.Job))
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportValidateCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
|
|
importName := "import_test.zip"
|
|
importFilePath := filepath.Join(server.GetPackagePath(), "tests", importName)
|
|
|
|
s.RunForSystemAdminAndLocal("defaults", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().StringArray("team", nil, "")
|
|
cmd.Flags().Bool("check-missing-teams", false, "")
|
|
cmd.Flags().Bool("ignore-attachments", false, "")
|
|
cmd.Flags().Bool("check-server-duplicates", true, "")
|
|
|
|
err := importValidateCmdF(c, cmd, []string{importFilePath})
|
|
s.Require().Nil(err)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
|
|
s.Require().Equal(Statistics{
|
|
Teams: 2,
|
|
Channels: 24,
|
|
Users: 16,
|
|
Posts: 2001,
|
|
DirectChannels: 79,
|
|
DirectPosts: 901,
|
|
Attachments: 2,
|
|
}, printer.GetLines()[0].(Statistics))
|
|
s.Require().Equal(struct {
|
|
UnusedAttachments []string `json:"unused_attachments"`
|
|
}{
|
|
UnusedAttachments: []string{"data/test2.png", "data/test_img_diff_A.png", "data/test_img_diff_B.png"},
|
|
}, printer.GetLines()[1].(struct {
|
|
UnusedAttachments []string `json:"unused_attachments"`
|
|
}))
|
|
s.Require().Equal("Validation complete\n", printer.GetLines()[3])
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("ignore attachments", func(c client.Client) {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().StringArray("team", nil, "")
|
|
cmd.Flags().Bool("check-missing-teams", false, "")
|
|
cmd.Flags().Bool("ignore-attachments", true, "")
|
|
cmd.Flags().Bool("check-server-duplicates", true, "")
|
|
|
|
err := importValidateCmdF(c, cmd, []string{importFilePath})
|
|
s.Require().Nil(err)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
|
|
s.Require().Equal(Statistics{
|
|
Teams: 2,
|
|
Channels: 24,
|
|
Users: 16,
|
|
Posts: 2001,
|
|
DirectChannels: 79,
|
|
DirectPosts: 901,
|
|
Attachments: 0,
|
|
}, printer.GetLines()[0].(Statistics))
|
|
s.Require().Equal("Validation complete\n", printer.GetLines()[2])
|
|
})
|
|
}
|
|
|
|
func (s *MmctlE2ETestSuite) TestImportDeleteCmdF() {
|
|
s.SetupTestHelper().InitBasic()
|
|
s.Run("no permissions", func() {
|
|
printer.Clean()
|
|
|
|
err := importDeleteCmdF(s.th.Client, &cobra.Command{}, []string{"import1.zip"})
|
|
s.Require().EqualError(err, "failed to delete import: You do not have the appropriate permissions.")
|
|
s.Require().Empty(printer.GetLines())
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
})
|
|
|
|
s.RunForSystemAdminAndLocal("delete import", func(c client.Client) {
|
|
importName := "import_test.zip"
|
|
importFilePath := filepath.Join(server.GetPackagePath(), "tests", importName)
|
|
importPath, err := filepath.Abs(filepath.Join(*s.th.App.Config().FileSettings.Directory,
|
|
*s.th.App.Config().ImportSettings.Directory))
|
|
s.Require().Nil(err)
|
|
|
|
cmd := &cobra.Command{}
|
|
|
|
newImportName := "new_import_test.zip"
|
|
err = utils.CopyFile(importFilePath, filepath.Join(importPath, newImportName))
|
|
s.Require().Nil(err)
|
|
|
|
printer.Clean()
|
|
imports, appErr := s.th.App.ListImports()
|
|
s.Require().Nil(appErr)
|
|
s.Require().NotEmpty(imports)
|
|
s.Require().Equal(newImportName, imports[0])
|
|
|
|
err = importDeleteCmdF(c, cmd, []string{newImportName})
|
|
s.Require().Nil(err)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Equal(fmt.Sprintf(`Import file "%s" has been deleted`, newImportName), printer.GetLines()[0])
|
|
|
|
imports, appErr = s.th.App.ListImports()
|
|
s.Require().Nil(appErr)
|
|
s.Require().Empty(imports)
|
|
|
|
//idempotency check
|
|
|
|
err = importDeleteCmdF(c, cmd, []string{newImportName})
|
|
s.Require().Nil(err)
|
|
s.Require().Empty(printer.GetErrorLines())
|
|
s.Require().Len(printer.GetLines(), 2)
|
|
s.Equal(fmt.Sprintf(`Import file "%s" has been deleted`, newImportName), printer.GetLines()[0])
|
|
})
|
|
}
|