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.
239 lines
6.9 KiB
Go
239 lines
6.9 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package app
|
|
|
|
import (
|
|
"fmt"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hanzoteam/server/server/public/model"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
type MockReportable struct {
|
|
TestField1 string
|
|
TestField2 int
|
|
TestField3 time.Time
|
|
}
|
|
|
|
func (mr *MockReportable) ToReport() []string {
|
|
return []string{
|
|
mr.TestField1,
|
|
strconv.Itoa(mr.TestField2),
|
|
mr.TestField3.Format("2006-01-02"),
|
|
}
|
|
}
|
|
|
|
var testData []model.ReportableObject = []model.ReportableObject{
|
|
&MockReportable{
|
|
TestField1: "some-name",
|
|
TestField2: 400,
|
|
TestField3: time.Date(2024, 1, 1, 0, 0, 0, 0, time.Local),
|
|
},
|
|
&MockReportable{
|
|
TestField1: "some-other-name",
|
|
TestField2: 500,
|
|
TestField3: time.Date(2023, 1, 1, 0, 0, 0, 0, time.Local),
|
|
},
|
|
&MockReportable{
|
|
TestField1: "some-other-other-name",
|
|
TestField2: 600,
|
|
TestField3: time.Date(2022, 1, 1, 0, 0, 0, 0, time.Local),
|
|
},
|
|
}
|
|
|
|
func TestSaveReportChunk(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic(t)
|
|
|
|
t.Run("should write CSV chunk to file", func(t *testing.T) {
|
|
prefix := model.NewId()
|
|
err := th.App.SaveReportChunk("csv", prefix, 999, []model.ReportableObject{testData[0]})
|
|
require.Nil(t, err)
|
|
|
|
filePath := fmt.Sprintf("admin_reports/batch_report_%s__999.csv", prefix)
|
|
bytes, err := th.App.ReadFile(filePath)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, bytes)
|
|
require.Equal(t, "some-name,400,2024-01-01\n", string(bytes))
|
|
})
|
|
|
|
t.Run("should fail if the report format is not supported", func(t *testing.T) {
|
|
err := th.App.SaveReportChunk("zzz", model.NewId(), 999, []model.ReportableObject{testData[0]})
|
|
require.NotNil(t, err)
|
|
})
|
|
}
|
|
|
|
func TestCompileReportChunks(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic(t)
|
|
|
|
prefix := model.NewId()
|
|
err := th.App.SaveReportChunk("csv", prefix, 0, []model.ReportableObject{testData[0]})
|
|
require.Nil(t, err)
|
|
err = th.App.SaveReportChunk("csv", prefix, 1, []model.ReportableObject{testData[1]})
|
|
require.Nil(t, err)
|
|
err = th.App.SaveReportChunk("csv", prefix, 2, []model.ReportableObject{testData[2]})
|
|
require.Nil(t, err)
|
|
|
|
t.Run("should compile a bunch of report chunks", func(t *testing.T) {
|
|
compileErr := th.App.CompileReportChunks("csv", prefix, 3, []string{"Name", "NumPosts", "StartDate"})
|
|
require.Nil(t, compileErr)
|
|
|
|
filePath := fmt.Sprintf("admin_reports/batch_report_%s.csv", prefix)
|
|
bytes, readErr := th.App.ReadFile(filePath)
|
|
require.Nil(t, readErr)
|
|
require.NotNil(t, bytes)
|
|
|
|
expected := `Name,NumPosts,StartDate
|
|
some-name,400,2024-01-01
|
|
some-other-name,500,2023-01-01
|
|
some-other-other-name,600,2022-01-01
|
|
`
|
|
require.Equal(t, expected, string(bytes))
|
|
})
|
|
|
|
t.Run("should fail if the report format is not supported", func(t *testing.T) {
|
|
err = th.App.CompileReportChunks("zzz", prefix, 3, []string{"Name", "NumPosts", "StartDate"})
|
|
require.NotNil(t, err)
|
|
})
|
|
|
|
t.Run("should fail if a chunk is missing", func(t *testing.T) {
|
|
err = th.App.CompileReportChunks("csv", prefix, 4, []string{"Name", "NumPosts", "StartDate"})
|
|
require.NotNil(t, err)
|
|
})
|
|
}
|
|
|
|
func TestCheckForExistingJobs(t *testing.T) {
|
|
mainHelper.Parallel(t)
|
|
th := Setup(t).InitBasic(t)
|
|
|
|
t.Run("should return error if job with same options exists in pending jobs", func(t *testing.T) {
|
|
app := th.App
|
|
options := map[string]string{
|
|
"date_range": "last_30_days",
|
|
"requesting_user_id": th.BasicUser.Id,
|
|
"role": "user",
|
|
"team": "",
|
|
"hide_active": "false",
|
|
"hide_inactive": "false",
|
|
"guest_filter": "",
|
|
}
|
|
|
|
jobType := model.JobTypeExportUsersToCSV
|
|
|
|
// Create a pending job with same options
|
|
job, err := app.Srv().Jobs.CreateJob(th.Context, jobType, options)
|
|
defer func() {
|
|
_ = app.Srv().Jobs.RequestCancellation(th.Context, job.Id)
|
|
}()
|
|
require.Nil(t, err)
|
|
require.NotNil(t, job)
|
|
|
|
// checkForExistingJobs
|
|
appErr := app.checkForExistingJobs(th.Context, options, jobType)
|
|
require.NotNil(t, appErr)
|
|
require.Equal(t, "app.report.start_users_batch_export.job_exists", appErr.Id)
|
|
})
|
|
|
|
t.Run("should return error if job with same options exists in in-progress jobs", func(t *testing.T) {
|
|
app := th.App
|
|
options := map[string]string{
|
|
"date_range": "last_30_days",
|
|
"requesting_user_id": th.BasicUser.Id,
|
|
"role": "user",
|
|
"team": "",
|
|
"hide_active": "false",
|
|
"hide_inactive": "false",
|
|
"guest_filter": "",
|
|
}
|
|
|
|
jobType := model.JobTypeExportUsersToCSV
|
|
|
|
// Create an in-progress job with same options
|
|
job, err := app.Srv().Jobs.CreateJob(th.Context, jobType, options)
|
|
defer func() {
|
|
_ = app.Srv().Jobs.RequestCancellation(th.Context, job.Id)
|
|
}()
|
|
require.Nil(t, err)
|
|
require.NotNil(t, job)
|
|
|
|
// Manually set job status to in-progress
|
|
err = app.Srv().Jobs.SetJobProgress(job, 60)
|
|
require.Nil(t, err)
|
|
|
|
// Call checkForExistingJobs
|
|
appErr := app.checkForExistingJobs(th.Context, options, jobType)
|
|
require.NotNil(t, appErr)
|
|
require.Equal(t, "app.report.start_users_batch_export.job_exists", appErr.Id)
|
|
})
|
|
|
|
t.Run("should not return error if existing jobs have different options", func(t *testing.T) {
|
|
app := th.App
|
|
options := map[string]string{
|
|
"date_range": "last_30_days",
|
|
"requesting_user_id": th.BasicUser.Id,
|
|
"role": "user",
|
|
"team": "",
|
|
"hide_active": "false",
|
|
"hide_inactive": "false",
|
|
"guest_filter": "",
|
|
}
|
|
|
|
jobType := model.JobTypeExportUsersToCSV
|
|
|
|
differentOptions := map[string]string{
|
|
"date_range": "all_time",
|
|
"requesting_user_id": th.BasicUser2.Id,
|
|
"role": "admin",
|
|
"team": "",
|
|
"hide_active": "false",
|
|
"hide_inactive": "false",
|
|
"guest_filter": "",
|
|
}
|
|
|
|
job, err := app.Srv().Jobs.CreateJob(th.Context, jobType, differentOptions)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, job)
|
|
|
|
// Call checkForExistingJobs
|
|
appErr := app.checkForExistingJobs(th.Context, options, jobType)
|
|
require.Nil(t, appErr)
|
|
})
|
|
|
|
t.Run("should not return error if existing job has different guest_filter", func(t *testing.T) {
|
|
app := th.App
|
|
options := map[string]string{
|
|
"date_range": "last_30_days",
|
|
"requesting_user_id": th.BasicUser.Id,
|
|
"role": "",
|
|
"team": "",
|
|
"hide_active": "false",
|
|
"hide_inactive": "false",
|
|
"guest_filter": "single_channel",
|
|
}
|
|
|
|
jobType := model.JobTypeExportUsersToCSV
|
|
|
|
existingJobOptions := map[string]string{
|
|
"date_range": "last_30_days",
|
|
"requesting_user_id": th.BasicUser.Id,
|
|
"role": "",
|
|
"team": "",
|
|
"hide_active": "false",
|
|
"hide_inactive": "false",
|
|
"guest_filter": "multi_channel",
|
|
}
|
|
|
|
job, err := app.Srv().Jobs.CreateJob(th.Context, jobType, existingJobOptions)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, job)
|
|
|
|
appErr := app.checkForExistingJobs(th.Context, options, jobType)
|
|
require.Nil(t, appErr)
|
|
})
|
|
}
|