[MM-56000] Add LDAP job command to mmctl (#25633)

Этот коммит содержится в:
Ben Schumacher
2024-04-22 12:19:53 +02:00
коммит произвёл GitHub
родитель 8348acac49
Коммит 3dae305dc7
8 изменённых файлов: 476 добавлений и 7 удалений

Просмотреть файл

@@ -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)
})
})
}