Includes mmctl into the mono-repo (#23091)
* Includes mmctl into the mono-repo * Update to use the new public module paths * Adds docs check to the mmctl CI * Fix public utils import path * Tidy up modules * Fix linter * Update CI tasks to use the new file structure * Update CI references
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
412109b02e
Коммит
951456c780
114
server/cmd/mmctl/commands/integrity_test.go
Обычный файл
114
server/cmd/mmctl/commands/integrity_test.go
Обычный файл
@@ -0,0 +1,114 @@
|
||||
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
||||
// See LICENSE.txt for license information.
|
||||
|
||||
package commands
|
||||
|
||||
import (
|
||||
"errors"
|
||||
|
||||
"github.com/hashicorp/go-multierror"
|
||||
|
||||
"github.com/mattermost/mattermost-server/server/v8/cmd/mmctl/printer"
|
||||
|
||||
"github.com/mattermost/mattermost-server/server/public/model"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
func (s *MmctlUnitTestSuite) TestIntegrityCmd() {
|
||||
s.Run("Integrity check succeeds", func() {
|
||||
printer.Clean()
|
||||
cmd := &cobra.Command{}
|
||||
cmd.Flags().Bool("confirm", true, "")
|
||||
|
||||
mockData := model.RelationalIntegrityCheckData{
|
||||
ParentName: "parent",
|
||||
ChildName: "child",
|
||||
ParentIdAttr: "parentIdAttr",
|
||||
ChildIdAttr: "childIdAttr",
|
||||
Records: []model.OrphanedRecord{
|
||||
{
|
||||
ParentId: model.NewString("parentId"),
|
||||
ChildId: model.NewString("childId"),
|
||||
},
|
||||
},
|
||||
}
|
||||
mockResults := []model.IntegrityCheckResult{
|
||||
{
|
||||
Data: mockData,
|
||||
Err: nil,
|
||||
},
|
||||
}
|
||||
s.client.
|
||||
EXPECT().
|
||||
CheckIntegrity().
|
||||
Return(mockResults, &model.Response{}, nil).
|
||||
Times(1)
|
||||
|
||||
err := integrityCmdF(s.client, cmd, []string{})
|
||||
s.Require().Nil(err)
|
||||
s.Require().Len(printer.GetLines(), 1)
|
||||
s.Require().Len(printer.GetErrorLines(), 0)
|
||||
s.Require().Equal(mockData, printer.GetLines()[0])
|
||||
})
|
||||
|
||||
s.Run("Integrity check fails", func() {
|
||||
printer.Clean()
|
||||
cmd := &cobra.Command{}
|
||||
cmd.Flags().Bool("confirm", true, "")
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
CheckIntegrity().
|
||||
Return(nil, &model.Response{}, errors.New("mock error")).
|
||||
Times(1)
|
||||
|
||||
err := integrityCmdF(s.client, cmd, []string{})
|
||||
s.Require().NotNil(err)
|
||||
s.Require().Len(printer.GetLines(), 0)
|
||||
s.Require().Len(printer.GetErrorLines(), 0)
|
||||
s.Require().Equal("unable to perform integrity check. Error: mock error", err.Error())
|
||||
})
|
||||
|
||||
s.Run("Integrity check with errors", func() {
|
||||
printer.Clean()
|
||||
cmd := &cobra.Command{}
|
||||
cmd.Flags().Bool("confirm", true, "")
|
||||
|
||||
mockData := model.RelationalIntegrityCheckData{
|
||||
ParentName: "parent",
|
||||
ChildName: "child",
|
||||
ParentIdAttr: "parentIdAttr",
|
||||
ChildIdAttr: "childIdAttr",
|
||||
Records: []model.OrphanedRecord{
|
||||
{
|
||||
ParentId: model.NewString("parentId"),
|
||||
ChildId: model.NewString("childId"),
|
||||
},
|
||||
},
|
||||
}
|
||||
mockResults := []model.IntegrityCheckResult{
|
||||
{
|
||||
Data: nil,
|
||||
Err: errors.New("test error"),
|
||||
},
|
||||
{
|
||||
Data: mockData,
|
||||
Err: nil,
|
||||
},
|
||||
}
|
||||
s.client.
|
||||
EXPECT().
|
||||
CheckIntegrity().
|
||||
Return(mockResults, &model.Response{}, nil).
|
||||
Times(1)
|
||||
var expected error
|
||||
expected = multierror.Append(expected, errors.New("test error"))
|
||||
|
||||
err := integrityCmdF(s.client, cmd, []string{})
|
||||
s.Require().EqualError(err, expected.Error())
|
||||
s.Require().Len(printer.GetLines(), 1)
|
||||
s.Require().Len(printer.GetErrorLines(), 1)
|
||||
s.Require().Equal(mockData, printer.GetLines()[0])
|
||||
s.Require().Equal("test error", printer.GetErrorLines()[0])
|
||||
})
|
||||
}
|
||||
Ссылка в новой задаче
Block a user