7dadc5dfe9
- Abstracting from the exec form while generating. - Removal of single quotes (usage of only double-quotes). - Bash-nesting using escaping of special characters.
602 lines
17 KiB
Go
602 lines
17 KiB
Go
// Code generated by mockery v2.21.0. DO NOT EDIT.
|
|
|
|
package nomad
|
|
|
|
import (
|
|
context "context"
|
|
|
|
api "github.com/hashicorp/nomad/api"
|
|
config "github.com/openHPI/poseidon/internal/config"
|
|
|
|
io "io"
|
|
|
|
mock "github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
// apiQuerierMock is an autogenerated mock type for the apiQuerier type
|
|
type apiQuerierMock struct {
|
|
mock.Mock
|
|
}
|
|
|
|
type apiQuerierMock_Expecter struct {
|
|
mock *mock.Mock
|
|
}
|
|
|
|
func (_m *apiQuerierMock) EXPECT() *apiQuerierMock_Expecter {
|
|
return &apiQuerierMock_Expecter{mock: &_m.Mock}
|
|
}
|
|
|
|
// DeleteJob provides a mock function with given fields: jobID
|
|
func (_m *apiQuerierMock) DeleteJob(jobID string) error {
|
|
ret := _m.Called(jobID)
|
|
|
|
var r0 error
|
|
if rf, ok := ret.Get(0).(func(string) error); ok {
|
|
r0 = rf(jobID)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// apiQuerierMock_DeleteJob_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteJob'
|
|
type apiQuerierMock_DeleteJob_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// DeleteJob is a helper method to define mock.On call
|
|
// - jobID string
|
|
func (_e *apiQuerierMock_Expecter) DeleteJob(jobID interface{}) *apiQuerierMock_DeleteJob_Call {
|
|
return &apiQuerierMock_DeleteJob_Call{Call: _e.mock.On("DeleteJob", jobID)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_DeleteJob_Call) Run(run func(jobID string)) *apiQuerierMock_DeleteJob_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_DeleteJob_Call) Return(err error) *apiQuerierMock_DeleteJob_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_DeleteJob_Call) RunAndReturn(run func(string) error) *apiQuerierMock_DeleteJob_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// EventStream provides a mock function with given fields: ctx
|
|
func (_m *apiQuerierMock) EventStream(ctx context.Context) (<-chan *api.Events, error) {
|
|
ret := _m.Called(ctx)
|
|
|
|
var r0 <-chan *api.Events
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(context.Context) (<-chan *api.Events, error)); ok {
|
|
return rf(ctx)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(context.Context) <-chan *api.Events); ok {
|
|
r0 = rf(ctx)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(<-chan *api.Events)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(context.Context) error); ok {
|
|
r1 = rf(ctx)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_EventStream_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'EventStream'
|
|
type apiQuerierMock_EventStream_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// EventStream is a helper method to define mock.On call
|
|
// - ctx context.Context
|
|
func (_e *apiQuerierMock_Expecter) EventStream(ctx interface{}) *apiQuerierMock_EventStream_Call {
|
|
return &apiQuerierMock_EventStream_Call{Call: _e.mock.On("EventStream", ctx)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_EventStream_Call) Run(run func(ctx context.Context)) *apiQuerierMock_EventStream_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(context.Context))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_EventStream_Call) Return(_a0 <-chan *api.Events, _a1 error) *apiQuerierMock_EventStream_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_EventStream_Call) RunAndReturn(run func(context.Context) (<-chan *api.Events, error)) *apiQuerierMock_EventStream_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// Execute provides a mock function with given fields: jobID, ctx, command, tty, stdin, stdout, stderr
|
|
func (_m *apiQuerierMock) Execute(jobID string, ctx context.Context, command string, tty bool, stdin io.Reader, stdout io.Writer, stderr io.Writer) (int, error) {
|
|
ret := _m.Called(jobID, ctx, command, tty, stdin, stdout, stderr)
|
|
|
|
var r0 int
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string, context.Context, string, bool, io.Reader, io.Writer, io.Writer) (int, error)); ok {
|
|
return rf(jobID, ctx, command, tty, stdin, stdout, stderr)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string, context.Context, string, bool, io.Reader, io.Writer, io.Writer) int); ok {
|
|
r0 = rf(jobID, ctx, command, tty, stdin, stdout, stderr)
|
|
} else {
|
|
r0 = ret.Get(0).(int)
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string, context.Context, string, bool, io.Reader, io.Writer, io.Writer) error); ok {
|
|
r1 = rf(jobID, ctx, command, tty, stdin, stdout, stderr)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_Execute_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Execute'
|
|
type apiQuerierMock_Execute_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// Execute is a helper method to define mock.On call
|
|
// - jobID string
|
|
// - ctx context.Context
|
|
// - command string
|
|
// - tty bool
|
|
// - stdin io.Reader
|
|
// - stdout io.Writer
|
|
// - stderr io.Writer
|
|
func (_e *apiQuerierMock_Expecter) Execute(jobID interface{}, ctx interface{}, command interface{}, tty interface{}, stdin interface{}, stdout interface{}, stderr interface{}) *apiQuerierMock_Execute_Call {
|
|
return &apiQuerierMock_Execute_Call{Call: _e.mock.On("Execute", jobID, ctx, command, tty, stdin, stdout, stderr)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_Execute_Call) Run(run func(jobID string, ctx context.Context, command string, tty bool, stdin io.Reader, stdout io.Writer, stderr io.Writer)) *apiQuerierMock_Execute_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string), args[1].(context.Context), args[2].(string), args[3].(bool), args[4].(io.Reader), args[5].(io.Writer), args[6].(io.Writer))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_Execute_Call) Return(_a0 int, _a1 error) *apiQuerierMock_Execute_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_Execute_Call) RunAndReturn(run func(string, context.Context, string, bool, io.Reader, io.Writer, io.Writer) (int, error)) *apiQuerierMock_Execute_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// JobScale provides a mock function with given fields: jobID
|
|
func (_m *apiQuerierMock) JobScale(jobID string) (uint, error) {
|
|
ret := _m.Called(jobID)
|
|
|
|
var r0 uint
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string) (uint, error)); ok {
|
|
return rf(jobID)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string) uint); ok {
|
|
r0 = rf(jobID)
|
|
} else {
|
|
r0 = ret.Get(0).(uint)
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string) error); ok {
|
|
r1 = rf(jobID)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_JobScale_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'JobScale'
|
|
type apiQuerierMock_JobScale_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// JobScale is a helper method to define mock.On call
|
|
// - jobID string
|
|
func (_e *apiQuerierMock_Expecter) JobScale(jobID interface{}) *apiQuerierMock_JobScale_Call {
|
|
return &apiQuerierMock_JobScale_Call{Call: _e.mock.On("JobScale", jobID)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_JobScale_Call) Run(run func(jobID string)) *apiQuerierMock_JobScale_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_JobScale_Call) Return(jobScale uint, err error) *apiQuerierMock_JobScale_Call {
|
|
_c.Call.Return(jobScale, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_JobScale_Call) RunAndReturn(run func(string) (uint, error)) *apiQuerierMock_JobScale_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// LoadJobList provides a mock function with given fields:
|
|
func (_m *apiQuerierMock) LoadJobList() ([]*api.JobListStub, error) {
|
|
ret := _m.Called()
|
|
|
|
var r0 []*api.JobListStub
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func() ([]*api.JobListStub, error)); ok {
|
|
return rf()
|
|
}
|
|
if rf, ok := ret.Get(0).(func() []*api.JobListStub); ok {
|
|
r0 = rf()
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*api.JobListStub)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func() error); ok {
|
|
r1 = rf()
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_LoadJobList_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LoadJobList'
|
|
type apiQuerierMock_LoadJobList_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// LoadJobList is a helper method to define mock.On call
|
|
func (_e *apiQuerierMock_Expecter) LoadJobList() *apiQuerierMock_LoadJobList_Call {
|
|
return &apiQuerierMock_LoadJobList_Call{Call: _e.mock.On("LoadJobList")}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_LoadJobList_Call) Run(run func()) *apiQuerierMock_LoadJobList_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run()
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_LoadJobList_Call) Return(list []*api.JobListStub, err error) *apiQuerierMock_LoadJobList_Call {
|
|
_c.Call.Return(list, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_LoadJobList_Call) RunAndReturn(run func() ([]*api.JobListStub, error)) *apiQuerierMock_LoadJobList_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// RegisterNomadJob provides a mock function with given fields: job
|
|
func (_m *apiQuerierMock) RegisterNomadJob(job *api.Job) (string, error) {
|
|
ret := _m.Called(job)
|
|
|
|
var r0 string
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(*api.Job) (string, error)); ok {
|
|
return rf(job)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(*api.Job) string); ok {
|
|
r0 = rf(job)
|
|
} else {
|
|
r0 = ret.Get(0).(string)
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(*api.Job) error); ok {
|
|
r1 = rf(job)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_RegisterNomadJob_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RegisterNomadJob'
|
|
type apiQuerierMock_RegisterNomadJob_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// RegisterNomadJob is a helper method to define mock.On call
|
|
// - job *api.Job
|
|
func (_e *apiQuerierMock_Expecter) RegisterNomadJob(job interface{}) *apiQuerierMock_RegisterNomadJob_Call {
|
|
return &apiQuerierMock_RegisterNomadJob_Call{Call: _e.mock.On("RegisterNomadJob", job)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_RegisterNomadJob_Call) Run(run func(job *api.Job)) *apiQuerierMock_RegisterNomadJob_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(*api.Job))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_RegisterNomadJob_Call) Return(_a0 string, _a1 error) *apiQuerierMock_RegisterNomadJob_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_RegisterNomadJob_Call) RunAndReturn(run func(*api.Job) (string, error)) *apiQuerierMock_RegisterNomadJob_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// SetJobScale provides a mock function with given fields: jobID, count, reason
|
|
func (_m *apiQuerierMock) SetJobScale(jobID string, count uint, reason string) error {
|
|
ret := _m.Called(jobID, count, reason)
|
|
|
|
var r0 error
|
|
if rf, ok := ret.Get(0).(func(string, uint, string) error); ok {
|
|
r0 = rf(jobID, count, reason)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// apiQuerierMock_SetJobScale_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetJobScale'
|
|
type apiQuerierMock_SetJobScale_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// SetJobScale is a helper method to define mock.On call
|
|
// - jobID string
|
|
// - count uint
|
|
// - reason string
|
|
func (_e *apiQuerierMock_Expecter) SetJobScale(jobID interface{}, count interface{}, reason interface{}) *apiQuerierMock_SetJobScale_Call {
|
|
return &apiQuerierMock_SetJobScale_Call{Call: _e.mock.On("SetJobScale", jobID, count, reason)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_SetJobScale_Call) Run(run func(jobID string, count uint, reason string)) *apiQuerierMock_SetJobScale_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string), args[1].(uint), args[2].(string))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_SetJobScale_Call) Return(err error) *apiQuerierMock_SetJobScale_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_SetJobScale_Call) RunAndReturn(run func(string, uint, string) error) *apiQuerierMock_SetJobScale_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// allocation provides a mock function with given fields: jobID
|
|
func (_m *apiQuerierMock) allocation(jobID string) (*api.Allocation, error) {
|
|
ret := _m.Called(jobID)
|
|
|
|
var r0 *api.Allocation
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string) (*api.Allocation, error)); ok {
|
|
return rf(jobID)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string) *api.Allocation); ok {
|
|
r0 = rf(jobID)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*api.Allocation)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string) error); ok {
|
|
r1 = rf(jobID)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_allocation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'allocation'
|
|
type apiQuerierMock_allocation_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// allocation is a helper method to define mock.On call
|
|
// - jobID string
|
|
func (_e *apiQuerierMock_Expecter) allocation(jobID interface{}) *apiQuerierMock_allocation_Call {
|
|
return &apiQuerierMock_allocation_Call{Call: _e.mock.On("allocation", jobID)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_allocation_Call) Run(run func(jobID string)) *apiQuerierMock_allocation_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_allocation_Call) Return(_a0 *api.Allocation, _a1 error) *apiQuerierMock_allocation_Call {
|
|
_c.Call.Return(_a0, _a1)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_allocation_Call) RunAndReturn(run func(string) (*api.Allocation, error)) *apiQuerierMock_allocation_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// init provides a mock function with given fields: nomadConfig
|
|
func (_m *apiQuerierMock) init(nomadConfig *config.Nomad) error {
|
|
ret := _m.Called(nomadConfig)
|
|
|
|
var r0 error
|
|
if rf, ok := ret.Get(0).(func(*config.Nomad) error); ok {
|
|
r0 = rf(nomadConfig)
|
|
} else {
|
|
r0 = ret.Error(0)
|
|
}
|
|
|
|
return r0
|
|
}
|
|
|
|
// apiQuerierMock_init_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'init'
|
|
type apiQuerierMock_init_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// init is a helper method to define mock.On call
|
|
// - nomadConfig *config.Nomad
|
|
func (_e *apiQuerierMock_Expecter) init(nomadConfig interface{}) *apiQuerierMock_init_Call {
|
|
return &apiQuerierMock_init_Call{Call: _e.mock.On("init", nomadConfig)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_init_Call) Run(run func(nomadConfig *config.Nomad)) *apiQuerierMock_init_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(*config.Nomad))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_init_Call) Return(err error) *apiQuerierMock_init_Call {
|
|
_c.Call.Return(err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_init_Call) RunAndReturn(run func(*config.Nomad) error) *apiQuerierMock_init_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// job provides a mock function with given fields: jobID
|
|
func (_m *apiQuerierMock) job(jobID string) (*api.Job, error) {
|
|
ret := _m.Called(jobID)
|
|
|
|
var r0 *api.Job
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string) (*api.Job, error)); ok {
|
|
return rf(jobID)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string) *api.Job); ok {
|
|
r0 = rf(jobID)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).(*api.Job)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string) error); ok {
|
|
r1 = rf(jobID)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_job_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'job'
|
|
type apiQuerierMock_job_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// job is a helper method to define mock.On call
|
|
// - jobID string
|
|
func (_e *apiQuerierMock_Expecter) job(jobID interface{}) *apiQuerierMock_job_Call {
|
|
return &apiQuerierMock_job_Call{Call: _e.mock.On("job", jobID)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_job_Call) Run(run func(jobID string)) *apiQuerierMock_job_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_job_Call) Return(job *api.Job, err error) *apiQuerierMock_job_Call {
|
|
_c.Call.Return(job, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_job_Call) RunAndReturn(run func(string) (*api.Job, error)) *apiQuerierMock_job_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
// listJobs provides a mock function with given fields: prefix
|
|
func (_m *apiQuerierMock) listJobs(prefix string) ([]*api.JobListStub, error) {
|
|
ret := _m.Called(prefix)
|
|
|
|
var r0 []*api.JobListStub
|
|
var r1 error
|
|
if rf, ok := ret.Get(0).(func(string) ([]*api.JobListStub, error)); ok {
|
|
return rf(prefix)
|
|
}
|
|
if rf, ok := ret.Get(0).(func(string) []*api.JobListStub); ok {
|
|
r0 = rf(prefix)
|
|
} else {
|
|
if ret.Get(0) != nil {
|
|
r0 = ret.Get(0).([]*api.JobListStub)
|
|
}
|
|
}
|
|
|
|
if rf, ok := ret.Get(1).(func(string) error); ok {
|
|
r1 = rf(prefix)
|
|
} else {
|
|
r1 = ret.Error(1)
|
|
}
|
|
|
|
return r0, r1
|
|
}
|
|
|
|
// apiQuerierMock_listJobs_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'listJobs'
|
|
type apiQuerierMock_listJobs_Call struct {
|
|
*mock.Call
|
|
}
|
|
|
|
// listJobs is a helper method to define mock.On call
|
|
// - prefix string
|
|
func (_e *apiQuerierMock_Expecter) listJobs(prefix interface{}) *apiQuerierMock_listJobs_Call {
|
|
return &apiQuerierMock_listJobs_Call{Call: _e.mock.On("listJobs", prefix)}
|
|
}
|
|
|
|
func (_c *apiQuerierMock_listJobs_Call) Run(run func(prefix string)) *apiQuerierMock_listJobs_Call {
|
|
_c.Call.Run(func(args mock.Arguments) {
|
|
run(args[0].(string))
|
|
})
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_listJobs_Call) Return(allocationListStub []*api.JobListStub, err error) *apiQuerierMock_listJobs_Call {
|
|
_c.Call.Return(allocationListStub, err)
|
|
return _c
|
|
}
|
|
|
|
func (_c *apiQuerierMock_listJobs_Call) RunAndReturn(run func(string) ([]*api.JobListStub, error)) *apiQuerierMock_listJobs_Call {
|
|
_c.Call.Return(run)
|
|
return _c
|
|
}
|
|
|
|
type mockConstructorTestingTnewApiQuerierMock interface {
|
|
mock.TestingT
|
|
Cleanup(func())
|
|
}
|
|
|
|
// newApiQuerierMock creates a new instance of apiQuerierMock. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
|
|
func newApiQuerierMock(t mockConstructorTestingTnewApiQuerierMock) *apiQuerierMock {
|
|
mock := &apiQuerierMock{}
|
|
mock.Mock.Test(t)
|
|
|
|
t.Cleanup(func() { mock.AssertExpectations(t) })
|
|
|
|
return mock
|
|
}
|