2023-09-04 21:19:48 +08:00
// Code generated by mockery v2.32.4. DO NOT EDIT.
2023-05-10 17:45:20 +08:00
package datacoord
import (
context "context"
datapb "github.com/milvus-io/milvus/internal/proto/datapb"
mock "github.com/stretchr/testify/mock"
)
// NMockHandler is an autogenerated mock type for the Handler type
type NMockHandler struct {
mock . Mock
}
type NMockHandler_Expecter struct {
mock * mock . Mock
}
func ( _m * NMockHandler ) EXPECT ( ) * NMockHandler_Expecter {
return & NMockHandler_Expecter { mock : & _m . Mock }
}
2023-10-24 17:46:15 +08:00
// CheckShouldDropChannel provides a mock function with given fields: ch
func ( _m * NMockHandler ) CheckShouldDropChannel ( ch string ) bool {
ret := _m . Called ( ch )
2023-05-10 17:45:20 +08:00
var r0 bool
2023-10-24 17:46:15 +08:00
if rf , ok := ret . Get ( 0 ) . ( func ( string ) bool ) ; ok {
r0 = rf ( ch )
2023-05-10 17:45:20 +08:00
} else {
r0 = ret . Get ( 0 ) . ( bool )
}
return r0
}
// NMockHandler_CheckShouldDropChannel_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CheckShouldDropChannel'
type NMockHandler_CheckShouldDropChannel_Call struct {
* mock . Call
}
// CheckShouldDropChannel is a helper method to define mock.On call
2023-10-24 17:46:15 +08:00
// - ch string
func ( _e * NMockHandler_Expecter ) CheckShouldDropChannel ( ch interface { } ) * NMockHandler_CheckShouldDropChannel_Call {
return & NMockHandler_CheckShouldDropChannel_Call { Call : _e . mock . On ( "CheckShouldDropChannel" , ch ) }
2023-05-10 17:45:20 +08:00
}
2023-10-24 17:46:15 +08:00
func ( _c * NMockHandler_CheckShouldDropChannel_Call ) Run ( run func ( ch string ) ) * NMockHandler_CheckShouldDropChannel_Call {
2023-05-10 17:45:20 +08:00
_c . Call . Run ( func ( args mock . Arguments ) {
2023-10-24 17:46:15 +08:00
run ( args [ 0 ] . ( string ) )
2023-05-10 17:45:20 +08:00
} )
return _c
}
func ( _c * NMockHandler_CheckShouldDropChannel_Call ) Return ( _a0 bool ) * NMockHandler_CheckShouldDropChannel_Call {
_c . Call . Return ( _a0 )
return _c
}
2023-10-24 17:46:15 +08:00
func ( _c * NMockHandler_CheckShouldDropChannel_Call ) RunAndReturn ( run func ( string ) bool ) * NMockHandler_CheckShouldDropChannel_Call {
2023-09-04 21:19:48 +08:00
_c . Call . Return ( run )
return _c
}
2024-04-15 16:33:19 +08:00
// FinishDropChannel provides a mock function with given fields: ch, collectionID
func ( _m * NMockHandler ) FinishDropChannel ( ch string , collectionID int64 ) error {
ret := _m . Called ( ch , collectionID )
2023-05-10 17:45:20 +08:00
var r0 error
2024-04-15 16:33:19 +08:00
if rf , ok := ret . Get ( 0 ) . ( func ( string , int64 ) error ) ; ok {
r0 = rf ( ch , collectionID )
2023-05-10 17:45:20 +08:00
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// NMockHandler_FinishDropChannel_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'FinishDropChannel'
type NMockHandler_FinishDropChannel_Call struct {
* mock . Call
}
// FinishDropChannel is a helper method to define mock.On call
2023-10-24 17:46:15 +08:00
// - ch string
2024-04-15 16:33:19 +08:00
// - collectionID int64
func ( _e * NMockHandler_Expecter ) FinishDropChannel ( ch interface { } , collectionID interface { } ) * NMockHandler_FinishDropChannel_Call {
return & NMockHandler_FinishDropChannel_Call { Call : _e . mock . On ( "FinishDropChannel" , ch , collectionID ) }
2023-05-10 17:45:20 +08:00
}
2024-04-15 16:33:19 +08:00
func ( _c * NMockHandler_FinishDropChannel_Call ) Run ( run func ( ch string , collectionID int64 ) ) * NMockHandler_FinishDropChannel_Call {
2023-05-10 17:45:20 +08:00
_c . Call . Run ( func ( args mock . Arguments ) {
2024-04-15 16:33:19 +08:00
run ( args [ 0 ] . ( string ) , args [ 1 ] . ( int64 ) )
2023-05-10 17:45:20 +08:00
} )
return _c
}
func ( _c * NMockHandler_FinishDropChannel_Call ) Return ( _a0 error ) * NMockHandler_FinishDropChannel_Call {
_c . Call . Return ( _a0 )
return _c
}
2024-04-15 16:33:19 +08:00
func ( _c * NMockHandler_FinishDropChannel_Call ) RunAndReturn ( run func ( string , int64 ) error ) * NMockHandler_FinishDropChannel_Call {
2023-09-04 21:19:48 +08:00
_c . Call . Return ( run )
return _c
}
2023-05-10 17:45:20 +08:00
// GetCollection provides a mock function with given fields: ctx, collectionID
func ( _m * NMockHandler ) GetCollection ( ctx context . Context , collectionID int64 ) ( * collectionInfo , error ) {
ret := _m . Called ( ctx , collectionID )
var r0 * collectionInfo
2023-09-04 21:19:48 +08:00
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) ( * collectionInfo , error ) ) ; ok {
return rf ( ctx , collectionID )
}
2023-05-10 17:45:20 +08:00
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , int64 ) * collectionInfo ) ; ok {
r0 = rf ( ctx , collectionID )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * collectionInfo )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , int64 ) error ) ; ok {
r1 = rf ( ctx , collectionID )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// NMockHandler_GetCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetCollection'
type NMockHandler_GetCollection_Call struct {
* mock . Call
}
// GetCollection is a helper method to define mock.On call
2023-09-04 21:19:48 +08:00
// - ctx context.Context
// - collectionID int64
2023-05-10 17:45:20 +08:00
func ( _e * NMockHandler_Expecter ) GetCollection ( ctx interface { } , collectionID interface { } ) * NMockHandler_GetCollection_Call {
return & NMockHandler_GetCollection_Call { Call : _e . mock . On ( "GetCollection" , ctx , collectionID ) }
}
func ( _c * NMockHandler_GetCollection_Call ) Run ( run func ( ctx context . Context , collectionID int64 ) ) * NMockHandler_GetCollection_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( int64 ) )
} )
return _c
}
func ( _c * NMockHandler_GetCollection_Call ) Return ( _a0 * collectionInfo , _a1 error ) * NMockHandler_GetCollection_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
2023-09-04 21:19:48 +08:00
func ( _c * NMockHandler_GetCollection_Call ) RunAndReturn ( run func ( context . Context , int64 ) ( * collectionInfo , error ) ) * NMockHandler_GetCollection_Call {
_c . Call . Return ( run )
return _c
}
2023-06-25 17:20:43 +08:00
// GetDataVChanPositions provides a mock function with given fields: ch, partitionID
2023-11-13 11:16:18 +08:00
func ( _m * NMockHandler ) GetDataVChanPositions ( ch RWChannel , partitionID int64 ) * datapb . VchannelInfo {
2023-05-10 17:45:20 +08:00
ret := _m . Called ( ch , partitionID )
var r0 * datapb . VchannelInfo
2023-11-13 11:16:18 +08:00
if rf , ok := ret . Get ( 0 ) . ( func ( RWChannel , int64 ) * datapb . VchannelInfo ) ; ok {
2023-05-10 17:45:20 +08:00
r0 = rf ( ch , partitionID )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . VchannelInfo )
}
}
return r0
}
// NMockHandler_GetDataVChanPositions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetDataVChanPositions'
type NMockHandler_GetDataVChanPositions_Call struct {
* mock . Call
}
// GetDataVChanPositions is a helper method to define mock.On call
2023-11-13 11:16:18 +08:00
// - ch RWChannel
2023-09-04 21:19:48 +08:00
// - partitionID int64
2023-06-25 17:20:43 +08:00
func ( _e * NMockHandler_Expecter ) GetDataVChanPositions ( ch interface { } , partitionID interface { } ) * NMockHandler_GetDataVChanPositions_Call {
return & NMockHandler_GetDataVChanPositions_Call { Call : _e . mock . On ( "GetDataVChanPositions" , ch , partitionID ) }
2023-05-10 17:45:20 +08:00
}
2023-11-13 11:16:18 +08:00
func ( _c * NMockHandler_GetDataVChanPositions_Call ) Run ( run func ( ch RWChannel , partitionID int64 ) ) * NMockHandler_GetDataVChanPositions_Call {
2023-05-10 17:45:20 +08:00
_c . Call . Run ( func ( args mock . Arguments ) {
2023-11-13 11:16:18 +08:00
run ( args [ 0 ] . ( RWChannel ) , args [ 1 ] . ( int64 ) )
2023-05-10 17:45:20 +08:00
} )
return _c
}
func ( _c * NMockHandler_GetDataVChanPositions_Call ) Return ( _a0 * datapb . VchannelInfo ) * NMockHandler_GetDataVChanPositions_Call {
_c . Call . Return ( _a0 )
return _c
}
2023-11-13 11:16:18 +08:00
func ( _c * NMockHandler_GetDataVChanPositions_Call ) RunAndReturn ( run func ( RWChannel , int64 ) * datapb . VchannelInfo ) * NMockHandler_GetDataVChanPositions_Call {
2023-09-04 21:19:48 +08:00
_c . Call . Return ( run )
return _c
}
2023-06-25 17:20:43 +08:00
// GetQueryVChanPositions provides a mock function with given fields: ch, partitionIDs
2023-11-13 11:16:18 +08:00
func ( _m * NMockHandler ) GetQueryVChanPositions ( ch RWChannel , partitionIDs ... int64 ) * datapb . VchannelInfo {
2023-05-10 17:45:20 +08:00
_va := make ( [ ] interface { } , len ( partitionIDs ) )
for _i := range partitionIDs {
_va [ _i ] = partitionIDs [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ch )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . VchannelInfo
2023-11-13 11:16:18 +08:00
if rf , ok := ret . Get ( 0 ) . ( func ( RWChannel , ... int64 ) * datapb . VchannelInfo ) ; ok {
2023-05-10 17:45:20 +08:00
r0 = rf ( ch , partitionIDs ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . VchannelInfo )
}
}
return r0
}
// NMockHandler_GetQueryVChanPositions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetQueryVChanPositions'
type NMockHandler_GetQueryVChanPositions_Call struct {
* mock . Call
}
// GetQueryVChanPositions is a helper method to define mock.On call
2023-11-13 11:16:18 +08:00
// - ch RWChannel
2023-09-04 21:19:48 +08:00
// - partitionIDs ...int64
2023-06-25 17:20:43 +08:00
func ( _e * NMockHandler_Expecter ) GetQueryVChanPositions ( ch interface { } , partitionIDs ... interface { } ) * NMockHandler_GetQueryVChanPositions_Call {
2023-05-10 17:45:20 +08:00
return & NMockHandler_GetQueryVChanPositions_Call { Call : _e . mock . On ( "GetQueryVChanPositions" ,
2023-06-25 17:20:43 +08:00
append ( [ ] interface { } { ch } , partitionIDs ... ) ... ) }
2023-05-10 17:45:20 +08:00
}
2023-11-13 11:16:18 +08:00
func ( _c * NMockHandler_GetQueryVChanPositions_Call ) Run ( run func ( ch RWChannel , partitionIDs ... int64 ) ) * NMockHandler_GetQueryVChanPositions_Call {
2023-05-10 17:45:20 +08:00
_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 )
}
}
2023-11-13 11:16:18 +08:00
run ( args [ 0 ] . ( RWChannel ) , variadicArgs ... )
2023-05-10 17:45:20 +08:00
} )
return _c
}
func ( _c * NMockHandler_GetQueryVChanPositions_Call ) Return ( _a0 * datapb . VchannelInfo ) * NMockHandler_GetQueryVChanPositions_Call {
_c . Call . Return ( _a0 )
return _c
}
2023-11-13 11:16:18 +08:00
func ( _c * NMockHandler_GetQueryVChanPositions_Call ) RunAndReturn ( run func ( RWChannel , ... int64 ) * datapb . VchannelInfo ) * NMockHandler_GetQueryVChanPositions_Call {
2023-09-04 21:19:48 +08:00
_c . Call . Return ( run )
return _c
2023-05-10 17:45:20 +08:00
}
// NewNMockHandler creates a new instance of NMockHandler. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
2023-09-04 21:19:48 +08:00
// The first argument is typically a *testing.T value.
func NewNMockHandler ( t interface {
mock . TestingT
Cleanup ( func ( ) )
} ) * NMockHandler {
2023-05-10 17:45:20 +08:00
mock := & NMockHandler { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}