2023-01-16 13:59:42 +08:00
// Code generated by mockery v2.16.0. DO NOT EDIT.
2022-09-15 18:48:32 +08:00
package meta
import (
context "context"
2023-04-26 10:14:41 +08:00
"github.com/milvus-io/milvus/internal/proto/indexpb"
2022-09-15 18:48:32 +08:00
datapb "github.com/milvus-io/milvus/internal/proto/datapb"
2023-01-16 13:59:42 +08:00
2022-09-15 18:48:32 +08:00
mock "github.com/stretchr/testify/mock"
querypb "github.com/milvus-io/milvus/internal/proto/querypb"
2023-06-09 01:28:37 +08:00
schemapb "github.com/milvus-io/milvus-proto/go-api/v2/schemapb"
2022-09-15 18:48:32 +08:00
)
// 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
2023-04-18 18:30:32 +08:00
// - ctx context.Context
// - collectionID int64
2022-09-15 18:48:32 +08:00
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
2023-04-18 18:30:32 +08:00
// - ctx context.Context
// - collectionID int64
// - segmentID int64
2022-09-15 18:48:32 +08:00
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
}
2023-04-26 10:14:41 +08:00
// 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
}
2022-09-15 18:48:32 +08:00
// 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
2023-04-18 18:30:32 +08:00
// - ctx context.Context
// - collectionID int64
2022-09-15 18:48:32 +08:00
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
2023-04-18 18:30:32 +08:00
// - ctx context.Context
// - collectionID int64
// - partitionID int64
2022-09-15 18:48:32 +08:00
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
}
2023-04-18 18:30:32 +08:00
// 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
}
2022-09-15 18:48:32 +08:00
// GetSegmentInfo provides a mock function with given fields: ctx, segmentID
2022-12-13 16:17:22 +08:00
func ( _m * MockBroker ) GetSegmentInfo ( ctx context . Context , segmentID ... int64 ) ( * datapb . GetSegmentInfoResponse , error ) {
2022-09-15 18:48:32 +08:00
_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 ... )
2022-12-13 16:17:22 +08:00
var r0 * datapb . GetSegmentInfoResponse
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , ... int64 ) * datapb . GetSegmentInfoResponse ) ; ok {
2022-09-15 18:48:32 +08:00
r0 = rf ( ctx , segmentID ... )
} else {
if ret . Get ( 0 ) != nil {
2022-12-13 16:17:22 +08:00
r0 = ret . Get ( 0 ) . ( * datapb . GetSegmentInfoResponse )
2022-09-15 18:48:32 +08:00
}
}
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
2023-04-18 18:30:32 +08:00
// - ctx context.Context
// - segmentID ...int64
2022-09-15 18:48:32 +08:00
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
}
2022-12-13 16:17:22 +08:00
func ( _c * MockBroker_GetSegmentInfo_Call ) Return ( _a0 * datapb . GetSegmentInfoResponse , _a1 error ) * MockBroker_GetSegmentInfo_Call {
2022-09-15 18:48:32 +08:00
_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
}