mirror of
https://github.com/hanzoai/team.git
synced 2026-09-30 01:57:35 +00:00
zeekay
8feb24ea1f
The server module named github.com/hanzoai/team/server/v8 — a path that resolves to the retired workspace's repository, not this one — while server/public still published github.com/mattermost/mattermost/server/public. Both are github.com/hanzoteam/server/... now, which is where this code actually lives. That second half is what made the binary unbootable. Upstream's public path stayed claimed by upstream, mattermost-plugin-ai imported it, and the build linked public/model twice; public/plugin's gob.Register then ran twice under one name and panicked at init, before main. Reverting the rename traded the brand for a working binary. Rebranding the plugin alongside it needs no such trade: the agents fork is github.com/hanzoteam/agents/v2 against our public module, so no module named mattermost/mattermost is in the graph at all. Measured: `go list -deps ./cmd/mattermost` names exactly one public/model, and it is ours.
365 lines
11 KiB
Go
365 lines
11 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
//go:build unit
|
|
|
|
package commands
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/hanzoteam/server/server/v8/cmd/mmctl/printer"
|
|
)
|
|
|
|
func (s *MmctlUnitTestSuite) TestReportPostsCmdF() {
|
|
s.Run("no channel specified", func() {
|
|
printer.Clean()
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{""})
|
|
s.Require().EqualError(err, "Unable to find channel ''")
|
|
})
|
|
|
|
s.Run("invalid time-field", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "invalid_field", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().EqualError(err, "time-field must be either 'create_at' or 'update_at'")
|
|
})
|
|
|
|
s.Run("invalid sort-direction", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "invalid", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().EqualError(err, "sort-direction must be either 'asc' or 'desc'")
|
|
})
|
|
|
|
s.Run("invalid per-page (too large)", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 2000, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Contains(err.Error(), "per-page must be between 1 and")
|
|
})
|
|
|
|
s.Run("invalid per-page (zero)", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 0, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Contains(err.Error(), "per-page must be between 1 and")
|
|
})
|
|
|
|
s.Run("API error when getting posts", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockError := errors.New("api error occurred")
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsForReporting(context.TODO(), model.ReportPostOptions{
|
|
ChannelId: channelID,
|
|
TimeField: "create_at",
|
|
SortDirection: "asc",
|
|
PerPage: 100,
|
|
IncludeDeleted: false,
|
|
ExcludeSystemPosts: false,
|
|
IncludeMetadata: false,
|
|
}, model.ReportPostOptionsCursor{
|
|
Cursor: "",
|
|
}).
|
|
Return(nil, &model.Response{}, mockError).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Contains(err.Error(), "failed to get posts for reporting")
|
|
})
|
|
|
|
s.Run("successfully get posts with no next cursor", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockPost1 := &model.Post{Id: "post1", Message: "message1", UserId: userID, ChannelId: channelID}
|
|
mockPost2 := &model.Post{Id: "post2", Message: "message2", UserId: userID, ChannelId: channelID}
|
|
mockResponse := &model.ReportPostListResponse{
|
|
Posts: []*model.Post{
|
|
mockPost1,
|
|
mockPost2,
|
|
},
|
|
NextCursor: nil,
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsForReporting(context.TODO(), model.ReportPostOptions{
|
|
ChannelId: channelID,
|
|
TimeField: "create_at",
|
|
SortDirection: "asc",
|
|
PerPage: 100,
|
|
IncludeDeleted: false,
|
|
ExcludeSystemPosts: false,
|
|
IncludeMetadata: false,
|
|
}, model.ReportPostOptionsCursor{
|
|
Cursor: "",
|
|
}).
|
|
Return(mockResponse, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("successfully get posts with next cursor", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockPost1 := &model.Post{Id: "post1", Message: "message1", UserId: userID, ChannelId: channelID}
|
|
nextCursor := &model.ReportPostOptionsCursor{
|
|
Cursor: "MTphYmMxMjM6Y3JlYXRlX2F0OmZhbHNlOmZhbHNlOmFzYzoxNzM1NDg4MTIzNDU2OnBvc3Qx",
|
|
}
|
|
mockResponse := &model.ReportPostListResponse{
|
|
Posts: []*model.Post{
|
|
mockPost1,
|
|
},
|
|
NextCursor: nextCursor,
|
|
}
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsForReporting(context.TODO(), model.ReportPostOptions{
|
|
ChannelId: channelID,
|
|
TimeField: "create_at",
|
|
SortDirection: "asc",
|
|
PerPage: 100,
|
|
IncludeDeleted: false,
|
|
ExcludeSystemPosts: false,
|
|
IncludeMetadata: false,
|
|
}, model.ReportPostOptionsCursor{
|
|
Cursor: "",
|
|
}).
|
|
Return(mockResponse, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().String("cursor", "", "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("successfully get posts with custom cursor", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockPost1 := &model.Post{Id: "post1", Message: "message1", UserId: userID, ChannelId: channelID}
|
|
mockResponse := &model.ReportPostListResponse{
|
|
Posts: []*model.Post{
|
|
mockPost1,
|
|
},
|
|
NextCursor: nil,
|
|
}
|
|
|
|
customCursor := "MTphYmMxMjM6Y3JlYXRlX2F0OmZhbHNlOmZhbHNlOmFzYzoxNzM1NDg4MDAwMDAwOmN1c3RvbXBvc3QxMjM"
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsForReporting(context.TODO(), model.ReportPostOptions{
|
|
ChannelId: channelID,
|
|
TimeField: "create_at",
|
|
SortDirection: "asc",
|
|
PerPage: 100,
|
|
IncludeDeleted: false,
|
|
ExcludeSystemPosts: false,
|
|
IncludeMetadata: false,
|
|
}, model.ReportPostOptionsCursor{
|
|
Cursor: customCursor,
|
|
}).
|
|
Return(mockResponse, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "create_at", "")
|
|
cmd.Flags().String("sort-direction", "asc", "")
|
|
cmd.Flags().Int64("end-time", int64(0), "")
|
|
cmd.Flags().String("cursor", customCursor, "")
|
|
cmd.Flags().Int("per-page", 100, "")
|
|
cmd.Flags().Bool("include-deleted", false, "")
|
|
cmd.Flags().Bool("exclude-system-posts", false, "")
|
|
cmd.Flags().Bool("include-metadata", false, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
|
|
s.Run("successfully get posts with all options enabled", func() {
|
|
printer.Clean()
|
|
mockChannel := model.Channel{Name: channelName, Id: channelID}
|
|
mockPost1 := &model.Post{Id: "post1", Message: "message1", UserId: userID, ChannelId: channelID}
|
|
mockResponse := &model.ReportPostListResponse{
|
|
Posts: []*model.Post{
|
|
mockPost1,
|
|
},
|
|
NextCursor: nil,
|
|
}
|
|
|
|
customCursor := "MTphYmMxMjM6dXBkYXRlX2F0OnRydWU6dHJ1ZTpkZXNjOjE3MzU0MDAwMDAwMDA6"
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetChannel(context.TODO(), channelName).
|
|
Return(&mockChannel, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
s.client.
|
|
EXPECT().
|
|
GetPostsForReporting(context.TODO(), model.ReportPostOptions{
|
|
ChannelId: channelID,
|
|
TimeField: "update_at",
|
|
SortDirection: "desc",
|
|
PerPage: 500,
|
|
IncludeDeleted: true,
|
|
ExcludeSystemPosts: true,
|
|
IncludeMetadata: true,
|
|
}, model.ReportPostOptionsCursor{
|
|
Cursor: customCursor,
|
|
}).
|
|
Return(mockResponse, &model.Response{}, nil).
|
|
Times(1)
|
|
|
|
cmd := &cobra.Command{}
|
|
cmd.Flags().String("time-field", "update_at", "")
|
|
cmd.Flags().String("sort-direction", "desc", "")
|
|
cmd.Flags().String("cursor", customCursor, "")
|
|
cmd.Flags().Int("per-page", 500, "")
|
|
cmd.Flags().Bool("include-deleted", true, "")
|
|
cmd.Flags().Bool("exclude-system-posts", true, "")
|
|
cmd.Flags().Bool("include-metadata", true, "")
|
|
|
|
err := reportPostsCmdF(s.client, cmd, []string{channelName})
|
|
s.Require().Nil(err)
|
|
s.Len(printer.GetErrorLines(), 0)
|
|
})
|
|
}
|