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.
592 lines
15 KiB
Go
592 lines
15 KiB
Go
// Code generated by mockery v2.53.4. DO NOT EDIT.
|
|
|
|
// Regenerate this file using `make store-mocks`.
|
|
|
|
package mocks
|
|
|
|
import (
|
|
model "github.com/hanzoteam/server/server/public/model"
|
|
request "github.com/hanzoteam/server/server/public/shared/request"
|
|
mock "github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
// JobStore is an autogenerated mock type for the JobStore type
|
|
type JobStore struct {
|
|
mock.Mock
|
|
}
|
|
|
|
// Cleanup provides a mock function with given fields: expiryTime, batchSize
|
|
func (_m *JobStore) Cleanup(expiryTime int64, batchSize int) error {
|
|
ret := _m.Called(expiryTime, batchSize)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Cleanup")
|
|
}
|
|
|
|
var r0 error
|
|
if rf, ok := ret.Get(0).(func(int64, int) error); ok {
|
|
r0 = rf(expiryTime, batchSize)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// Delete provides a mock function with given fields: id
|
|
func (_m *JobStore) Delete(id string) (string, error) {
|
|
ret := _m.Called(id)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Delete")
|
|
}
|
|
|
|
var r0 string
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string) (string, error)); ok {
|
|
return rf(id)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string) string); ok {
|
|
r0 = rf(id)
|
|
} else {
|
|
r0 = ret.Get(0).(string)
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string) error); ok {
|
|
r1 = rf(id)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// Get provides a mock function with given fields: rctx, id
|
|
func (_m *JobStore) Get(rctx request.CTX, id string) (*model.Job, error) {
|
|
ret := _m.Called(rctx, id)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Get")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string) (*model.Job, error)); ok {
|
|
return rf(rctx, id)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string) *model.Job); ok {
|
|
r0 = rf(rctx, id)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, string) error); ok {
|
|
r1 = rf(rctx, id)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetAllByStatus provides a mock function with given fields: rctx, status
|
|
func (_m *JobStore) GetAllByStatus(rctx request.CTX, status string) ([]*model.Job, error) {
|
|
ret := _m.Called(rctx, status)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetAllByStatus")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string) ([]*model.Job, error)); ok {
|
|
return rf(rctx, status)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string) []*model.Job); ok {
|
|
r0 = rf(rctx, status)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, string) error); ok {
|
|
r1 = rf(rctx, status)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetAllByType provides a mock function with given fields: rctx, jobType
|
|
func (_m *JobStore) GetAllByType(rctx request.CTX, jobType string) ([]*model.Job, error) {
|
|
ret := _m.Called(rctx, jobType)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetAllByType")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string) ([]*model.Job, error)); ok {
|
|
return rf(rctx, jobType)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string) []*model.Job); ok {
|
|
r0 = rf(rctx, jobType)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, string) error); ok {
|
|
r1 = rf(rctx, jobType)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetAllByTypeAndStatus provides a mock function with given fields: rctx, jobType, status
|
|
func (_m *JobStore) GetAllByTypeAndStatus(rctx request.CTX, jobType string, status string) ([]*model.Job, error) {
|
|
ret := _m.Called(rctx, jobType, status)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetAllByTypeAndStatus")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string, string) ([]*model.Job, error)); ok {
|
|
return rf(rctx, jobType, status)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string, string) []*model.Job); ok {
|
|
r0 = rf(rctx, jobType, status)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, string, string) error); ok {
|
|
r1 = rf(rctx, jobType, status)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetAllByTypePage provides a mock function with given fields: rctx, jobType, offset, limit
|
|
func (_m *JobStore) GetAllByTypePage(rctx request.CTX, jobType string, offset int, limit int) ([]*model.Job, error) {
|
|
ret := _m.Called(rctx, jobType, offset, limit)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetAllByTypePage")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string, int, int) ([]*model.Job, error)); ok {
|
|
return rf(rctx, jobType, offset, limit)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string, int, int) []*model.Job); ok {
|
|
r0 = rf(rctx, jobType, offset, limit)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, string, int, int) error); ok {
|
|
r1 = rf(rctx, jobType, offset, limit)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetAllByTypesAndStatusesPage provides a mock function with given fields: rctx, jobType, status, offset, limit
|
|
func (_m *JobStore) GetAllByTypesAndStatusesPage(rctx request.CTX, jobType []string, status []string, offset int, limit int) ([]*model.Job, error) {
|
|
ret := _m.Called(rctx, jobType, status, offset, limit)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetAllByTypesAndStatusesPage")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, []string, []string, int, int) ([]*model.Job, error)); ok {
|
|
return rf(rctx, jobType, status, offset, limit)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, []string, []string, int, int) []*model.Job); ok {
|
|
r0 = rf(rctx, jobType, status, offset, limit)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, []string, []string, int, int) error); ok {
|
|
r1 = rf(rctx, jobType, status, offset, limit)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetAllByTypesPage provides a mock function with given fields: rctx, jobTypes, offset, limit
|
|
func (_m *JobStore) GetAllByTypesPage(rctx request.CTX, jobTypes []string, offset int, limit int) ([]*model.Job, error) {
|
|
ret := _m.Called(rctx, jobTypes, offset, limit)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetAllByTypesPage")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, []string, int, int) ([]*model.Job, error)); ok {
|
|
return rf(rctx, jobTypes, offset, limit)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, []string, int, int) []*model.Job); ok {
|
|
r0 = rf(rctx, jobTypes, offset, limit)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, []string, int, int) error); ok {
|
|
r1 = rf(rctx, jobTypes, offset, limit)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetByTypeAndData provides a mock function with given fields: rctx, jobType, data, useMaster, statuses
|
|
func (_m *JobStore) GetByTypeAndData(rctx request.CTX, jobType string, data map[string]string, useMaster bool, statuses ...string) ([]*model.Job, error) {
|
|
_va := make([]interface{}, len(statuses))
|
|
for _i := range statuses {
|
|
_va[_i] = statuses[_i]
|
|
}
|
|
var _ca []interface{}
|
|
_ca = append(_ca, rctx, jobType, data, useMaster)
|
|
_ca = append(_ca, _va...)
|
|
ret := _m.Called(_ca...)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetByTypeAndData")
|
|
}
|
|
|
|
var r0 []*model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string, map[string]string, bool, ...string) ([]*model.Job, error)); ok {
|
|
return rf(rctx, jobType, data, useMaster, statuses...)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(request.CTX, string, map[string]string, bool, ...string) []*model.Job); ok {
|
|
r0 = rf(rctx, jobType, data, useMaster, statuses...)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(request.CTX, string, map[string]string, bool, ...string) error); ok {
|
|
r1 = rf(rctx, jobType, data, useMaster, statuses...)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetCountByStatusAndType provides a mock function with given fields: status, jobType
|
|
func (_m *JobStore) GetCountByStatusAndType(status string, jobType string) (int64, error) {
|
|
ret := _m.Called(status, jobType)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetCountByStatusAndType")
|
|
}
|
|
|
|
var r0 int64
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string, string) (int64, error)); ok {
|
|
return rf(status, jobType)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string, string) int64); ok {
|
|
r0 = rf(status, jobType)
|
|
} else {
|
|
r0 = ret.Get(0).(int64)
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string, string) error); ok {
|
|
r1 = rf(status, jobType)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetNewestJobByStatusAndType provides a mock function with given fields: status, jobType
|
|
func (_m *JobStore) GetNewestJobByStatusAndType(status string, jobType string) (*model.Job, error) {
|
|
ret := _m.Called(status, jobType)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetNewestJobByStatusAndType")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string, string) (*model.Job, error)); ok {
|
|
return rf(status, jobType)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string, string) *model.Job); ok {
|
|
r0 = rf(status, jobType)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string, string) error); ok {
|
|
r1 = rf(status, jobType)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// GetNewestJobByStatusesAndType provides a mock function with given fields: statuses, jobType
|
|
func (_m *JobStore) GetNewestJobByStatusesAndType(statuses []string, jobType string) (*model.Job, error) {
|
|
ret := _m.Called(statuses, jobType)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for GetNewestJobByStatusesAndType")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func([]string, string) (*model.Job, error)); ok {
|
|
return rf(statuses, jobType)
|
|
}
|
|
if rf, ok := ret.Get(0).(func([]string, string) *model.Job); ok {
|
|
r0 = rf(statuses, jobType)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func([]string, string) error); ok {
|
|
r1 = rf(statuses, jobType)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// Save provides a mock function with given fields: job
|
|
func (_m *JobStore) Save(job *model.Job) (*model.Job, error) {
|
|
ret := _m.Called(job)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for Save")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(*model.Job) (*model.Job, error)); ok {
|
|
return rf(job)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(*model.Job) *model.Job); ok {
|
|
r0 = rf(job)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(*model.Job) error); ok {
|
|
r1 = rf(job)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// SaveOnce provides a mock function with given fields: job
|
|
func (_m *JobStore) SaveOnce(job *model.Job) (*model.Job, error) {
|
|
ret := _m.Called(job)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for SaveOnce")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(*model.Job) (*model.Job, error)); ok {
|
|
return rf(job)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(*model.Job) *model.Job); ok {
|
|
r0 = rf(job)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(*model.Job) error); ok {
|
|
r1 = rf(job)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// SaveOnceByTypeAndData provides a mock function with given fields: job, data
|
|
func (_m *JobStore) SaveOnceByTypeAndData(job *model.Job, data map[string]string) (*model.Job, error) {
|
|
ret := _m.Called(job, data)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for SaveOnceByTypeAndData")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(*model.Job, map[string]string) (*model.Job, error)); ok {
|
|
return rf(job, data)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(*model.Job, map[string]string) *model.Job); ok {
|
|
r0 = rf(job, data)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(*model.Job, map[string]string) error); ok {
|
|
r1 = rf(job, data)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// UpdateOptimistically provides a mock function with given fields: job, currentStatus
|
|
func (_m *JobStore) UpdateOptimistically(job *model.Job, currentStatus string) (*model.Job, error) {
|
|
ret := _m.Called(job, currentStatus)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for UpdateOptimistically")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(*model.Job, string) (*model.Job, error)); ok {
|
|
return rf(job, currentStatus)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(*model.Job, string) *model.Job); ok {
|
|
r0 = rf(job, currentStatus)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(*model.Job, string) error); ok {
|
|
r1 = rf(job, currentStatus)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// UpdateStatus provides a mock function with given fields: id, status
|
|
func (_m *JobStore) UpdateStatus(id string, status string) (*model.Job, error) {
|
|
ret := _m.Called(id, status)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for UpdateStatus")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string, string) (*model.Job, error)); ok {
|
|
return rf(id, status)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string, string) *model.Job); ok {
|
|
r0 = rf(id, status)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string, string) error); ok {
|
|
r1 = rf(id, status)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// UpdateStatusOptimistically provides a mock function with given fields: id, currentStatus, newStatus
|
|
func (_m *JobStore) UpdateStatusOptimistically(id string, currentStatus string, newStatus string) (*model.Job, error) {
|
|
ret := _m.Called(id, currentStatus, newStatus)
|
|
|
|
if len(ret) == 0 {
|
|
panic("no return value specified for UpdateStatusOptimistically")
|
|
}
|
|
|
|
var r0 *model.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string, string, string) (*model.Job, error)); ok {
|
|
return rf(id, currentStatus, newStatus)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string, string, string) *model.Job); ok {
|
|
r0 = rf(id, currentStatus, newStatus)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*model.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string, string, string) error); ok {
|
|
r1 = rf(id, currentStatus, newStatus)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// NewJobStore creates a new instance of JobStore. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
|
// The first argument is typically a *testing.T value.
|
|
func NewJobStore(t interface {
|
|
mock.TestingT
|
|
Cleanup(func())
|
|
}) *JobStore {
|
|
mock := &JobStore{}
|
|
mock.Mock.Test(t)
|
|
|
|
t.Cleanup(func() { mock.AssertExpectations(t) })
|
|
|
|
return mock
|
|
}
|