// Code generated by mockery v2.16.0. DO NOT EDIT. package meta import ( context "context" "github.com/milvus-io/milvus/internal/proto/indexpb" datapb "github.com/milvus-io/milvus/internal/proto/datapb" mock "github.com/stretchr/testify/mock" querypb "github.com/milvus-io/milvus/internal/proto/querypb" schemapb "github.com/milvus-io/milvus-proto/go-api/v2/schemapb" ) // MockBroker is an autogenerated mock type for the Broker type type MockBroker struct { mock.Mock } type MockBroker_Expecter struct { mock *mock.Mock } func (_m *MockBroker) EXPECT() *MockBroker_Expecter { return &MockBroker_Expecter{mock: &_m.Mock} } // GetCollectionSchema provides a mock function with given fields: ctx, collectionID func (_m *MockBroker) GetCollectionSchema(ctx context.Context, collectionID int64) (*schemapb.CollectionSchema, error) { ret := _m.Called(ctx, collectionID) var r0 *schemapb.CollectionSchema if rf, ok := ret.Get(0).(func(context.Context, int64) *schemapb.CollectionSchema); ok { r0 = rf(ctx, collectionID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*schemapb.CollectionSchema) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int64) error); ok { r1 = rf(ctx, collectionID) } else { r1 = ret.Error(1) } return r0, r1 } // MockBroker_GetCollectionSchema_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetCollectionSchema' type MockBroker_GetCollectionSchema_Call struct { *mock.Call } // GetCollectionSchema is a helper method to define mock.On call // - ctx context.Context // - collectionID int64 func (_e *MockBroker_Expecter) GetCollectionSchema(ctx interface{}, collectionID interface{}) *MockBroker_GetCollectionSchema_Call { return &MockBroker_GetCollectionSchema_Call{Call: _e.mock.On("GetCollectionSchema", ctx, collectionID)} } func (_c *MockBroker_GetCollectionSchema_Call) Run(run func(ctx context.Context, collectionID int64)) *MockBroker_GetCollectionSchema_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(context.Context), args[1].(int64)) }) return _c } func (_c *MockBroker_GetCollectionSchema_Call) Return(_a0 *schemapb.CollectionSchema, _a1 error) *MockBroker_GetCollectionSchema_Call { _c.Call.Return(_a0, _a1) return _c } // GetIndexInfo provides a mock function with given fields: ctx, collectionID, segmentID func (_m *MockBroker) GetIndexInfo(ctx context.Context, collectionID int64, segmentID int64) ([]*querypb.FieldIndexInfo, error) { ret := _m.Called(ctx, collectionID, segmentID) var r0 []*querypb.FieldIndexInfo if rf, ok := ret.Get(0).(func(context.Context, int64, int64) []*querypb.FieldIndexInfo); ok { r0 = rf(ctx, collectionID, segmentID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*querypb.FieldIndexInfo) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int64, int64) error); ok { r1 = rf(ctx, collectionID, segmentID) } else { r1 = ret.Error(1) } return r0, r1 } // MockBroker_GetIndexInfo_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetIndexInfo' type MockBroker_GetIndexInfo_Call struct { *mock.Call } // GetIndexInfo is a helper method to define mock.On call // - ctx context.Context // - collectionID int64 // - segmentID int64 func (_e *MockBroker_Expecter) GetIndexInfo(ctx interface{}, collectionID interface{}, segmentID interface{}) *MockBroker_GetIndexInfo_Call { return &MockBroker_GetIndexInfo_Call{Call: _e.mock.On("GetIndexInfo", ctx, collectionID, segmentID)} } func (_c *MockBroker_GetIndexInfo_Call) Run(run func(ctx context.Context, collectionID int64, segmentID int64)) *MockBroker_GetIndexInfo_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(context.Context), args[1].(int64), args[2].(int64)) }) return _c } func (_c *MockBroker_GetIndexInfo_Call) Return(_a0 []*querypb.FieldIndexInfo, _a1 error) *MockBroker_GetIndexInfo_Call { _c.Call.Return(_a0, _a1) return _c } // DescribeIndex provides a mock function with given fields: ctx, collectionID, segmentID func (_m *MockBroker) DescribeIndex(ctx context.Context, collectionID int64) ([]*indexpb.IndexInfo, error) { ret := _m.Called(ctx, collectionID) var r0 []*indexpb.IndexInfo if rf, ok := ret.Get(0).(func(context.Context, int64) []*indexpb.IndexInfo); ok { r0 = rf(ctx, collectionID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*indexpb.IndexInfo) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int64) error); ok { r1 = rf(ctx, collectionID) } else { r1 = ret.Error(1) } return r0, r1 } // MockBroker_DescribeIndex_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DescribeIndex' type MockBroker_DescribeIndex_Call struct { *mock.Call } // DescribeIndex is a helper method to define mock.On call // - ctx context.Context // - collectionID int64 func (_e *MockBroker_Expecter) DescribeIndex(ctx interface{}, collectionID interface{}) *MockBroker_DescribeIndex_Call { return &MockBroker_DescribeIndex_Call{Call: _e.mock.On("DescribeIndex", ctx, collectionID)} } func (_c *MockBroker_DescribeIndex_Call) Run(run func(ctx context.Context, collectionID int64)) *MockBroker_DescribeIndex_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(context.Context), args[1].(int64)) }) return _c } func (_c *MockBroker_DescribeIndex_Call) Return(_a0 []*indexpb.IndexInfo, _a1 error) *MockBroker_DescribeIndex_Call { _c.Call.Return(_a0, _a1) return _c } // GetPartitions provides a mock function with given fields: ctx, collectionID func (_m *MockBroker) GetPartitions(ctx context.Context, collectionID int64) ([]int64, error) { ret := _m.Called(ctx, collectionID) var r0 []int64 if rf, ok := ret.Get(0).(func(context.Context, int64) []int64); ok { r0 = rf(ctx, collectionID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]int64) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, int64) error); ok { r1 = rf(ctx, collectionID) } else { r1 = ret.Error(1) } return r0, r1 } // MockBroker_GetPartitions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetPartitions' type MockBroker_GetPartitions_Call struct { *mock.Call } // GetPartitions is a helper method to define mock.On call // - ctx context.Context // - collectionID int64 func (_e *MockBroker_Expecter) GetPartitions(ctx interface{}, collectionID interface{}) *MockBroker_GetPartitions_Call { return &MockBroker_GetPartitions_Call{Call: _e.mock.On("GetPartitions", ctx, collectionID)} } func (_c *MockBroker_GetPartitions_Call) Run(run func(ctx context.Context, collectionID int64)) *MockBroker_GetPartitions_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(context.Context), args[1].(int64)) }) return _c } func (_c *MockBroker_GetPartitions_Call) Return(_a0 []int64, _a1 error) *MockBroker_GetPartitions_Call { _c.Call.Return(_a0, _a1) return _c } // GetRecoveryInfo provides a mock function with given fields: ctx, collectionID, partitionID func (_m *MockBroker) GetRecoveryInfo(ctx context.Context, collectionID int64, partitionID int64) ([]*datapb.VchannelInfo, []*datapb.SegmentBinlogs, error) { ret := _m.Called(ctx, collectionID, partitionID) var r0 []*datapb.VchannelInfo if rf, ok := ret.Get(0).(func(context.Context, int64, int64) []*datapb.VchannelInfo); ok { r0 = rf(ctx, collectionID, partitionID) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*datapb.VchannelInfo) } } var r1 []*datapb.SegmentBinlogs if rf, ok := ret.Get(1).(func(context.Context, int64, int64) []*datapb.SegmentBinlogs); ok { r1 = rf(ctx, collectionID, partitionID) } else { if ret.Get(1) != nil { r1 = ret.Get(1).([]*datapb.SegmentBinlogs) } } var r2 error if rf, ok := ret.Get(2).(func(context.Context, int64, int64) error); ok { r2 = rf(ctx, collectionID, partitionID) } else { r2 = ret.Error(2) } return r0, r1, r2 } // MockBroker_GetRecoveryInfo_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRecoveryInfo' type MockBroker_GetRecoveryInfo_Call struct { *mock.Call } // GetRecoveryInfo is a helper method to define mock.On call // - ctx context.Context // - collectionID int64 // - partitionID int64 func (_e *MockBroker_Expecter) GetRecoveryInfo(ctx interface{}, collectionID interface{}, partitionID interface{}) *MockBroker_GetRecoveryInfo_Call { return &MockBroker_GetRecoveryInfo_Call{Call: _e.mock.On("GetRecoveryInfo", ctx, collectionID, partitionID)} } func (_c *MockBroker_GetRecoveryInfo_Call) Run(run func(ctx context.Context, collectionID int64, partitionID int64)) *MockBroker_GetRecoveryInfo_Call { _c.Call.Run(func(args mock.Arguments) { run(args[0].(context.Context), args[1].(int64), args[2].(int64)) }) return _c } func (_c *MockBroker_GetRecoveryInfo_Call) Return(_a0 []*datapb.VchannelInfo, _a1 []*datapb.SegmentBinlogs, _a2 error) *MockBroker_GetRecoveryInfo_Call { _c.Call.Return(_a0, _a1, _a2) return _c } // GetRecoveryInfoV2 provides a mock function with given fields: ctx, collectionID, partitionIDs func (_m *MockBroker) GetRecoveryInfoV2(ctx context.Context, collectionID int64, partitionIDs ...int64) ([]*datapb.VchannelInfo, []*datapb.SegmentInfo, error) { _va := make([]interface{}, len(partitionIDs)) for _i := range partitionIDs { _va[_i] = partitionIDs[_i] } var _ca []interface{} _ca = append(_ca, ctx, collectionID) _ca = append(_ca, _va...) ret := _m.Called(_ca...) var r0 []*datapb.VchannelInfo if rf, ok := ret.Get(0).(func(context.Context, int64, ...int64) []*datapb.VchannelInfo); ok { r0 = rf(ctx, collectionID, partitionIDs...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).([]*datapb.VchannelInfo) } } var r1 []*datapb.SegmentInfo if rf, ok := ret.Get(1).(func(context.Context, int64, ...int64) []*datapb.SegmentInfo); ok { r1 = rf(ctx, collectionID, partitionIDs...) } else { if ret.Get(1) != nil { r1 = ret.Get(1).([]*datapb.SegmentInfo) } } var r2 error if rf, ok := ret.Get(2).(func(context.Context, int64, ...int64) error); ok { r2 = rf(ctx, collectionID, partitionIDs...) } else { r2 = ret.Error(2) } return r0, r1, r2 } // MockBroker_GetRecoveryInfoV2_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRecoveryInfoV2' type MockBroker_GetRecoveryInfoV2_Call struct { *mock.Call } // GetRecoveryInfoV2 is a helper method to define mock.On call // - ctx context.Context // - collectionID int64 // - partitionIDs ...int64 func (_e *MockBroker_Expecter) GetRecoveryInfoV2(ctx interface{}, collectionID interface{}, partitionIDs ...interface{}) *MockBroker_GetRecoveryInfoV2_Call { return &MockBroker_GetRecoveryInfoV2_Call{Call: _e.mock.On("GetRecoveryInfoV2", append([]interface{}{ctx, collectionID}, partitionIDs...)...)} } func (_c *MockBroker_GetRecoveryInfoV2_Call) Run(run func(ctx context.Context, collectionID int64, partitionIDs ...int64)) *MockBroker_GetRecoveryInfoV2_Call { _c.Call.Run(func(args mock.Arguments) { variadicArgs := make([]int64, len(args)-2) for i, a := range args[2:] { if a != nil { variadicArgs[i] = a.(int64) } } run(args[0].(context.Context), args[1].(int64), variadicArgs...) }) return _c } func (_c *MockBroker_GetRecoveryInfoV2_Call) Return(_a0 []*datapb.VchannelInfo, _a1 []*datapb.SegmentInfo, _a2 error) *MockBroker_GetRecoveryInfoV2_Call { _c.Call.Return(_a0, _a1, _a2) return _c } // GetSegmentInfo provides a mock function with given fields: ctx, segmentID func (_m *MockBroker) GetSegmentInfo(ctx context.Context, segmentID ...int64) (*datapb.GetSegmentInfoResponse, error) { _va := make([]interface{}, len(segmentID)) for _i := range segmentID { _va[_i] = segmentID[_i] } var _ca []interface{} _ca = append(_ca, ctx) _ca = append(_ca, _va...) ret := _m.Called(_ca...) var r0 *datapb.GetSegmentInfoResponse if rf, ok := ret.Get(0).(func(context.Context, ...int64) *datapb.GetSegmentInfoResponse); ok { r0 = rf(ctx, segmentID...) } else { if ret.Get(0) != nil { r0 = ret.Get(0).(*datapb.GetSegmentInfoResponse) } } var r1 error if rf, ok := ret.Get(1).(func(context.Context, ...int64) error); ok { r1 = rf(ctx, segmentID...) } else { r1 = ret.Error(1) } return r0, r1 } // MockBroker_GetSegmentInfo_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetSegmentInfo' type MockBroker_GetSegmentInfo_Call struct { *mock.Call } // GetSegmentInfo is a helper method to define mock.On call // - ctx context.Context // - segmentID ...int64 func (_e *MockBroker_Expecter) GetSegmentInfo(ctx interface{}, segmentID ...interface{}) *MockBroker_GetSegmentInfo_Call { return &MockBroker_GetSegmentInfo_Call{Call: _e.mock.On("GetSegmentInfo", append([]interface{}{ctx}, segmentID...)...)} } func (_c *MockBroker_GetSegmentInfo_Call) Run(run func(ctx context.Context, segmentID ...int64)) *MockBroker_GetSegmentInfo_Call { _c.Call.Run(func(args mock.Arguments) { variadicArgs := make([]int64, len(args)-1) for i, a := range args[1:] { if a != nil { variadicArgs[i] = a.(int64) } } run(args[0].(context.Context), variadicArgs...) }) return _c } func (_c *MockBroker_GetSegmentInfo_Call) Return(_a0 *datapb.GetSegmentInfoResponse, _a1 error) *MockBroker_GetSegmentInfo_Call { _c.Call.Return(_a0, _a1) return _c } type mockConstructorTestingTNewMockBroker interface { mock.TestingT Cleanup(func()) } // NewMockBroker creates a new instance of MockBroker. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations. func NewMockBroker(t mockConstructorTestingTNewMockBroker) *MockBroker { mock := &MockBroker{} mock.Mock.Test(t) t.Cleanup(func() { mock.AssertExpectations(t) }) return mock }