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>
380 lines
11 KiB
Go
380 lines
11 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/mattermost/mattermost/server/v8/cmd/mmctl/printer"
|
|
)
|
|
|
|
func (s *MmctlUnitTestSuite) TestPostCreateCmdF() {
|
|
s.Run("create a post with empty text", func() {
|
|
cmd := &cobra.Command{}
|
|
|
|
err := postCreateCmdF(s.client, cmd, []string{"some-channel", ""})
|
|
s.Require().EqualError(err, "message cannot be empty")
|
|
})
|
|
|
|
s.Run("no channel specified", func() {
|
|
msgArg := "some text"
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("message", msgArg, "")
|
|
|
|
err := postCreateCmdF(s.client, cmd, []string{"", msgArg})
|
|
s.Require().EqualError(err, "Unable to find channel ''")
|
|
})
|
|
|
|
s.Run("wrong reply msg", func() {
|
|
msgArg := "some text"
|
|
replyToArg := "a-non-existing-post"
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("message", msgArg, "")
|
|
cmd.Flags().String("reply-to", replyToArg, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPost(context.TODO(), replyToArg, "").
|
|
Return(nil, &model.Response{}, errors.New("some-error")).
|
|
Times(1)
|
|
|
|
err := postCreateCmdF(s.client, cmd, []string{msgArg})
|
|
s.Require().Contains(err.Error(), "some-error")
|
|
})
|
|
|
|
s.Run("error when creating a post", func() {
|
|
msgArg := "some text"
|
|
channelArg := "example-channel"
|
|
mockChannel := model.Channel{Name: channelArg}
|
|
mockPost := &model.Post{Message: msgArg}
|
|
data, err := mockPost.ToJSON()
|
|
s.Require().NoError(err)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("message", msgArg, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
DoAPIPost(context.TODO(), "/posts?set_online=false", data).
|
|
Return(nil, errors.New("some-error")).
|
|
Times(1)
|
|
|
|
err = postCreateCmdF(s.client, cmd, []string{channelArg, msgArg})
|
|
s.Require().Contains(err.Error(), "could not create post")
|
|
})
|
|
|
|
s.Run("create a post", func() {
|
|
msgArg := "some text"
|
|
channelArg := "example-channel"
|
|
mockChannel := model.Channel{Name: channelArg}
|
|
mockPost := model.Post{Message: msgArg}
|
|
data, err := mockPost.ToJSON()
|
|
s.Require().NoError(err)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("message", msgArg, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
DoAPIPost(context.TODO(), "/posts?set_online=false", data).
|
|
Return(nil, nil).
|
|
Times(1)
|
|
|
|
err = postCreateCmdF(s.client, cmd, []string{channelArg, msgArg})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("reply to an existing post", func() {
|
|
msgArg := "some text"
|
|
replyToArg := "an-existing-post"
|
|
rootID := "some-root-id"
|
|
channelArg := "example-channel"
|
|
mockChannel := model.Channel{Name: channelArg}
|
|
mockReplyTo := model.Post{RootId: rootID}
|
|
mockPost := model.Post{Message: msgArg, RootId: rootID}
|
|
data, err := mockPost.ToJSON()
|
|
s.Require().NoError(err)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("reply-to", replyToArg, "")
|
|
cmd.Flags().String("message", msgArg, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelArg, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPost(context.TODO(), replyToArg, "").
|
|
Return(&mockReplyTo, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
DoAPIPost(context.TODO(), "/posts?set_online=false", data).
|
|
Return(nil, nil).
|
|
Times(1)
|
|
|
|
err = postCreateCmdF(s.client, cmd, []string{channelArg, msgArg})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestPostListCmdF() {
|
|
s.Run("no channel specified", func() {
|
|
sinceArg := "invalid-date"
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("since", sinceArg, "")
|
|
|
|
err := postListCmdF(s.client, cmd, []string{"", sinceArg})
|
|
s.Require().EqualError(err, "Unable to find channel ''")
|
|
})
|
|
|
|
s.Run("invalid time for since flag", func() {
|
|
sinceArg := "invalid-date"
|
|
mockChannel := model.Channel{Name: channelName}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("since", sinceArg, "")
|
|
|
|
err := postListCmdF(s.client, cmd, []string{channelName, sinceArg})
|
|
s.Require().Contains(err.Error(), "invalid since time 'invalid-date'")
|
|
})
|
|
|
|
s.Run("list posts for a channel", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockPost := &model.Post{Message: "some text", Id: "some-id", UserId: userID, CreateAt: model.GetMillisForTime(time.Now())}
|
|
mockPostList := model.NewPostList()
|
|
mockPostList.AddPost(mockPost)
|
|
mockPostList.AddOrder(mockPost.Id)
|
|
mockUser := model.User{Id: userID, Username: "some-user"}
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Int("number", 1, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsForChannel(context.TODO(), channelID, 0, 1, "", false, false).
|
|
Return(mockPostList, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetUser(context.TODO(), userID, "").
|
|
Return(&mockUser, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
printer.Clean()
|
|
err := postListCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetLines(), 1)
|
|
s.Require().Equal(printer.GetLines()[0], mockPost)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("list posts for a channel from a certain time (valid date)", func() {
|
|
printer.Clean()
|
|
|
|
ISO8601ValidString := "2006-01-02T15:04:05-07:00"
|
|
|
|
sinceArg := "2006-01-02T15:04:05-07:00"
|
|
sinceTime, err := time.Parse(ISO8601ValidString, sinceArg)
|
|
s.Require().Nil(err)
|
|
|
|
sinceTimeMillis := model.GetMillisForTime(sinceTime)
|
|
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockPost := &model.Post{Message: "some text", Id: "some-id", UserId: userID}
|
|
mockPostList := model.NewPostList()
|
|
mockPostList.AddPost(mockPost)
|
|
mockPostList.AddOrder(mockPost.Id)
|
|
mockUser := model.User{Id: userID, Username: "some-user"}
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Int("number", 1, "")
|
|
cmd.Flags().String("since", sinceArg, "")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName, "").
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsSince(context.TODO(), channelID, sinceTimeMillis, false).
|
|
Return(mockPostList, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetUser(context.TODO(), userID, "").
|
|
Return(&mockUser, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
err = postListCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Nil(err)
|
|
s.Require().Equal(printer.GetLines()[0], mockPost)
|
|
s.Len(printer.GetLines(), 1)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|
|
|
|
func (s *MmctlUnitTestSuite) TestDeletePostsCmdF() {
|
|
postID1 := "ux9bxc1b8bf1zdoj1tfu14836e"
|
|
postID2 := "ux9bxc1b8bf1zdoj1tfu14836f"
|
|
|
|
s.Run("invalid post id", func() {
|
|
id := "invalid-id"
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("confirm", true, "")
|
|
cmd.Flags().Bool("permanent", false, "")
|
|
|
|
err := deletePostsCmdF(s.client, cmd, []string{id})
|
|
s.Require().Nil(err)
|
|
s.Require().Equal("Invalid postID: invalid-id", printer.GetErrorLines()[0])
|
|
})
|
|
|
|
s.Run("successfully permanently delete one post", func() {
|
|
printer.Clean()
|
|
s.client.
|
|
EXPECT().
|
|
PermanentDeletePost(context.TODO(), postID1).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("confirm", true, "")
|
|
cmd.Flags().Bool("permanent", true, "")
|
|
|
|
err := deletePostsCmdF(s.client, cmd, []string{postID1})
|
|
s.Require().Nil(err)
|
|
s.Require().Equal(postID1+" successfully deleted", printer.GetLines()[0])
|
|
})
|
|
|
|
s.Run("successfully soft delete one post", func() {
|
|
printer.Clean()
|
|
s.client.
|
|
EXPECT().
|
|
DeletePost(context.TODO(), postID1).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("confirm", true, "")
|
|
cmd.Flags().Bool("permanent", false, "")
|
|
|
|
err := deletePostsCmdF(s.client, cmd, []string{postID1})
|
|
s.Require().Nil(err)
|
|
s.Require().Equal(postID1+" successfully deleted", printer.GetLines()[0])
|
|
})
|
|
|
|
s.Run("successfully delete multiple posts", func() {
|
|
printer.Clean()
|
|
s.client.
|
|
EXPECT().
|
|
PermanentDeletePost(context.TODO(), postID1).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
PermanentDeletePost(context.TODO(), postID2).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("confirm", true, "")
|
|
cmd.Flags().Bool("permanent", true, "")
|
|
|
|
err := deletePostsCmdF(s.client, cmd, []string{postID1, postID2})
|
|
s.Require().Nil(err)
|
|
s.Require().Equal(postID1+" successfully deleted", printer.GetLines()[0])
|
|
s.Require().Equal(postID2+" successfully deleted", printer.GetLines()[1])
|
|
})
|
|
|
|
s.Run("PermanentDeletePost api request returns an error", func() {
|
|
printer.Clean()
|
|
|
|
mockError := errors.New("an error occurred on deleting a post")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
PermanentDeletePost(context.TODO(), postID1).
|
|
Return(&model.Response{StatusCode: http.StatusBadRequest}, mockError).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("confirm", true, "")
|
|
cmd.Flags().Bool("permanent", true, "")
|
|
|
|
err := deletePostsCmdF(s.client, cmd, []string{postID1})
|
|
s.Require().ErrorContains(err, "an error occurred on deleting a post")
|
|
s.Require().Len(printer.GetErrorLines(), 1)
|
|
s.Require().Equal("Error deleting post: "+postID1+". Error: an error occurred on deleting a post",
|
|
printer.GetErrorLines()[0])
|
|
})
|
|
|
|
s.Run("Delete multiple posts but one fails with an error", func() {
|
|
printer.Clean()
|
|
mockError := errors.New("an error occurred on deleting a post")
|
|
s.client.
|
|
EXPECT().
|
|
PermanentDeletePost(context.TODO(), postID1).
|
|
Return(&model.Response{StatusCode: http.StatusOK}, nil).
|
|
Times(1)
|
|
s.client.
|
|
EXPECT().
|
|
PermanentDeletePost(context.TODO(), postID2).
|
|
Return(&model.Response{StatusCode: http.StatusBadRequest}, mockError).
|
|
Times(1)
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().Bool("confirm", true, "")
|
|
cmd.Flags().Bool("permanent", true, "")
|
|
|
|
err := deletePostsCmdF(s.client, cmd, []string{postID1, postID2})
|
|
s.Require().ErrorContains(err, "an error occurred on deleting a post")
|
|
s.Require().Len(printer.GetLines(), 1)
|
|
s.Require().Len(printer.GetErrorLines(), 1)
|
|
s.Require().Equal(postID1+" successfully deleted", printer.GetLines()[0])
|
|
s.Require().Equal("Error deleting post: "+postID2+". Error: an error occurred on deleting a post",
|
|
printer.GetErrorLines()[0])
|
|
})
|
|
}
|