[MM-56000] Add LDAP job command to mmctl (#25633)
Этот коммит содержится в:
коммит произвёл
GitHub
родитель
8348acac49
Коммит
3dae305dc7
@@ -5,14 +5,14 @@ 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"
|
||||
|
||||
"github.com/mattermost/mattermost/server/public/model"
|
||||
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"
|
||||
)
|
||||
|
||||
func (s *MmctlUnitTestSuite) TestLdapSyncCmd() {
|
||||
@@ -114,3 +114,157 @@ func (s *MmctlUnitTestSuite) TestLdapMigrateID() {
|
||||
s.Require().Len(printer.GetLines(), 0)
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MmctlUnitTestSuite) TestLdapJobListCmdF() {
|
||||
s.Run("no LDAP jobs", func() {
|
||||
printer.Clean()
|
||||
var mockJobs []*model.Job
|
||||
|
||||
cmd := &cobra.Command{}
|
||||
perPage := 10
|
||||
cmd.Flags().Int("page", 0, "")
|
||||
cmd.Flags().Int("per-page", perPage, "")
|
||||
cmd.Flags().Bool("all", false, "")
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
GetJobsByType(context.TODO(), model.JobTypeLdapSync, 0, perPage).
|
||||
Return(mockJobs, &model.Response{}, nil).
|
||||
Times(1)
|
||||
|
||||
err := ldapJobListCmdF(s.client, cmd, nil)
|
||||
s.Require().Nil(err)
|
||||
s.Len(printer.GetLines(), 1)
|
||||
s.Empty(printer.GetErrorLines())
|
||||
s.Equal("No jobs found", printer.GetLines()[0])
|
||||
})
|
||||
|
||||
s.Run("some LDAP jobs", func() {
|
||||
printer.Clean()
|
||||
mockJobs := []*model.Job{
|
||||
{
|
||||
Id: model.NewId(),
|
||||
},
|
||||
{
|
||||
Id: model.NewId(),
|
||||
},
|
||||
{
|
||||
Id: model.NewId(),
|
||||
},
|
||||
}
|
||||
|
||||
cmd := &cobra.Command{}
|
||||
perPage := 3
|
||||
cmd.Flags().Int("page", 0, "")
|
||||
cmd.Flags().Int("per-page", perPage, "")
|
||||
cmd.Flags().Bool("all", false, "")
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
GetJobsByType(context.TODO(), model.JobTypeLdapSync, 0, perPage).
|
||||
Return(mockJobs, &model.Response{}, nil).
|
||||
Times(1)
|
||||
|
||||
err := ldapJobListCmdF(s.client, cmd, nil)
|
||||
s.Require().Nil(err)
|
||||
s.Len(printer.GetLines(), len(mockJobs))
|
||||
s.Empty(printer.GetErrorLines())
|
||||
for i, line := range printer.GetLines() {
|
||||
s.Equal(mockJobs[i], line.(*model.Job))
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
func (s *MmctlUnitTestSuite) TestLdapJobShowCmdF() {
|
||||
s.Run("not found", func() {
|
||||
printer.Clean()
|
||||
|
||||
jobID := model.NewId()
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
GetJob(context.TODO(), jobID).
|
||||
Return(nil, &model.Response{StatusCode: http.StatusNotFound}, errors.New("not found")).
|
||||
Times(1)
|
||||
|
||||
err := ldapJobShowCmdF(s.client, &cobra.Command{}, []string{jobID})
|
||||
s.Require().NotNil(err)
|
||||
s.Empty(printer.GetLines())
|
||||
s.Empty(printer.GetErrorLines())
|
||||
})
|
||||
|
||||
s.Run("found", func() {
|
||||
printer.Clean()
|
||||
mockJob := &model.Job{
|
||||
Id: model.NewId(),
|
||||
}
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
GetJob(context.TODO(), mockJob.Id).
|
||||
Return(mockJob, &model.Response{}, nil).
|
||||
Times(1)
|
||||
|
||||
err := ldapJobShowCmdF(s.client, &cobra.Command{}, []string{mockJob.Id})
|
||||
s.Require().Nil(err)
|
||||
s.Len(printer.GetLines(), 1)
|
||||
s.Empty(printer.GetErrorLines())
|
||||
s.Equal(mockJob, printer.GetLines()[0].(*model.Job))
|
||||
})
|
||||
|
||||
s.Run("shell completion", func() {
|
||||
s.Run("no match for empty argument", func() {
|
||||
r, dir := ldapJobShowCompletionF(context.Background(), s.client, nil, nil, "")
|
||||
s.Equal(cobra.ShellCompDirectiveNoFileComp, dir)
|
||||
s.Equal([]string{}, r)
|
||||
})
|
||||
|
||||
s.Run("one element matches", func() {
|
||||
mockJobs := []*model.Job{
|
||||
{
|
||||
Id: "0_id",
|
||||
},
|
||||
{
|
||||
Id: "1_id",
|
||||
},
|
||||
{
|
||||
Id: "2_id",
|
||||
},
|
||||
}
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
GetJobsByType(context.Background(), model.JobTypeLdapSync, 0, DefaultPageSize).
|
||||
Return(mockJobs, &model.Response{}, nil).
|
||||
Times(1)
|
||||
|
||||
r, dir := ldapJobShowCompletionF(context.Background(), s.client, nil, nil, "1")
|
||||
s.Equal(cobra.ShellCompDirectiveNoFileComp, dir)
|
||||
s.Equal([]string{"1_id"}, r)
|
||||
})
|
||||
|
||||
s.Run("more elements then the limit match", func() {
|
||||
var mockJobs []*model.Job
|
||||
for i := 0; i < 100; i++ {
|
||||
mockJobs = append(mockJobs, &model.Job{
|
||||
Id: fmt.Sprintf("id_%d", i),
|
||||
})
|
||||
}
|
||||
|
||||
var expected []string
|
||||
for i := 0; i < shellCompletionMaxItems; i++ {
|
||||
expected = append(expected, fmt.Sprintf("id_%d", i))
|
||||
}
|
||||
|
||||
s.client.
|
||||
EXPECT().
|
||||
GetJobsByType(context.Background(), model.JobTypeLdapSync, 0, DefaultPageSize).
|
||||
Return(mockJobs, &model.Response{}, nil).
|
||||
Times(1)
|
||||
|
||||
r, dir := ldapJobShowCompletionF(context.Background(), s.client, nil, nil, "id_")
|
||||
s.Equal(cobra.ShellCompDirectiveNoFileComp, dir)
|
||||
s.Equal(expected, r)
|
||||
})
|
||||
})
|
||||
}
|
||||
|
||||
Ссылка в новой задаче
Block a user