2023-09-26 01:57:25 +00:00
// Code generated by mockery v2.32.4. DO NOT EDIT.
package mocks
import (
context "context"
commonpb "github.com/milvus-io/milvus-proto/go-api/v2/commonpb"
datapb "github.com/milvus-io/milvus/internal/proto/datapb"
grpc "google.golang.org/grpc"
internalpb "github.com/milvus-io/milvus/internal/proto/internalpb"
milvuspb "github.com/milvus-io/milvus-proto/go-api/v2/milvuspb"
mock "github.com/stretchr/testify/mock"
)
// MockDataNodeClient is an autogenerated mock type for the DataNodeClient type
type MockDataNodeClient struct {
mock . Mock
}
type MockDataNodeClient_Expecter struct {
mock * mock . Mock
}
func ( _m * MockDataNodeClient ) EXPECT ( ) * MockDataNodeClient_Expecter {
return & MockDataNodeClient_Expecter { mock : & _m . Mock }
}
// CheckChannelOperationProgress provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) CheckChannelOperationProgress ( ctx context . Context , in * datapb . ChannelWatchInfo , opts ... grpc . CallOption ) ( * datapb . ChannelOperationProgressResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . ChannelOperationProgressResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ChannelWatchInfo , ... grpc . CallOption ) ( * datapb . ChannelOperationProgressResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ChannelWatchInfo , ... grpc . CallOption ) * datapb . ChannelOperationProgressResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . ChannelOperationProgressResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . ChannelWatchInfo , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_CheckChannelOperationProgress_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CheckChannelOperationProgress'
type MockDataNodeClient_CheckChannelOperationProgress_Call struct {
* mock . Call
}
// CheckChannelOperationProgress is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.ChannelWatchInfo
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) CheckChannelOperationProgress ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_CheckChannelOperationProgress_Call {
return & MockDataNodeClient_CheckChannelOperationProgress_Call { Call : _e . mock . On ( "CheckChannelOperationProgress" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_CheckChannelOperationProgress_Call ) Run ( run func ( ctx context . Context , in * datapb . ChannelWatchInfo , opts ... grpc . CallOption ) ) * MockDataNodeClient_CheckChannelOperationProgress_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . ChannelWatchInfo ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_CheckChannelOperationProgress_Call ) Return ( _a0 * datapb . ChannelOperationProgressResponse , _a1 error ) * MockDataNodeClient_CheckChannelOperationProgress_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_CheckChannelOperationProgress_Call ) RunAndReturn ( run func ( context . Context , * datapb . ChannelWatchInfo , ... grpc . CallOption ) ( * datapb . ChannelOperationProgressResponse , error ) ) * MockDataNodeClient_CheckChannelOperationProgress_Call {
_c . Call . Return ( run )
return _c
}
// Close provides a mock function with given fields:
func ( _m * MockDataNodeClient ) Close ( ) error {
ret := _m . Called ( )
var r0 error
if rf , ok := ret . Get ( 0 ) . ( func ( ) error ) ; ok {
r0 = rf ( )
} else {
r0 = ret . Error ( 0 )
}
return r0
}
// MockDataNodeClient_Close_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Close'
type MockDataNodeClient_Close_Call struct {
* mock . Call
}
// Close is a helper method to define mock.On call
func ( _e * MockDataNodeClient_Expecter ) Close ( ) * MockDataNodeClient_Close_Call {
return & MockDataNodeClient_Close_Call { Call : _e . mock . On ( "Close" ) }
}
func ( _c * MockDataNodeClient_Close_Call ) Run ( run func ( ) ) * MockDataNodeClient_Close_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
run ( )
} )
return _c
}
func ( _c * MockDataNodeClient_Close_Call ) Return ( _a0 error ) * MockDataNodeClient_Close_Call {
_c . Call . Return ( _a0 )
return _c
}
func ( _c * MockDataNodeClient_Close_Call ) RunAndReturn ( run func ( ) error ) * MockDataNodeClient_Close_Call {
_c . Call . Return ( run )
return _c
}
2024-06-16 14:07:57 +00:00
// CompactionV2 provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) CompactionV2 ( ctx context . Context , in * datapb . CompactionPlan , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
2023-09-26 01:57:25 +00:00
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . CompactionPlan , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . CompactionPlan , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . CompactionPlan , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
2024-06-16 14:07:57 +00:00
// MockDataNodeClient_CompactionV2_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CompactionV2'
type MockDataNodeClient_CompactionV2_Call struct {
2023-09-26 01:57:25 +00:00
* mock . Call
}
2024-06-16 14:07:57 +00:00
// CompactionV2 is a helper method to define mock.On call
// - ctx context.Context
// - in *datapb.CompactionPlan
// - opts ...grpc.CallOption
func ( _e * MockDataNodeClient_Expecter ) CompactionV2 ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_CompactionV2_Call {
return & MockDataNodeClient_CompactionV2_Call { Call : _e . mock . On ( "CompactionV2" ,
2023-09-26 01:57:25 +00:00
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
2024-06-16 14:07:57 +00:00
func ( _c * MockDataNodeClient_CompactionV2_Call ) Run ( run func ( ctx context . Context , in * datapb . CompactionPlan , opts ... grpc . CallOption ) ) * MockDataNodeClient_CompactionV2_Call {
2023-09-26 01:57:25 +00:00
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . CompactionPlan ) , variadicArgs ... )
} )
return _c
}
2024-06-16 14:07:57 +00:00
func ( _c * MockDataNodeClient_CompactionV2_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_CompactionV2_Call {
2023-09-26 01:57:25 +00:00
_c . Call . Return ( _a0 , _a1 )
return _c
}
2024-06-16 14:07:57 +00:00
func ( _c * MockDataNodeClient_CompactionV2_Call ) RunAndReturn ( run func ( context . Context , * datapb . CompactionPlan , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_CompactionV2_Call {
2023-09-26 01:57:25 +00:00
_c . Call . Return ( run )
return _c
}
2024-06-05 12:27:51 +00:00
// DropCompactionPlan provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) DropCompactionPlan ( ctx context . Context , in * datapb . DropCompactionPlanRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . DropCompactionPlanRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . DropCompactionPlanRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . DropCompactionPlanRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_DropCompactionPlan_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropCompactionPlan'
type MockDataNodeClient_DropCompactionPlan_Call struct {
* mock . Call
}
// DropCompactionPlan is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.DropCompactionPlanRequest
// - opts ...grpc.CallOption
2024-06-05 12:27:51 +00:00
func ( _e * MockDataNodeClient_Expecter ) DropCompactionPlan ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_DropCompactionPlan_Call {
return & MockDataNodeClient_DropCompactionPlan_Call { Call : _e . mock . On ( "DropCompactionPlan" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_DropCompactionPlan_Call ) Run ( run func ( ctx context . Context , in * datapb . DropCompactionPlanRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_DropCompactionPlan_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . DropCompactionPlanRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_DropCompactionPlan_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_DropCompactionPlan_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_DropCompactionPlan_Call ) RunAndReturn ( run func ( context . Context , * datapb . DropCompactionPlanRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_DropCompactionPlan_Call {
_c . Call . Return ( run )
return _c
}
2023-12-04 11:56:35 +00:00
// DropImport provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) DropImport ( ctx context . Context , in * datapb . DropImportRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . DropImportRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . DropImportRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . DropImportRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_DropImport_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropImport'
type MockDataNodeClient_DropImport_Call struct {
* mock . Call
}
// DropImport is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.DropImportRequest
// - opts ...grpc.CallOption
2023-12-04 11:56:35 +00:00
func ( _e * MockDataNodeClient_Expecter ) DropImport ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_DropImport_Call {
return & MockDataNodeClient_DropImport_Call { Call : _e . mock . On ( "DropImport" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_DropImport_Call ) Run ( run func ( ctx context . Context , in * datapb . DropImportRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_DropImport_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . DropImportRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_DropImport_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_DropImport_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_DropImport_Call ) RunAndReturn ( run func ( context . Context , * datapb . DropImportRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_DropImport_Call {
_c . Call . Return ( run )
return _c
}
2023-09-26 01:57:25 +00:00
// FlushChannels provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) FlushChannels ( ctx context . Context , in * datapb . FlushChannelsRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . FlushChannelsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . FlushChannelsRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . FlushChannelsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_FlushChannels_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'FlushChannels'
type MockDataNodeClient_FlushChannels_Call struct {
* mock . Call
}
// FlushChannels is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.FlushChannelsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) FlushChannels ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_FlushChannels_Call {
return & MockDataNodeClient_FlushChannels_Call { Call : _e . mock . On ( "FlushChannels" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_FlushChannels_Call ) Run ( run func ( ctx context . Context , in * datapb . FlushChannelsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_FlushChannels_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . FlushChannelsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_FlushChannels_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_FlushChannels_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_FlushChannels_Call ) RunAndReturn ( run func ( context . Context , * datapb . FlushChannelsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_FlushChannels_Call {
_c . Call . Return ( run )
return _c
}
// FlushSegments provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) FlushSegments ( ctx context . Context , in * datapb . FlushSegmentsRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . FlushSegmentsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . FlushSegmentsRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . FlushSegmentsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_FlushSegments_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'FlushSegments'
type MockDataNodeClient_FlushSegments_Call struct {
* mock . Call
}
// FlushSegments is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.FlushSegmentsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) FlushSegments ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_FlushSegments_Call {
return & MockDataNodeClient_FlushSegments_Call { Call : _e . mock . On ( "FlushSegments" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_FlushSegments_Call ) Run ( run func ( ctx context . Context , in * datapb . FlushSegmentsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_FlushSegments_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . FlushSegmentsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_FlushSegments_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_FlushSegments_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_FlushSegments_Call ) RunAndReturn ( run func ( context . Context , * datapb . FlushSegmentsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_FlushSegments_Call {
_c . Call . Return ( run )
return _c
}
// GetCompactionState provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) GetCompactionState ( ctx context . Context , in * datapb . CompactionStateRequest , opts ... grpc . CallOption ) ( * datapb . CompactionStateResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . CompactionStateResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . CompactionStateRequest , ... grpc . CallOption ) ( * datapb . CompactionStateResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . CompactionStateRequest , ... grpc . CallOption ) * datapb . CompactionStateResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . CompactionStateResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . CompactionStateRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_GetCompactionState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetCompactionState'
type MockDataNodeClient_GetCompactionState_Call struct {
* mock . Call
}
// GetCompactionState is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.CompactionStateRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) GetCompactionState ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_GetCompactionState_Call {
return & MockDataNodeClient_GetCompactionState_Call { Call : _e . mock . On ( "GetCompactionState" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_GetCompactionState_Call ) Run ( run func ( ctx context . Context , in * datapb . CompactionStateRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_GetCompactionState_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . CompactionStateRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_GetCompactionState_Call ) Return ( _a0 * datapb . CompactionStateResponse , _a1 error ) * MockDataNodeClient_GetCompactionState_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_GetCompactionState_Call ) RunAndReturn ( run func ( context . Context , * datapb . CompactionStateRequest , ... grpc . CallOption ) ( * datapb . CompactionStateResponse , error ) ) * MockDataNodeClient_GetCompactionState_Call {
_c . Call . Return ( run )
return _c
}
// GetComponentStates provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) GetComponentStates ( ctx context . Context , in * milvuspb . GetComponentStatesRequest , opts ... grpc . CallOption ) ( * milvuspb . ComponentStates , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * milvuspb . ComponentStates
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * milvuspb . GetComponentStatesRequest , ... grpc . CallOption ) ( * milvuspb . ComponentStates , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * milvuspb . GetComponentStatesRequest , ... grpc . CallOption ) * milvuspb . ComponentStates ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * milvuspb . ComponentStates )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * milvuspb . GetComponentStatesRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_GetComponentStates_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetComponentStates'
type MockDataNodeClient_GetComponentStates_Call struct {
* mock . Call
}
// GetComponentStates is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *milvuspb.GetComponentStatesRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) GetComponentStates ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_GetComponentStates_Call {
return & MockDataNodeClient_GetComponentStates_Call { Call : _e . mock . On ( "GetComponentStates" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_GetComponentStates_Call ) Run ( run func ( ctx context . Context , in * milvuspb . GetComponentStatesRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_GetComponentStates_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * milvuspb . GetComponentStatesRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_GetComponentStates_Call ) Return ( _a0 * milvuspb . ComponentStates , _a1 error ) * MockDataNodeClient_GetComponentStates_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_GetComponentStates_Call ) RunAndReturn ( run func ( context . Context , * milvuspb . GetComponentStatesRequest , ... grpc . CallOption ) ( * milvuspb . ComponentStates , error ) ) * MockDataNodeClient_GetComponentStates_Call {
_c . Call . Return ( run )
return _c
}
// GetMetrics provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) GetMetrics ( ctx context . Context , in * milvuspb . GetMetricsRequest , opts ... grpc . CallOption ) ( * milvuspb . GetMetricsResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * milvuspb . GetMetricsResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * milvuspb . GetMetricsRequest , ... grpc . CallOption ) ( * milvuspb . GetMetricsResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * milvuspb . GetMetricsRequest , ... grpc . CallOption ) * milvuspb . GetMetricsResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * milvuspb . GetMetricsResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * milvuspb . GetMetricsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_GetMetrics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetMetrics'
type MockDataNodeClient_GetMetrics_Call struct {
* mock . Call
}
// GetMetrics is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *milvuspb.GetMetricsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) GetMetrics ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_GetMetrics_Call {
return & MockDataNodeClient_GetMetrics_Call { Call : _e . mock . On ( "GetMetrics" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_GetMetrics_Call ) Run ( run func ( ctx context . Context , in * milvuspb . GetMetricsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_GetMetrics_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * milvuspb . GetMetricsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_GetMetrics_Call ) Return ( _a0 * milvuspb . GetMetricsResponse , _a1 error ) * MockDataNodeClient_GetMetrics_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_GetMetrics_Call ) RunAndReturn ( run func ( context . Context , * milvuspb . GetMetricsRequest , ... grpc . CallOption ) ( * milvuspb . GetMetricsResponse , error ) ) * MockDataNodeClient_GetMetrics_Call {
_c . Call . Return ( run )
return _c
}
// GetStatisticsChannel provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) GetStatisticsChannel ( ctx context . Context , in * internalpb . GetStatisticsChannelRequest , opts ... grpc . CallOption ) ( * milvuspb . StringResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * milvuspb . StringResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * internalpb . GetStatisticsChannelRequest , ... grpc . CallOption ) ( * milvuspb . StringResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * internalpb . GetStatisticsChannelRequest , ... grpc . CallOption ) * milvuspb . StringResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * milvuspb . StringResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * internalpb . GetStatisticsChannelRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_GetStatisticsChannel_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetStatisticsChannel'
type MockDataNodeClient_GetStatisticsChannel_Call struct {
* mock . Call
}
// GetStatisticsChannel is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *internalpb.GetStatisticsChannelRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) GetStatisticsChannel ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_GetStatisticsChannel_Call {
return & MockDataNodeClient_GetStatisticsChannel_Call { Call : _e . mock . On ( "GetStatisticsChannel" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_GetStatisticsChannel_Call ) Run ( run func ( ctx context . Context , in * internalpb . GetStatisticsChannelRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_GetStatisticsChannel_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * internalpb . GetStatisticsChannelRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_GetStatisticsChannel_Call ) Return ( _a0 * milvuspb . StringResponse , _a1 error ) * MockDataNodeClient_GetStatisticsChannel_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_GetStatisticsChannel_Call ) RunAndReturn ( run func ( context . Context , * internalpb . GetStatisticsChannelRequest , ... grpc . CallOption ) ( * milvuspb . StringResponse , error ) ) * MockDataNodeClient_GetStatisticsChannel_Call {
_c . Call . Return ( run )
return _c
}
2023-12-04 11:56:35 +00:00
// ImportV2 provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) ImportV2 ( ctx context . Context , in * datapb . ImportRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ImportRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ImportRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . ImportRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_ImportV2_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ImportV2'
type MockDataNodeClient_ImportV2_Call struct {
* mock . Call
}
// ImportV2 is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.ImportRequest
// - opts ...grpc.CallOption
2023-12-04 11:56:35 +00:00
func ( _e * MockDataNodeClient_Expecter ) ImportV2 ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_ImportV2_Call {
return & MockDataNodeClient_ImportV2_Call { Call : _e . mock . On ( "ImportV2" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_ImportV2_Call ) Run ( run func ( ctx context . Context , in * datapb . ImportRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_ImportV2_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . ImportRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_ImportV2_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_ImportV2_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_ImportV2_Call ) RunAndReturn ( run func ( context . Context , * datapb . ImportRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_ImportV2_Call {
_c . Call . Return ( run )
return _c
}
2023-09-26 01:57:25 +00:00
// NotifyChannelOperation provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) NotifyChannelOperation ( ctx context . Context , in * datapb . ChannelOperationsRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ChannelOperationsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ChannelOperationsRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . ChannelOperationsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_NotifyChannelOperation_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'NotifyChannelOperation'
type MockDataNodeClient_NotifyChannelOperation_Call struct {
* mock . Call
}
// NotifyChannelOperation is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.ChannelOperationsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) NotifyChannelOperation ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_NotifyChannelOperation_Call {
return & MockDataNodeClient_NotifyChannelOperation_Call { Call : _e . mock . On ( "NotifyChannelOperation" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_NotifyChannelOperation_Call ) Run ( run func ( ctx context . Context , in * datapb . ChannelOperationsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_NotifyChannelOperation_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . ChannelOperationsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_NotifyChannelOperation_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_NotifyChannelOperation_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_NotifyChannelOperation_Call ) RunAndReturn ( run func ( context . Context , * datapb . ChannelOperationsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_NotifyChannelOperation_Call {
_c . Call . Return ( run )
return _c
}
2023-12-04 11:56:35 +00:00
// PreImport provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) PreImport ( ctx context . Context , in * datapb . PreImportRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . PreImportRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . PreImportRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . PreImportRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_PreImport_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'PreImport'
type MockDataNodeClient_PreImport_Call struct {
* mock . Call
}
// PreImport is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.PreImportRequest
// - opts ...grpc.CallOption
2023-12-04 11:56:35 +00:00
func ( _e * MockDataNodeClient_Expecter ) PreImport ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_PreImport_Call {
return & MockDataNodeClient_PreImport_Call { Call : _e . mock . On ( "PreImport" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_PreImport_Call ) Run ( run func ( ctx context . Context , in * datapb . PreImportRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_PreImport_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . PreImportRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_PreImport_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_PreImport_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_PreImport_Call ) RunAndReturn ( run func ( context . Context , * datapb . PreImportRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_PreImport_Call {
_c . Call . Return ( run )
return _c
}
// QueryImport provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) QueryImport ( ctx context . Context , in * datapb . QueryImportRequest , opts ... grpc . CallOption ) ( * datapb . QueryImportResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . QueryImportResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . QueryImportRequest , ... grpc . CallOption ) ( * datapb . QueryImportResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . QueryImportRequest , ... grpc . CallOption ) * datapb . QueryImportResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . QueryImportResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . QueryImportRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_QueryImport_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'QueryImport'
type MockDataNodeClient_QueryImport_Call struct {
* mock . Call
}
// QueryImport is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.QueryImportRequest
// - opts ...grpc.CallOption
2023-12-04 11:56:35 +00:00
func ( _e * MockDataNodeClient_Expecter ) QueryImport ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_QueryImport_Call {
return & MockDataNodeClient_QueryImport_Call { Call : _e . mock . On ( "QueryImport" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_QueryImport_Call ) Run ( run func ( ctx context . Context , in * datapb . QueryImportRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_QueryImport_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . QueryImportRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_QueryImport_Call ) Return ( _a0 * datapb . QueryImportResponse , _a1 error ) * MockDataNodeClient_QueryImport_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_QueryImport_Call ) RunAndReturn ( run func ( context . Context , * datapb . QueryImportRequest , ... grpc . CallOption ) ( * datapb . QueryImportResponse , error ) ) * MockDataNodeClient_QueryImport_Call {
_c . Call . Return ( run )
return _c
}
// QueryPreImport provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) QueryPreImport ( ctx context . Context , in * datapb . QueryPreImportRequest , opts ... grpc . CallOption ) ( * datapb . QueryPreImportResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . QueryPreImportResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . QueryPreImportRequest , ... grpc . CallOption ) ( * datapb . QueryPreImportResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . QueryPreImportRequest , ... grpc . CallOption ) * datapb . QueryPreImportResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . QueryPreImportResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . QueryPreImportRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_QueryPreImport_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'QueryPreImport'
type MockDataNodeClient_QueryPreImport_Call struct {
* mock . Call
}
// QueryPreImport is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.QueryPreImportRequest
// - opts ...grpc.CallOption
2023-12-04 11:56:35 +00:00
func ( _e * MockDataNodeClient_Expecter ) QueryPreImport ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_QueryPreImport_Call {
return & MockDataNodeClient_QueryPreImport_Call { Call : _e . mock . On ( "QueryPreImport" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_QueryPreImport_Call ) Run ( run func ( ctx context . Context , in * datapb . QueryPreImportRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_QueryPreImport_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . QueryPreImportRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_QueryPreImport_Call ) Return ( _a0 * datapb . QueryPreImportResponse , _a1 error ) * MockDataNodeClient_QueryPreImport_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_QueryPreImport_Call ) RunAndReturn ( run func ( context . Context , * datapb . QueryPreImportRequest , ... grpc . CallOption ) ( * datapb . QueryPreImportResponse , error ) ) * MockDataNodeClient_QueryPreImport_Call {
_c . Call . Return ( run )
return _c
}
2024-05-17 10:19:38 +00:00
// QuerySlot provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) QuerySlot ( ctx context . Context , in * datapb . QuerySlotRequest , opts ... grpc . CallOption ) ( * datapb . QuerySlotResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . QuerySlotResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . QuerySlotRequest , ... grpc . CallOption ) ( * datapb . QuerySlotResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . QuerySlotRequest , ... grpc . CallOption ) * datapb . QuerySlotResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . QuerySlotResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . QuerySlotRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_QuerySlot_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'QuerySlot'
type MockDataNodeClient_QuerySlot_Call struct {
* mock . Call
}
// QuerySlot is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.QuerySlotRequest
// - opts ...grpc.CallOption
2024-05-17 10:19:38 +00:00
func ( _e * MockDataNodeClient_Expecter ) QuerySlot ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_QuerySlot_Call {
return & MockDataNodeClient_QuerySlot_Call { Call : _e . mock . On ( "QuerySlot" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_QuerySlot_Call ) Run ( run func ( ctx context . Context , in * datapb . QuerySlotRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_QuerySlot_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . QuerySlotRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_QuerySlot_Call ) Return ( _a0 * datapb . QuerySlotResponse , _a1 error ) * MockDataNodeClient_QuerySlot_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_QuerySlot_Call ) RunAndReturn ( run func ( context . Context , * datapb . QuerySlotRequest , ... grpc . CallOption ) ( * datapb . QuerySlotResponse , error ) ) * MockDataNodeClient_QuerySlot_Call {
_c . Call . Return ( run )
return _c
}
2023-09-26 01:57:25 +00:00
// ResendSegmentStats provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) ResendSegmentStats ( ctx context . Context , in * datapb . ResendSegmentStatsRequest , opts ... grpc . CallOption ) ( * datapb . ResendSegmentStatsResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * datapb . ResendSegmentStatsResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ResendSegmentStatsRequest , ... grpc . CallOption ) ( * datapb . ResendSegmentStatsResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . ResendSegmentStatsRequest , ... grpc . CallOption ) * datapb . ResendSegmentStatsResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * datapb . ResendSegmentStatsResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . ResendSegmentStatsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_ResendSegmentStats_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ResendSegmentStats'
type MockDataNodeClient_ResendSegmentStats_Call struct {
* mock . Call
}
// ResendSegmentStats is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.ResendSegmentStatsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) ResendSegmentStats ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_ResendSegmentStats_Call {
return & MockDataNodeClient_ResendSegmentStats_Call { Call : _e . mock . On ( "ResendSegmentStats" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_ResendSegmentStats_Call ) Run ( run func ( ctx context . Context , in * datapb . ResendSegmentStatsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_ResendSegmentStats_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . ResendSegmentStatsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_ResendSegmentStats_Call ) Return ( _a0 * datapb . ResendSegmentStatsResponse , _a1 error ) * MockDataNodeClient_ResendSegmentStats_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_ResendSegmentStats_Call ) RunAndReturn ( run func ( context . Context , * datapb . ResendSegmentStatsRequest , ... grpc . CallOption ) ( * datapb . ResendSegmentStatsResponse , error ) ) * MockDataNodeClient_ResendSegmentStats_Call {
_c . Call . Return ( run )
return _c
}
// ShowConfigurations provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) ShowConfigurations ( ctx context . Context , in * internalpb . ShowConfigurationsRequest , opts ... grpc . CallOption ) ( * internalpb . ShowConfigurationsResponse , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * internalpb . ShowConfigurationsResponse
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * internalpb . ShowConfigurationsRequest , ... grpc . CallOption ) ( * internalpb . ShowConfigurationsResponse , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * internalpb . ShowConfigurationsRequest , ... grpc . CallOption ) * internalpb . ShowConfigurationsResponse ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * internalpb . ShowConfigurationsResponse )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * internalpb . ShowConfigurationsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_ShowConfigurations_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ShowConfigurations'
type MockDataNodeClient_ShowConfigurations_Call struct {
* mock . Call
}
// ShowConfigurations is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *internalpb.ShowConfigurationsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) ShowConfigurations ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_ShowConfigurations_Call {
return & MockDataNodeClient_ShowConfigurations_Call { Call : _e . mock . On ( "ShowConfigurations" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_ShowConfigurations_Call ) Run ( run func ( ctx context . Context , in * internalpb . ShowConfigurationsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_ShowConfigurations_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * internalpb . ShowConfigurationsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_ShowConfigurations_Call ) Return ( _a0 * internalpb . ShowConfigurationsResponse , _a1 error ) * MockDataNodeClient_ShowConfigurations_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_ShowConfigurations_Call ) RunAndReturn ( run func ( context . Context , * internalpb . ShowConfigurationsRequest , ... grpc . CallOption ) ( * internalpb . ShowConfigurationsResponse , error ) ) * MockDataNodeClient_ShowConfigurations_Call {
_c . Call . Return ( run )
return _c
}
// SyncSegments provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) SyncSegments ( ctx context . Context , in * datapb . SyncSegmentsRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . SyncSegmentsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . SyncSegmentsRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . SyncSegmentsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_SyncSegments_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SyncSegments'
type MockDataNodeClient_SyncSegments_Call struct {
* mock . Call
}
// SyncSegments is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.SyncSegmentsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) SyncSegments ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_SyncSegments_Call {
return & MockDataNodeClient_SyncSegments_Call { Call : _e . mock . On ( "SyncSegments" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_SyncSegments_Call ) Run ( run func ( ctx context . Context , in * datapb . SyncSegmentsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_SyncSegments_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . SyncSegmentsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_SyncSegments_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_SyncSegments_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_SyncSegments_Call ) RunAndReturn ( run func ( context . Context , * datapb . SyncSegmentsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_SyncSegments_Call {
_c . Call . Return ( run )
return _c
}
// WatchDmChannels provides a mock function with given fields: ctx, in, opts
func ( _m * MockDataNodeClient ) WatchDmChannels ( ctx context . Context , in * datapb . WatchDmChannelsRequest , opts ... grpc . CallOption ) ( * commonpb . Status , error ) {
_va := make ( [ ] interface { } , len ( opts ) )
for _i := range opts {
_va [ _i ] = opts [ _i ]
}
var _ca [ ] interface { }
_ca = append ( _ca , ctx , in )
_ca = append ( _ca , _va ... )
ret := _m . Called ( _ca ... )
var r0 * commonpb . Status
var r1 error
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . WatchDmChannelsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) ; ok {
return rf ( ctx , in , opts ... )
}
if rf , ok := ret . Get ( 0 ) . ( func ( context . Context , * datapb . WatchDmChannelsRequest , ... grpc . CallOption ) * commonpb . Status ) ; ok {
r0 = rf ( ctx , in , opts ... )
} else {
if ret . Get ( 0 ) != nil {
r0 = ret . Get ( 0 ) . ( * commonpb . Status )
}
}
if rf , ok := ret . Get ( 1 ) . ( func ( context . Context , * datapb . WatchDmChannelsRequest , ... grpc . CallOption ) error ) ; ok {
r1 = rf ( ctx , in , opts ... )
} else {
r1 = ret . Error ( 1 )
}
return r0 , r1
}
// MockDataNodeClient_WatchDmChannels_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'WatchDmChannels'
type MockDataNodeClient_WatchDmChannels_Call struct {
* mock . Call
}
// WatchDmChannels is a helper method to define mock.On call
2024-06-16 14:07:57 +00:00
// - ctx context.Context
// - in *datapb.WatchDmChannelsRequest
// - opts ...grpc.CallOption
2023-09-26 01:57:25 +00:00
func ( _e * MockDataNodeClient_Expecter ) WatchDmChannels ( ctx interface { } , in interface { } , opts ... interface { } ) * MockDataNodeClient_WatchDmChannels_Call {
return & MockDataNodeClient_WatchDmChannels_Call { Call : _e . mock . On ( "WatchDmChannels" ,
append ( [ ] interface { } { ctx , in } , opts ... ) ... ) }
}
func ( _c * MockDataNodeClient_WatchDmChannels_Call ) Run ( run func ( ctx context . Context , in * datapb . WatchDmChannelsRequest , opts ... grpc . CallOption ) ) * MockDataNodeClient_WatchDmChannels_Call {
_c . Call . Run ( func ( args mock . Arguments ) {
variadicArgs := make ( [ ] grpc . CallOption , len ( args ) - 2 )
for i , a := range args [ 2 : ] {
if a != nil {
variadicArgs [ i ] = a . ( grpc . CallOption )
}
}
run ( args [ 0 ] . ( context . Context ) , args [ 1 ] . ( * datapb . WatchDmChannelsRequest ) , variadicArgs ... )
} )
return _c
}
func ( _c * MockDataNodeClient_WatchDmChannels_Call ) Return ( _a0 * commonpb . Status , _a1 error ) * MockDataNodeClient_WatchDmChannels_Call {
_c . Call . Return ( _a0 , _a1 )
return _c
}
func ( _c * MockDataNodeClient_WatchDmChannels_Call ) RunAndReturn ( run func ( context . Context , * datapb . WatchDmChannelsRequest , ... grpc . CallOption ) ( * commonpb . Status , error ) ) * MockDataNodeClient_WatchDmChannels_Call {
_c . Call . Return ( run )
return _c
}
// NewMockDataNodeClient creates a new instance of MockDataNodeClient. 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 NewMockDataNodeClient ( t interface {
mock . TestingT
Cleanup ( func ( ) )
} ) * MockDataNodeClient {
mock := & MockDataNodeClient { }
mock . Mock . Test ( t )
t . Cleanup ( func ( ) { mock . AssertExpectations ( t ) } )
return mock
}