mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-12 13:15:09 +08:00
185 lines
5.3 KiB
Go
185 lines
5.3 KiB
Go
|
// Code generated by mockery v2.32.4. DO NOT EDIT.
|
||
|
|
||
|
package function
|
||
|
|
||
|
import (
|
||
|
schemapb "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
|
||
|
mock "github.com/stretchr/testify/mock"
|
||
|
)
|
||
|
|
||
|
// MockFunctionRunner is an autogenerated mock type for the FunctionRunner type
|
||
|
type MockFunctionRunner struct {
|
||
|
mock.Mock
|
||
|
}
|
||
|
|
||
|
type MockFunctionRunner_Expecter struct {
|
||
|
mock *mock.Mock
|
||
|
}
|
||
|
|
||
|
func (_m *MockFunctionRunner) EXPECT() *MockFunctionRunner_Expecter {
|
||
|
return &MockFunctionRunner_Expecter{mock: &_m.Mock}
|
||
|
}
|
||
|
|
||
|
// BatchRun provides a mock function with given fields: inputs
|
||
|
func (_m *MockFunctionRunner) BatchRun(inputs ...interface{}) ([]interface{}, error) {
|
||
|
var _ca []interface{}
|
||
|
_ca = append(_ca, inputs...)
|
||
|
ret := _m.Called(_ca...)
|
||
|
|
||
|
var r0 []interface{}
|
||
|
var r1 error
|
||
|
if rf, ok := ret.Get(0).(func(...interface{}) ([]interface{}, error)); ok {
|
||
|
return rf(inputs...)
|
||
|
}
|
||
|
if rf, ok := ret.Get(0).(func(...interface{}) []interface{}); ok {
|
||
|
r0 = rf(inputs...)
|
||
|
} else {
|
||
|
if ret.Get(0) != nil {
|
||
|
r0 = ret.Get(0).([]interface{})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
if rf, ok := ret.Get(1).(func(...interface{}) error); ok {
|
||
|
r1 = rf(inputs...)
|
||
|
} else {
|
||
|
r1 = ret.Error(1)
|
||
|
}
|
||
|
|
||
|
return r0, r1
|
||
|
}
|
||
|
|
||
|
// MockFunctionRunner_BatchRun_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'BatchRun'
|
||
|
type MockFunctionRunner_BatchRun_Call struct {
|
||
|
*mock.Call
|
||
|
}
|
||
|
|
||
|
// BatchRun is a helper method to define mock.On call
|
||
|
// - inputs ...interface{}
|
||
|
func (_e *MockFunctionRunner_Expecter) BatchRun(inputs ...interface{}) *MockFunctionRunner_BatchRun_Call {
|
||
|
return &MockFunctionRunner_BatchRun_Call{Call: _e.mock.On("BatchRun",
|
||
|
append([]interface{}{}, inputs...)...)}
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_BatchRun_Call) Run(run func(inputs ...interface{})) *MockFunctionRunner_BatchRun_Call {
|
||
|
_c.Call.Run(func(args mock.Arguments) {
|
||
|
variadicArgs := make([]interface{}, len(args)-0)
|
||
|
for i, a := range args[0:] {
|
||
|
if a != nil {
|
||
|
variadicArgs[i] = a.(interface{})
|
||
|
}
|
||
|
}
|
||
|
run(variadicArgs...)
|
||
|
})
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_BatchRun_Call) Return(_a0 []interface{}, _a1 error) *MockFunctionRunner_BatchRun_Call {
|
||
|
_c.Call.Return(_a0, _a1)
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_BatchRun_Call) RunAndReturn(run func(...interface{}) ([]interface{}, error)) *MockFunctionRunner_BatchRun_Call {
|
||
|
_c.Call.Return(run)
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
// GetOutputFields provides a mock function with given fields:
|
||
|
func (_m *MockFunctionRunner) GetOutputFields() []*schemapb.FieldSchema {
|
||
|
ret := _m.Called()
|
||
|
|
||
|
var r0 []*schemapb.FieldSchema
|
||
|
if rf, ok := ret.Get(0).(func() []*schemapb.FieldSchema); ok {
|
||
|
r0 = rf()
|
||
|
} else {
|
||
|
if ret.Get(0) != nil {
|
||
|
r0 = ret.Get(0).([]*schemapb.FieldSchema)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return r0
|
||
|
}
|
||
|
|
||
|
// MockFunctionRunner_GetOutputFields_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetOutputFields'
|
||
|
type MockFunctionRunner_GetOutputFields_Call struct {
|
||
|
*mock.Call
|
||
|
}
|
||
|
|
||
|
// GetOutputFields is a helper method to define mock.On call
|
||
|
func (_e *MockFunctionRunner_Expecter) GetOutputFields() *MockFunctionRunner_GetOutputFields_Call {
|
||
|
return &MockFunctionRunner_GetOutputFields_Call{Call: _e.mock.On("GetOutputFields")}
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_GetOutputFields_Call) Run(run func()) *MockFunctionRunner_GetOutputFields_Call {
|
||
|
_c.Call.Run(func(args mock.Arguments) {
|
||
|
run()
|
||
|
})
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_GetOutputFields_Call) Return(_a0 []*schemapb.FieldSchema) *MockFunctionRunner_GetOutputFields_Call {
|
||
|
_c.Call.Return(_a0)
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_GetOutputFields_Call) RunAndReturn(run func() []*schemapb.FieldSchema) *MockFunctionRunner_GetOutputFields_Call {
|
||
|
_c.Call.Return(run)
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
// GetSchema provides a mock function with given fields:
|
||
|
func (_m *MockFunctionRunner) GetSchema() *schemapb.FunctionSchema {
|
||
|
ret := _m.Called()
|
||
|
|
||
|
var r0 *schemapb.FunctionSchema
|
||
|
if rf, ok := ret.Get(0).(func() *schemapb.FunctionSchema); ok {
|
||
|
r0 = rf()
|
||
|
} else {
|
||
|
if ret.Get(0) != nil {
|
||
|
r0 = ret.Get(0).(*schemapb.FunctionSchema)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return r0
|
||
|
}
|
||
|
|
||
|
// MockFunctionRunner_GetSchema_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetSchema'
|
||
|
type MockFunctionRunner_GetSchema_Call struct {
|
||
|
*mock.Call
|
||
|
}
|
||
|
|
||
|
// GetSchema is a helper method to define mock.On call
|
||
|
func (_e *MockFunctionRunner_Expecter) GetSchema() *MockFunctionRunner_GetSchema_Call {
|
||
|
return &MockFunctionRunner_GetSchema_Call{Call: _e.mock.On("GetSchema")}
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_GetSchema_Call) Run(run func()) *MockFunctionRunner_GetSchema_Call {
|
||
|
_c.Call.Run(func(args mock.Arguments) {
|
||
|
run()
|
||
|
})
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_GetSchema_Call) Return(_a0 *schemapb.FunctionSchema) *MockFunctionRunner_GetSchema_Call {
|
||
|
_c.Call.Return(_a0)
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
func (_c *MockFunctionRunner_GetSchema_Call) RunAndReturn(run func() *schemapb.FunctionSchema) *MockFunctionRunner_GetSchema_Call {
|
||
|
_c.Call.Return(run)
|
||
|
return _c
|
||
|
}
|
||
|
|
||
|
// NewMockFunctionRunner creates a new instance of MockFunctionRunner. 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 NewMockFunctionRunner(t interface {
|
||
|
mock.TestingT
|
||
|
Cleanup(func())
|
||
|
}) *MockFunctionRunner {
|
||
|
mock := &MockFunctionRunner{}
|
||
|
mock.Mock.Test(t)
|
||
|
|
||
|
t.Cleanup(func() { mock.AssertExpectations(t) })
|
||
|
|
||
|
return mock
|
||
|
}
|