// Code generated by mockery v2.16.0. DO NOT EDIT.

package mocks

import (
	context "context"

	commonpb "github.com/milvus-io/milvus-proto/go-api/commonpb"
	clientv3 "go.etcd.io/etcd/client/v3"

	internalpb "github.com/milvus-io/milvus/internal/proto/internalpb"

	milvuspb "github.com/milvus-io/milvus-proto/go-api/milvuspb"

	mock "github.com/stretchr/testify/mock"

	proxypb "github.com/milvus-io/milvus/internal/proto/proxypb"

	types "github.com/milvus-io/milvus/internal/types"
)

// Proxy is an autogenerated mock type for the ProxyComponent type
type Proxy struct {
	mock.Mock
}

type Proxy_Expecter struct {
	mock *mock.Mock
}

func (_m *Proxy) EXPECT() *Proxy_Expecter {
	return &Proxy_Expecter{mock: &_m.Mock}
}

// AlterAlias provides a mock function with given fields: ctx, request
func (_m *Proxy) AlterAlias(ctx context.Context, request *milvuspb.AlterAliasRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterAliasRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AlterAliasRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_AlterAlias_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AlterAlias'
type Proxy_AlterAlias_Call struct {
	*mock.Call
}

// AlterAlias is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.AlterAliasRequest
func (_e *Proxy_Expecter) AlterAlias(ctx interface{}, request interface{}) *Proxy_AlterAlias_Call {
	return &Proxy_AlterAlias_Call{Call: _e.mock.On("AlterAlias", ctx, request)}
}

func (_c *Proxy_AlterAlias_Call) Run(run func(ctx context.Context, request *milvuspb.AlterAliasRequest)) *Proxy_AlterAlias_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.AlterAliasRequest))
	})
	return _c
}

func (_c *Proxy_AlterAlias_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_AlterAlias_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// AlterCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) AlterCollection(ctx context.Context, request *milvuspb.AlterCollectionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.AlterCollectionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.AlterCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_AlterCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'AlterCollection'
type Proxy_AlterCollection_Call struct {
	*mock.Call
}

// AlterCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.AlterCollectionRequest
func (_e *Proxy_Expecter) AlterCollection(ctx interface{}, request interface{}) *Proxy_AlterCollection_Call {
	return &Proxy_AlterCollection_Call{Call: _e.mock.On("AlterCollection", ctx, request)}
}

func (_c *Proxy_AlterCollection_Call) Run(run func(ctx context.Context, request *milvuspb.AlterCollectionRequest)) *Proxy_AlterCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.AlterCollectionRequest))
	})
	return _c
}

func (_c *Proxy_AlterCollection_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_AlterCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CalcDistance provides a mock function with given fields: ctx, request
func (_m *Proxy) CalcDistance(ctx context.Context, request *milvuspb.CalcDistanceRequest) (*milvuspb.CalcDistanceResults, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.CalcDistanceResults
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CalcDistanceRequest) *milvuspb.CalcDistanceResults); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.CalcDistanceResults)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CalcDistanceRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CalcDistance_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CalcDistance'
type Proxy_CalcDistance_Call struct {
	*mock.Call
}

// CalcDistance is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.CalcDistanceRequest
func (_e *Proxy_Expecter) CalcDistance(ctx interface{}, request interface{}) *Proxy_CalcDistance_Call {
	return &Proxy_CalcDistance_Call{Call: _e.mock.On("CalcDistance", ctx, request)}
}

func (_c *Proxy_CalcDistance_Call) Run(run func(ctx context.Context, request *milvuspb.CalcDistanceRequest)) *Proxy_CalcDistance_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CalcDistanceRequest))
	})
	return _c
}

func (_c *Proxy_CalcDistance_Call) Return(_a0 *milvuspb.CalcDistanceResults, _a1 error) *Proxy_CalcDistance_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CheckHealth provides a mock function with given fields: ctx, req
func (_m *Proxy) CheckHealth(ctx context.Context, req *milvuspb.CheckHealthRequest) (*milvuspb.CheckHealthResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.CheckHealthResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CheckHealthRequest) *milvuspb.CheckHealthResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.CheckHealthResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CheckHealthRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CheckHealth_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CheckHealth'
type Proxy_CheckHealth_Call struct {
	*mock.Call
}

// CheckHealth is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.CheckHealthRequest
func (_e *Proxy_Expecter) CheckHealth(ctx interface{}, req interface{}) *Proxy_CheckHealth_Call {
	return &Proxy_CheckHealth_Call{Call: _e.mock.On("CheckHealth", ctx, req)}
}

func (_c *Proxy_CheckHealth_Call) Run(run func(ctx context.Context, req *milvuspb.CheckHealthRequest)) *Proxy_CheckHealth_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CheckHealthRequest))
	})
	return _c
}

func (_c *Proxy_CheckHealth_Call) Return(_a0 *milvuspb.CheckHealthResponse, _a1 error) *Proxy_CheckHealth_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Connect provides a mock function with given fields: ctx, req
func (_m *Proxy) Connect(ctx context.Context, req *milvuspb.ConnectRequest) (*milvuspb.ConnectResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.ConnectResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ConnectRequest) *milvuspb.ConnectResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ConnectResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ConnectRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Connect_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Connect'
type Proxy_Connect_Call struct {
	*mock.Call
}

// Connect is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.ConnectRequest
func (_e *Proxy_Expecter) Connect(ctx interface{}, req interface{}) *Proxy_Connect_Call {
	return &Proxy_Connect_Call{Call: _e.mock.On("Connect", ctx, req)}
}

func (_c *Proxy_Connect_Call) Run(run func(ctx context.Context, req *milvuspb.ConnectRequest)) *Proxy_Connect_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ConnectRequest))
	})
	return _c
}

func (_c *Proxy_Connect_Call) Return(_a0 *milvuspb.ConnectResponse, _a1 error) *Proxy_Connect_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreateAlias provides a mock function with given fields: ctx, request
func (_m *Proxy) CreateAlias(ctx context.Context, request *milvuspb.CreateAliasRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateAliasRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateAliasRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreateAlias_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateAlias'
type Proxy_CreateAlias_Call struct {
	*mock.Call
}

// CreateAlias is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.CreateAliasRequest
func (_e *Proxy_Expecter) CreateAlias(ctx interface{}, request interface{}) *Proxy_CreateAlias_Call {
	return &Proxy_CreateAlias_Call{Call: _e.mock.On("CreateAlias", ctx, request)}
}

func (_c *Proxy_CreateAlias_Call) Run(run func(ctx context.Context, request *milvuspb.CreateAliasRequest)) *Proxy_CreateAlias_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreateAliasRequest))
	})
	return _c
}

func (_c *Proxy_CreateAlias_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreateAlias_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreateCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) CreateCollection(ctx context.Context, request *milvuspb.CreateCollectionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCollectionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreateCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateCollection'
type Proxy_CreateCollection_Call struct {
	*mock.Call
}

// CreateCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.CreateCollectionRequest
func (_e *Proxy_Expecter) CreateCollection(ctx interface{}, request interface{}) *Proxy_CreateCollection_Call {
	return &Proxy_CreateCollection_Call{Call: _e.mock.On("CreateCollection", ctx, request)}
}

func (_c *Proxy_CreateCollection_Call) Run(run func(ctx context.Context, request *milvuspb.CreateCollectionRequest)) *Proxy_CreateCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreateCollectionRequest))
	})
	return _c
}

func (_c *Proxy_CreateCollection_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreateCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreateCredential provides a mock function with given fields: ctx, req
func (_m *Proxy) CreateCredential(ctx context.Context, req *milvuspb.CreateCredentialRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateCredentialRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateCredentialRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreateCredential_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateCredential'
type Proxy_CreateCredential_Call struct {
	*mock.Call
}

// CreateCredential is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.CreateCredentialRequest
func (_e *Proxy_Expecter) CreateCredential(ctx interface{}, req interface{}) *Proxy_CreateCredential_Call {
	return &Proxy_CreateCredential_Call{Call: _e.mock.On("CreateCredential", ctx, req)}
}

func (_c *Proxy_CreateCredential_Call) Run(run func(ctx context.Context, req *milvuspb.CreateCredentialRequest)) *Proxy_CreateCredential_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreateCredentialRequest))
	})
	return _c
}

func (_c *Proxy_CreateCredential_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreateCredential_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreateIndex provides a mock function with given fields: ctx, request
func (_m *Proxy) CreateIndex(ctx context.Context, request *milvuspb.CreateIndexRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateIndexRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateIndexRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreateIndex_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateIndex'
type Proxy_CreateIndex_Call struct {
	*mock.Call
}

// CreateIndex is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.CreateIndexRequest
func (_e *Proxy_Expecter) CreateIndex(ctx interface{}, request interface{}) *Proxy_CreateIndex_Call {
	return &Proxy_CreateIndex_Call{Call: _e.mock.On("CreateIndex", ctx, request)}
}

func (_c *Proxy_CreateIndex_Call) Run(run func(ctx context.Context, request *milvuspb.CreateIndexRequest)) *Proxy_CreateIndex_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreateIndexRequest))
	})
	return _c
}

func (_c *Proxy_CreateIndex_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreateIndex_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreatePartition provides a mock function with given fields: ctx, request
func (_m *Proxy) CreatePartition(ctx context.Context, request *milvuspb.CreatePartitionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreatePartitionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreatePartitionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreatePartition_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreatePartition'
type Proxy_CreatePartition_Call struct {
	*mock.Call
}

// CreatePartition is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.CreatePartitionRequest
func (_e *Proxy_Expecter) CreatePartition(ctx interface{}, request interface{}) *Proxy_CreatePartition_Call {
	return &Proxy_CreatePartition_Call{Call: _e.mock.On("CreatePartition", ctx, request)}
}

func (_c *Proxy_CreatePartition_Call) Run(run func(ctx context.Context, request *milvuspb.CreatePartitionRequest)) *Proxy_CreatePartition_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreatePartitionRequest))
	})
	return _c
}

func (_c *Proxy_CreatePartition_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreatePartition_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreateResourceGroup provides a mock function with given fields: ctx, req
func (_m *Proxy) CreateResourceGroup(ctx context.Context, req *milvuspb.CreateResourceGroupRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateResourceGroupRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateResourceGroupRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreateResourceGroup_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateResourceGroup'
type Proxy_CreateResourceGroup_Call struct {
	*mock.Call
}

// CreateResourceGroup is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.CreateResourceGroupRequest
func (_e *Proxy_Expecter) CreateResourceGroup(ctx interface{}, req interface{}) *Proxy_CreateResourceGroup_Call {
	return &Proxy_CreateResourceGroup_Call{Call: _e.mock.On("CreateResourceGroup", ctx, req)}
}

func (_c *Proxy_CreateResourceGroup_Call) Run(run func(ctx context.Context, req *milvuspb.CreateResourceGroupRequest)) *Proxy_CreateResourceGroup_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreateResourceGroupRequest))
	})
	return _c
}

func (_c *Proxy_CreateResourceGroup_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreateResourceGroup_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// CreateRole provides a mock function with given fields: ctx, req
func (_m *Proxy) CreateRole(ctx context.Context, req *milvuspb.CreateRoleRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.CreateRoleRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.CreateRoleRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_CreateRole_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'CreateRole'
type Proxy_CreateRole_Call struct {
	*mock.Call
}

// CreateRole is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.CreateRoleRequest
func (_e *Proxy_Expecter) CreateRole(ctx interface{}, req interface{}) *Proxy_CreateRole_Call {
	return &Proxy_CreateRole_Call{Call: _e.mock.On("CreateRole", ctx, req)}
}

func (_c *Proxy_CreateRole_Call) Run(run func(ctx context.Context, req *milvuspb.CreateRoleRequest)) *Proxy_CreateRole_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.CreateRoleRequest))
	})
	return _c
}

func (_c *Proxy_CreateRole_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_CreateRole_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Delete provides a mock function with given fields: ctx, request
func (_m *Proxy) Delete(ctx context.Context, request *milvuspb.DeleteRequest) (*milvuspb.MutationResult, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.MutationResult
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteRequest) *milvuspb.MutationResult); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.MutationResult)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Delete_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Delete'
type Proxy_Delete_Call struct {
	*mock.Call
}

// Delete is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DeleteRequest
func (_e *Proxy_Expecter) Delete(ctx interface{}, request interface{}) *Proxy_Delete_Call {
	return &Proxy_Delete_Call{Call: _e.mock.On("Delete", ctx, request)}
}

func (_c *Proxy_Delete_Call) Run(run func(ctx context.Context, request *milvuspb.DeleteRequest)) *Proxy_Delete_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DeleteRequest))
	})
	return _c
}

func (_c *Proxy_Delete_Call) Return(_a0 *milvuspb.MutationResult, _a1 error) *Proxy_Delete_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DeleteCredential provides a mock function with given fields: ctx, req
func (_m *Proxy) DeleteCredential(ctx context.Context, req *milvuspb.DeleteCredentialRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DeleteCredentialRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DeleteCredentialRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DeleteCredential_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DeleteCredential'
type Proxy_DeleteCredential_Call struct {
	*mock.Call
}

// DeleteCredential is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.DeleteCredentialRequest
func (_e *Proxy_Expecter) DeleteCredential(ctx interface{}, req interface{}) *Proxy_DeleteCredential_Call {
	return &Proxy_DeleteCredential_Call{Call: _e.mock.On("DeleteCredential", ctx, req)}
}

func (_c *Proxy_DeleteCredential_Call) Run(run func(ctx context.Context, req *milvuspb.DeleteCredentialRequest)) *Proxy_DeleteCredential_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DeleteCredentialRequest))
	})
	return _c
}

func (_c *Proxy_DeleteCredential_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DeleteCredential_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DescribeCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) DescribeCollection(ctx context.Context, request *milvuspb.DescribeCollectionRequest) (*milvuspb.DescribeCollectionResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.DescribeCollectionResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeCollectionRequest) *milvuspb.DescribeCollectionResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.DescribeCollectionResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DescribeCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DescribeCollection'
type Proxy_DescribeCollection_Call struct {
	*mock.Call
}

// DescribeCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DescribeCollectionRequest
func (_e *Proxy_Expecter) DescribeCollection(ctx interface{}, request interface{}) *Proxy_DescribeCollection_Call {
	return &Proxy_DescribeCollection_Call{Call: _e.mock.On("DescribeCollection", ctx, request)}
}

func (_c *Proxy_DescribeCollection_Call) Run(run func(ctx context.Context, request *milvuspb.DescribeCollectionRequest)) *Proxy_DescribeCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DescribeCollectionRequest))
	})
	return _c
}

func (_c *Proxy_DescribeCollection_Call) Return(_a0 *milvuspb.DescribeCollectionResponse, _a1 error) *Proxy_DescribeCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DescribeIndex provides a mock function with given fields: ctx, request
func (_m *Proxy) DescribeIndex(ctx context.Context, request *milvuspb.DescribeIndexRequest) (*milvuspb.DescribeIndexResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.DescribeIndexResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeIndexRequest) *milvuspb.DescribeIndexResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.DescribeIndexResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeIndexRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DescribeIndex_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DescribeIndex'
type Proxy_DescribeIndex_Call struct {
	*mock.Call
}

// DescribeIndex is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DescribeIndexRequest
func (_e *Proxy_Expecter) DescribeIndex(ctx interface{}, request interface{}) *Proxy_DescribeIndex_Call {
	return &Proxy_DescribeIndex_Call{Call: _e.mock.On("DescribeIndex", ctx, request)}
}

func (_c *Proxy_DescribeIndex_Call) Run(run func(ctx context.Context, request *milvuspb.DescribeIndexRequest)) *Proxy_DescribeIndex_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DescribeIndexRequest))
	})
	return _c
}

func (_c *Proxy_DescribeIndex_Call) Return(_a0 *milvuspb.DescribeIndexResponse, _a1 error) *Proxy_DescribeIndex_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DescribeResourceGroup provides a mock function with given fields: ctx, req
func (_m *Proxy) DescribeResourceGroup(ctx context.Context, req *milvuspb.DescribeResourceGroupRequest) (*milvuspb.DescribeResourceGroupResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.DescribeResourceGroupResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) *milvuspb.DescribeResourceGroupResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.DescribeResourceGroupResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DescribeResourceGroupRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DescribeResourceGroup_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DescribeResourceGroup'
type Proxy_DescribeResourceGroup_Call struct {
	*mock.Call
}

// DescribeResourceGroup is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.DescribeResourceGroupRequest
func (_e *Proxy_Expecter) DescribeResourceGroup(ctx interface{}, req interface{}) *Proxy_DescribeResourceGroup_Call {
	return &Proxy_DescribeResourceGroup_Call{Call: _e.mock.On("DescribeResourceGroup", ctx, req)}
}

func (_c *Proxy_DescribeResourceGroup_Call) Run(run func(ctx context.Context, req *milvuspb.DescribeResourceGroupRequest)) *Proxy_DescribeResourceGroup_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DescribeResourceGroupRequest))
	})
	return _c
}

func (_c *Proxy_DescribeResourceGroup_Call) Return(_a0 *milvuspb.DescribeResourceGroupResponse, _a1 error) *Proxy_DescribeResourceGroup_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DropAlias provides a mock function with given fields: ctx, request
func (_m *Proxy) DropAlias(ctx context.Context, request *milvuspb.DropAliasRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropAliasRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropAliasRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DropAlias_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropAlias'
type Proxy_DropAlias_Call struct {
	*mock.Call
}

// DropAlias is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DropAliasRequest
func (_e *Proxy_Expecter) DropAlias(ctx interface{}, request interface{}) *Proxy_DropAlias_Call {
	return &Proxy_DropAlias_Call{Call: _e.mock.On("DropAlias", ctx, request)}
}

func (_c *Proxy_DropAlias_Call) Run(run func(ctx context.Context, request *milvuspb.DropAliasRequest)) *Proxy_DropAlias_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DropAliasRequest))
	})
	return _c
}

func (_c *Proxy_DropAlias_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DropAlias_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DropCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) DropCollection(ctx context.Context, request *milvuspb.DropCollectionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropCollectionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DropCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropCollection'
type Proxy_DropCollection_Call struct {
	*mock.Call
}

// DropCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DropCollectionRequest
func (_e *Proxy_Expecter) DropCollection(ctx interface{}, request interface{}) *Proxy_DropCollection_Call {
	return &Proxy_DropCollection_Call{Call: _e.mock.On("DropCollection", ctx, request)}
}

func (_c *Proxy_DropCollection_Call) Run(run func(ctx context.Context, request *milvuspb.DropCollectionRequest)) *Proxy_DropCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DropCollectionRequest))
	})
	return _c
}

func (_c *Proxy_DropCollection_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DropCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DropIndex provides a mock function with given fields: ctx, request
func (_m *Proxy) DropIndex(ctx context.Context, request *milvuspb.DropIndexRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropIndexRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropIndexRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DropIndex_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropIndex'
type Proxy_DropIndex_Call struct {
	*mock.Call
}

// DropIndex is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DropIndexRequest
func (_e *Proxy_Expecter) DropIndex(ctx interface{}, request interface{}) *Proxy_DropIndex_Call {
	return &Proxy_DropIndex_Call{Call: _e.mock.On("DropIndex", ctx, request)}
}

func (_c *Proxy_DropIndex_Call) Run(run func(ctx context.Context, request *milvuspb.DropIndexRequest)) *Proxy_DropIndex_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DropIndexRequest))
	})
	return _c
}

func (_c *Proxy_DropIndex_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DropIndex_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DropPartition provides a mock function with given fields: ctx, request
func (_m *Proxy) DropPartition(ctx context.Context, request *milvuspb.DropPartitionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropPartitionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropPartitionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DropPartition_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropPartition'
type Proxy_DropPartition_Call struct {
	*mock.Call
}

// DropPartition is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DropPartitionRequest
func (_e *Proxy_Expecter) DropPartition(ctx interface{}, request interface{}) *Proxy_DropPartition_Call {
	return &Proxy_DropPartition_Call{Call: _e.mock.On("DropPartition", ctx, request)}
}

func (_c *Proxy_DropPartition_Call) Run(run func(ctx context.Context, request *milvuspb.DropPartitionRequest)) *Proxy_DropPartition_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DropPartitionRequest))
	})
	return _c
}

func (_c *Proxy_DropPartition_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DropPartition_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DropResourceGroup provides a mock function with given fields: ctx, req
func (_m *Proxy) DropResourceGroup(ctx context.Context, req *milvuspb.DropResourceGroupRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropResourceGroupRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropResourceGroupRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DropResourceGroup_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropResourceGroup'
type Proxy_DropResourceGroup_Call struct {
	*mock.Call
}

// DropResourceGroup is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.DropResourceGroupRequest
func (_e *Proxy_Expecter) DropResourceGroup(ctx interface{}, req interface{}) *Proxy_DropResourceGroup_Call {
	return &Proxy_DropResourceGroup_Call{Call: _e.mock.On("DropResourceGroup", ctx, req)}
}

func (_c *Proxy_DropResourceGroup_Call) Run(run func(ctx context.Context, req *milvuspb.DropResourceGroupRequest)) *Proxy_DropResourceGroup_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DropResourceGroupRequest))
	})
	return _c
}

func (_c *Proxy_DropResourceGroup_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DropResourceGroup_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// DropRole provides a mock function with given fields: ctx, req
func (_m *Proxy) DropRole(ctx context.Context, req *milvuspb.DropRoleRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DropRoleRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DropRoleRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_DropRole_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'DropRole'
type Proxy_DropRole_Call struct {
	*mock.Call
}

// DropRole is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.DropRoleRequest
func (_e *Proxy_Expecter) DropRole(ctx interface{}, req interface{}) *Proxy_DropRole_Call {
	return &Proxy_DropRole_Call{Call: _e.mock.On("DropRole", ctx, req)}
}

func (_c *Proxy_DropRole_Call) Run(run func(ctx context.Context, req *milvuspb.DropRoleRequest)) *Proxy_DropRole_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DropRoleRequest))
	})
	return _c
}

func (_c *Proxy_DropRole_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_DropRole_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Dummy provides a mock function with given fields: ctx, request
func (_m *Proxy) Dummy(ctx context.Context, request *milvuspb.DummyRequest) (*milvuspb.DummyResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.DummyResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.DummyRequest) *milvuspb.DummyResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.DummyResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.DummyRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Dummy_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Dummy'
type Proxy_Dummy_Call struct {
	*mock.Call
}

// Dummy is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.DummyRequest
func (_e *Proxy_Expecter) Dummy(ctx interface{}, request interface{}) *Proxy_Dummy_Call {
	return &Proxy_Dummy_Call{Call: _e.mock.On("Dummy", ctx, request)}
}

func (_c *Proxy_Dummy_Call) Run(run func(ctx context.Context, request *milvuspb.DummyRequest)) *Proxy_Dummy_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.DummyRequest))
	})
	return _c
}

func (_c *Proxy_Dummy_Call) Return(_a0 *milvuspb.DummyResponse, _a1 error) *Proxy_Dummy_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Flush provides a mock function with given fields: ctx, request
func (_m *Proxy) Flush(ctx context.Context, request *milvuspb.FlushRequest) (*milvuspb.FlushResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.FlushResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushRequest) *milvuspb.FlushResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.FlushResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.FlushRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Flush_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Flush'
type Proxy_Flush_Call struct {
	*mock.Call
}

// Flush is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.FlushRequest
func (_e *Proxy_Expecter) Flush(ctx interface{}, request interface{}) *Proxy_Flush_Call {
	return &Proxy_Flush_Call{Call: _e.mock.On("Flush", ctx, request)}
}

func (_c *Proxy_Flush_Call) Run(run func(ctx context.Context, request *milvuspb.FlushRequest)) *Proxy_Flush_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.FlushRequest))
	})
	return _c
}

func (_c *Proxy_Flush_Call) Return(_a0 *milvuspb.FlushResponse, _a1 error) *Proxy_Flush_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// FlushAll provides a mock function with given fields: ctx, request
func (_m *Proxy) FlushAll(ctx context.Context, request *milvuspb.FlushAllRequest) (*milvuspb.FlushAllResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.FlushAllResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.FlushAllRequest) *milvuspb.FlushAllResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.FlushAllResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.FlushAllRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_FlushAll_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'FlushAll'
type Proxy_FlushAll_Call struct {
	*mock.Call
}

// FlushAll is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.FlushAllRequest
func (_e *Proxy_Expecter) FlushAll(ctx interface{}, request interface{}) *Proxy_FlushAll_Call {
	return &Proxy_FlushAll_Call{Call: _e.mock.On("FlushAll", ctx, request)}
}

func (_c *Proxy_FlushAll_Call) Run(run func(ctx context.Context, request *milvuspb.FlushAllRequest)) *Proxy_FlushAll_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.FlushAllRequest))
	})
	return _c
}

func (_c *Proxy_FlushAll_Call) Return(_a0 *milvuspb.FlushAllResponse, _a1 error) *Proxy_FlushAll_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetAddress provides a mock function with given fields:
func (_m *Proxy) GetAddress() string {
	ret := _m.Called()

	var r0 string
	if rf, ok := ret.Get(0).(func() string); ok {
		r0 = rf()
	} else {
		r0 = ret.Get(0).(string)
	}

	return r0
}

// Proxy_GetAddress_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetAddress'
type Proxy_GetAddress_Call struct {
	*mock.Call
}

// GetAddress is a helper method to define mock.On call
func (_e *Proxy_Expecter) GetAddress() *Proxy_GetAddress_Call {
	return &Proxy_GetAddress_Call{Call: _e.mock.On("GetAddress")}
}

func (_c *Proxy_GetAddress_Call) Run(run func()) *Proxy_GetAddress_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Proxy_GetAddress_Call) Return(_a0 string) *Proxy_GetAddress_Call {
	_c.Call.Return(_a0)
	return _c
}

// GetCollectionStatistics provides a mock function with given fields: ctx, request
func (_m *Proxy) GetCollectionStatistics(ctx context.Context, request *milvuspb.GetCollectionStatisticsRequest) (*milvuspb.GetCollectionStatisticsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetCollectionStatisticsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) *milvuspb.GetCollectionStatisticsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetCollectionStatisticsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCollectionStatisticsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetCollectionStatistics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetCollectionStatistics'
type Proxy_GetCollectionStatistics_Call struct {
	*mock.Call
}

// GetCollectionStatistics is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetCollectionStatisticsRequest
func (_e *Proxy_Expecter) GetCollectionStatistics(ctx interface{}, request interface{}) *Proxy_GetCollectionStatistics_Call {
	return &Proxy_GetCollectionStatistics_Call{Call: _e.mock.On("GetCollectionStatistics", ctx, request)}
}

func (_c *Proxy_GetCollectionStatistics_Call) Run(run func(ctx context.Context, request *milvuspb.GetCollectionStatisticsRequest)) *Proxy_GetCollectionStatistics_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetCollectionStatisticsRequest))
	})
	return _c
}

func (_c *Proxy_GetCollectionStatistics_Call) Return(_a0 *milvuspb.GetCollectionStatisticsResponse, _a1 error) *Proxy_GetCollectionStatistics_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetCompactionState provides a mock function with given fields: ctx, req
func (_m *Proxy) GetCompactionState(ctx context.Context, req *milvuspb.GetCompactionStateRequest) (*milvuspb.GetCompactionStateResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.GetCompactionStateResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionStateRequest) *milvuspb.GetCompactionStateResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetCompactionStateResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCompactionStateRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetCompactionState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetCompactionState'
type Proxy_GetCompactionState_Call struct {
	*mock.Call
}

// GetCompactionState is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.GetCompactionStateRequest
func (_e *Proxy_Expecter) GetCompactionState(ctx interface{}, req interface{}) *Proxy_GetCompactionState_Call {
	return &Proxy_GetCompactionState_Call{Call: _e.mock.On("GetCompactionState", ctx, req)}
}

func (_c *Proxy_GetCompactionState_Call) Run(run func(ctx context.Context, req *milvuspb.GetCompactionStateRequest)) *Proxy_GetCompactionState_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetCompactionStateRequest))
	})
	return _c
}

func (_c *Proxy_GetCompactionState_Call) Return(_a0 *milvuspb.GetCompactionStateResponse, _a1 error) *Proxy_GetCompactionState_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetCompactionStateWithPlans provides a mock function with given fields: ctx, req
func (_m *Proxy) GetCompactionStateWithPlans(ctx context.Context, req *milvuspb.GetCompactionPlansRequest) (*milvuspb.GetCompactionPlansResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.GetCompactionPlansResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetCompactionPlansRequest) *milvuspb.GetCompactionPlansResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetCompactionPlansResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetCompactionPlansRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetCompactionStateWithPlans_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetCompactionStateWithPlans'
type Proxy_GetCompactionStateWithPlans_Call struct {
	*mock.Call
}

// GetCompactionStateWithPlans is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.GetCompactionPlansRequest
func (_e *Proxy_Expecter) GetCompactionStateWithPlans(ctx interface{}, req interface{}) *Proxy_GetCompactionStateWithPlans_Call {
	return &Proxy_GetCompactionStateWithPlans_Call{Call: _e.mock.On("GetCompactionStateWithPlans", ctx, req)}
}

func (_c *Proxy_GetCompactionStateWithPlans_Call) Run(run func(ctx context.Context, req *milvuspb.GetCompactionPlansRequest)) *Proxy_GetCompactionStateWithPlans_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetCompactionPlansRequest))
	})
	return _c
}

func (_c *Proxy_GetCompactionStateWithPlans_Call) Return(_a0 *milvuspb.GetCompactionPlansResponse, _a1 error) *Proxy_GetCompactionStateWithPlans_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetComponentStates provides a mock function with given fields: ctx
func (_m *Proxy) GetComponentStates(ctx context.Context) (*milvuspb.ComponentStates, error) {
	ret := _m.Called(ctx)

	var r0 *milvuspb.ComponentStates
	if rf, ok := ret.Get(0).(func(context.Context) *milvuspb.ComponentStates); ok {
		r0 = rf(ctx)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ComponentStates)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context) error); ok {
		r1 = rf(ctx)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetComponentStates_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetComponentStates'
type Proxy_GetComponentStates_Call struct {
	*mock.Call
}

// GetComponentStates is a helper method to define mock.On call
//  - ctx context.Context
func (_e *Proxy_Expecter) GetComponentStates(ctx interface{}) *Proxy_GetComponentStates_Call {
	return &Proxy_GetComponentStates_Call{Call: _e.mock.On("GetComponentStates", ctx)}
}

func (_c *Proxy_GetComponentStates_Call) Run(run func(ctx context.Context)) *Proxy_GetComponentStates_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context))
	})
	return _c
}

func (_c *Proxy_GetComponentStates_Call) Return(_a0 *milvuspb.ComponentStates, _a1 error) *Proxy_GetComponentStates_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetDdChannel provides a mock function with given fields: ctx, request
func (_m *Proxy) GetDdChannel(ctx context.Context, request *internalpb.GetDdChannelRequest) (*milvuspb.StringResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.StringResponse
	if rf, ok := ret.Get(0).(func(context.Context, *internalpb.GetDdChannelRequest) *milvuspb.StringResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.StringResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *internalpb.GetDdChannelRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetDdChannel_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetDdChannel'
type Proxy_GetDdChannel_Call struct {
	*mock.Call
}

// GetDdChannel is a helper method to define mock.On call
//  - ctx context.Context
//  - request *internalpb.GetDdChannelRequest
func (_e *Proxy_Expecter) GetDdChannel(ctx interface{}, request interface{}) *Proxy_GetDdChannel_Call {
	return &Proxy_GetDdChannel_Call{Call: _e.mock.On("GetDdChannel", ctx, request)}
}

func (_c *Proxy_GetDdChannel_Call) Run(run func(ctx context.Context, request *internalpb.GetDdChannelRequest)) *Proxy_GetDdChannel_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*internalpb.GetDdChannelRequest))
	})
	return _c
}

func (_c *Proxy_GetDdChannel_Call) Return(_a0 *milvuspb.StringResponse, _a1 error) *Proxy_GetDdChannel_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetFlushAllState provides a mock function with given fields: ctx, req
func (_m *Proxy) GetFlushAllState(ctx context.Context, req *milvuspb.GetFlushAllStateRequest) (*milvuspb.GetFlushAllStateResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.GetFlushAllStateResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushAllStateRequest) *milvuspb.GetFlushAllStateResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetFlushAllStateResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetFlushAllStateRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetFlushAllState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetFlushAllState'
type Proxy_GetFlushAllState_Call struct {
	*mock.Call
}

// GetFlushAllState is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.GetFlushAllStateRequest
func (_e *Proxy_Expecter) GetFlushAllState(ctx interface{}, req interface{}) *Proxy_GetFlushAllState_Call {
	return &Proxy_GetFlushAllState_Call{Call: _e.mock.On("GetFlushAllState", ctx, req)}
}

func (_c *Proxy_GetFlushAllState_Call) Run(run func(ctx context.Context, req *milvuspb.GetFlushAllStateRequest)) *Proxy_GetFlushAllState_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetFlushAllStateRequest))
	})
	return _c
}

func (_c *Proxy_GetFlushAllState_Call) Return(_a0 *milvuspb.GetFlushAllStateResponse, _a1 error) *Proxy_GetFlushAllState_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetFlushState provides a mock function with given fields: ctx, req
func (_m *Proxy) GetFlushState(ctx context.Context, req *milvuspb.GetFlushStateRequest) (*milvuspb.GetFlushStateResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.GetFlushStateResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetFlushStateRequest) *milvuspb.GetFlushStateResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetFlushStateResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetFlushStateRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetFlushState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetFlushState'
type Proxy_GetFlushState_Call struct {
	*mock.Call
}

// GetFlushState is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.GetFlushStateRequest
func (_e *Proxy_Expecter) GetFlushState(ctx interface{}, req interface{}) *Proxy_GetFlushState_Call {
	return &Proxy_GetFlushState_Call{Call: _e.mock.On("GetFlushState", ctx, req)}
}

func (_c *Proxy_GetFlushState_Call) Run(run func(ctx context.Context, req *milvuspb.GetFlushStateRequest)) *Proxy_GetFlushState_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetFlushStateRequest))
	})
	return _c
}

func (_c *Proxy_GetFlushState_Call) Return(_a0 *milvuspb.GetFlushStateResponse, _a1 error) *Proxy_GetFlushState_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetImportState provides a mock function with given fields: ctx, req
func (_m *Proxy) GetImportState(ctx context.Context, req *milvuspb.GetImportStateRequest) (*milvuspb.GetImportStateResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.GetImportStateResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetImportStateRequest) *milvuspb.GetImportStateResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetImportStateResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetImportStateRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetImportState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetImportState'
type Proxy_GetImportState_Call struct {
	*mock.Call
}

// GetImportState is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.GetImportStateRequest
func (_e *Proxy_Expecter) GetImportState(ctx interface{}, req interface{}) *Proxy_GetImportState_Call {
	return &Proxy_GetImportState_Call{Call: _e.mock.On("GetImportState", ctx, req)}
}

func (_c *Proxy_GetImportState_Call) Run(run func(ctx context.Context, req *milvuspb.GetImportStateRequest)) *Proxy_GetImportState_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetImportStateRequest))
	})
	return _c
}

func (_c *Proxy_GetImportState_Call) Return(_a0 *milvuspb.GetImportStateResponse, _a1 error) *Proxy_GetImportState_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetIndexBuildProgress provides a mock function with given fields: ctx, request
func (_m *Proxy) GetIndexBuildProgress(ctx context.Context, request *milvuspb.GetIndexBuildProgressRequest) (*milvuspb.GetIndexBuildProgressResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetIndexBuildProgressResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) *milvuspb.GetIndexBuildProgressResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetIndexBuildProgressResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexBuildProgressRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetIndexBuildProgress_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetIndexBuildProgress'
type Proxy_GetIndexBuildProgress_Call struct {
	*mock.Call
}

// GetIndexBuildProgress is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetIndexBuildProgressRequest
func (_e *Proxy_Expecter) GetIndexBuildProgress(ctx interface{}, request interface{}) *Proxy_GetIndexBuildProgress_Call {
	return &Proxy_GetIndexBuildProgress_Call{Call: _e.mock.On("GetIndexBuildProgress", ctx, request)}
}

func (_c *Proxy_GetIndexBuildProgress_Call) Run(run func(ctx context.Context, request *milvuspb.GetIndexBuildProgressRequest)) *Proxy_GetIndexBuildProgress_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetIndexBuildProgressRequest))
	})
	return _c
}

func (_c *Proxy_GetIndexBuildProgress_Call) Return(_a0 *milvuspb.GetIndexBuildProgressResponse, _a1 error) *Proxy_GetIndexBuildProgress_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetIndexState provides a mock function with given fields: ctx, request
func (_m *Proxy) GetIndexState(ctx context.Context, request *milvuspb.GetIndexStateRequest) (*milvuspb.GetIndexStateResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetIndexStateResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStateRequest) *milvuspb.GetIndexStateResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetIndexStateResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexStateRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetIndexState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetIndexState'
type Proxy_GetIndexState_Call struct {
	*mock.Call
}

// GetIndexState is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetIndexStateRequest
func (_e *Proxy_Expecter) GetIndexState(ctx interface{}, request interface{}) *Proxy_GetIndexState_Call {
	return &Proxy_GetIndexState_Call{Call: _e.mock.On("GetIndexState", ctx, request)}
}

func (_c *Proxy_GetIndexState_Call) Run(run func(ctx context.Context, request *milvuspb.GetIndexStateRequest)) *Proxy_GetIndexState_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetIndexStateRequest))
	})
	return _c
}

func (_c *Proxy_GetIndexState_Call) Return(_a0 *milvuspb.GetIndexStateResponse, _a1 error) *Proxy_GetIndexState_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetIndexStatistics provides a mock function with given fields: ctx, request
func (_m *Proxy) GetIndexStatistics(ctx context.Context, request *milvuspb.GetIndexStatisticsRequest) (*milvuspb.GetIndexStatisticsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetIndexStatisticsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetIndexStatisticsRequest) *milvuspb.GetIndexStatisticsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetIndexStatisticsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetIndexStatisticsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetIndexStatistics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetIndexStatistics'
type Proxy_GetIndexStatistics_Call struct {
	*mock.Call
}

// GetIndexStatistics is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetIndexStatisticsRequest
func (_e *Proxy_Expecter) GetIndexStatistics(ctx interface{}, request interface{}) *Proxy_GetIndexStatistics_Call {
	return &Proxy_GetIndexStatistics_Call{Call: _e.mock.On("GetIndexStatistics", ctx, request)}
}

func (_c *Proxy_GetIndexStatistics_Call) Run(run func(ctx context.Context, request *milvuspb.GetIndexStatisticsRequest)) *Proxy_GetIndexStatistics_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetIndexStatisticsRequest))
	})
	return _c
}

func (_c *Proxy_GetIndexStatistics_Call) Return(_a0 *milvuspb.GetIndexStatisticsResponse, _a1 error) *Proxy_GetIndexStatistics_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetLoadState provides a mock function with given fields: ctx, request
func (_m *Proxy) GetLoadState(ctx context.Context, request *milvuspb.GetLoadStateRequest) (*milvuspb.GetLoadStateResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetLoadStateResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadStateRequest) *milvuspb.GetLoadStateResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetLoadStateResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetLoadStateRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetLoadState_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetLoadState'
type Proxy_GetLoadState_Call struct {
	*mock.Call
}

// GetLoadState is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetLoadStateRequest
func (_e *Proxy_Expecter) GetLoadState(ctx interface{}, request interface{}) *Proxy_GetLoadState_Call {
	return &Proxy_GetLoadState_Call{Call: _e.mock.On("GetLoadState", ctx, request)}
}

func (_c *Proxy_GetLoadState_Call) Run(run func(ctx context.Context, request *milvuspb.GetLoadStateRequest)) *Proxy_GetLoadState_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetLoadStateRequest))
	})
	return _c
}

func (_c *Proxy_GetLoadState_Call) Return(_a0 *milvuspb.GetLoadStateResponse, _a1 error) *Proxy_GetLoadState_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetLoadingProgress provides a mock function with given fields: ctx, request
func (_m *Proxy) GetLoadingProgress(ctx context.Context, request *milvuspb.GetLoadingProgressRequest) (*milvuspb.GetLoadingProgressResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetLoadingProgressResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetLoadingProgressRequest) *milvuspb.GetLoadingProgressResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetLoadingProgressResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetLoadingProgressRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetLoadingProgress_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetLoadingProgress'
type Proxy_GetLoadingProgress_Call struct {
	*mock.Call
}

// GetLoadingProgress is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetLoadingProgressRequest
func (_e *Proxy_Expecter) GetLoadingProgress(ctx interface{}, request interface{}) *Proxy_GetLoadingProgress_Call {
	return &Proxy_GetLoadingProgress_Call{Call: _e.mock.On("GetLoadingProgress", ctx, request)}
}

func (_c *Proxy_GetLoadingProgress_Call) Run(run func(ctx context.Context, request *milvuspb.GetLoadingProgressRequest)) *Proxy_GetLoadingProgress_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetLoadingProgressRequest))
	})
	return _c
}

func (_c *Proxy_GetLoadingProgress_Call) Return(_a0 *milvuspb.GetLoadingProgressResponse, _a1 error) *Proxy_GetLoadingProgress_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetMetrics provides a mock function with given fields: ctx, request
func (_m *Proxy) GetMetrics(ctx context.Context, request *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetMetricsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) *milvuspb.GetMetricsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetMetricsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetMetricsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetMetrics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetMetrics'
type Proxy_GetMetrics_Call struct {
	*mock.Call
}

// GetMetrics is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetMetricsRequest
func (_e *Proxy_Expecter) GetMetrics(ctx interface{}, request interface{}) *Proxy_GetMetrics_Call {
	return &Proxy_GetMetrics_Call{Call: _e.mock.On("GetMetrics", ctx, request)}
}

func (_c *Proxy_GetMetrics_Call) Run(run func(ctx context.Context, request *milvuspb.GetMetricsRequest)) *Proxy_GetMetrics_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetMetricsRequest))
	})
	return _c
}

func (_c *Proxy_GetMetrics_Call) Return(_a0 *milvuspb.GetMetricsResponse, _a1 error) *Proxy_GetMetrics_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetPartitionStatistics provides a mock function with given fields: ctx, request
func (_m *Proxy) GetPartitionStatistics(ctx context.Context, request *milvuspb.GetPartitionStatisticsRequest) (*milvuspb.GetPartitionStatisticsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetPartitionStatisticsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) *milvuspb.GetPartitionStatisticsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetPartitionStatisticsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetPartitionStatisticsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetPartitionStatistics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetPartitionStatistics'
type Proxy_GetPartitionStatistics_Call struct {
	*mock.Call
}

// GetPartitionStatistics is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetPartitionStatisticsRequest
func (_e *Proxy_Expecter) GetPartitionStatistics(ctx interface{}, request interface{}) *Proxy_GetPartitionStatistics_Call {
	return &Proxy_GetPartitionStatistics_Call{Call: _e.mock.On("GetPartitionStatistics", ctx, request)}
}

func (_c *Proxy_GetPartitionStatistics_Call) Run(run func(ctx context.Context, request *milvuspb.GetPartitionStatisticsRequest)) *Proxy_GetPartitionStatistics_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetPartitionStatisticsRequest))
	})
	return _c
}

func (_c *Proxy_GetPartitionStatistics_Call) Return(_a0 *milvuspb.GetPartitionStatisticsResponse, _a1 error) *Proxy_GetPartitionStatistics_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetPersistentSegmentInfo provides a mock function with given fields: ctx, request
func (_m *Proxy) GetPersistentSegmentInfo(ctx context.Context, request *milvuspb.GetPersistentSegmentInfoRequest) (*milvuspb.GetPersistentSegmentInfoResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetPersistentSegmentInfoResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) *milvuspb.GetPersistentSegmentInfoResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetPersistentSegmentInfoResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetPersistentSegmentInfoRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetPersistentSegmentInfo_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetPersistentSegmentInfo'
type Proxy_GetPersistentSegmentInfo_Call struct {
	*mock.Call
}

// GetPersistentSegmentInfo is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetPersistentSegmentInfoRequest
func (_e *Proxy_Expecter) GetPersistentSegmentInfo(ctx interface{}, request interface{}) *Proxy_GetPersistentSegmentInfo_Call {
	return &Proxy_GetPersistentSegmentInfo_Call{Call: _e.mock.On("GetPersistentSegmentInfo", ctx, request)}
}

func (_c *Proxy_GetPersistentSegmentInfo_Call) Run(run func(ctx context.Context, request *milvuspb.GetPersistentSegmentInfoRequest)) *Proxy_GetPersistentSegmentInfo_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetPersistentSegmentInfoRequest))
	})
	return _c
}

func (_c *Proxy_GetPersistentSegmentInfo_Call) Return(_a0 *milvuspb.GetPersistentSegmentInfoResponse, _a1 error) *Proxy_GetPersistentSegmentInfo_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetProxyMetrics provides a mock function with given fields: ctx, request
func (_m *Proxy) GetProxyMetrics(ctx context.Context, request *milvuspb.GetMetricsRequest) (*milvuspb.GetMetricsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetMetricsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetMetricsRequest) *milvuspb.GetMetricsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetMetricsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetMetricsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetProxyMetrics_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetProxyMetrics'
type Proxy_GetProxyMetrics_Call struct {
	*mock.Call
}

// GetProxyMetrics is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetMetricsRequest
func (_e *Proxy_Expecter) GetProxyMetrics(ctx interface{}, request interface{}) *Proxy_GetProxyMetrics_Call {
	return &Proxy_GetProxyMetrics_Call{Call: _e.mock.On("GetProxyMetrics", ctx, request)}
}

func (_c *Proxy_GetProxyMetrics_Call) Run(run func(ctx context.Context, request *milvuspb.GetMetricsRequest)) *Proxy_GetProxyMetrics_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetMetricsRequest))
	})
	return _c
}

func (_c *Proxy_GetProxyMetrics_Call) Return(_a0 *milvuspb.GetMetricsResponse, _a1 error) *Proxy_GetProxyMetrics_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetQuerySegmentInfo provides a mock function with given fields: ctx, request
func (_m *Proxy) GetQuerySegmentInfo(ctx context.Context, request *milvuspb.GetQuerySegmentInfoRequest) (*milvuspb.GetQuerySegmentInfoResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.GetQuerySegmentInfoResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) *milvuspb.GetQuerySegmentInfoResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetQuerySegmentInfoResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetQuerySegmentInfoRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetQuerySegmentInfo_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetQuerySegmentInfo'
type Proxy_GetQuerySegmentInfo_Call struct {
	*mock.Call
}

// GetQuerySegmentInfo is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.GetQuerySegmentInfoRequest
func (_e *Proxy_Expecter) GetQuerySegmentInfo(ctx interface{}, request interface{}) *Proxy_GetQuerySegmentInfo_Call {
	return &Proxy_GetQuerySegmentInfo_Call{Call: _e.mock.On("GetQuerySegmentInfo", ctx, request)}
}

func (_c *Proxy_GetQuerySegmentInfo_Call) Run(run func(ctx context.Context, request *milvuspb.GetQuerySegmentInfoRequest)) *Proxy_GetQuerySegmentInfo_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetQuerySegmentInfoRequest))
	})
	return _c
}

func (_c *Proxy_GetQuerySegmentInfo_Call) Return(_a0 *milvuspb.GetQuerySegmentInfoResponse, _a1 error) *Proxy_GetQuerySegmentInfo_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetRateLimiter provides a mock function with given fields:
func (_m *Proxy) GetRateLimiter() (types.Limiter, error) {
	ret := _m.Called()

	var r0 types.Limiter
	if rf, ok := ret.Get(0).(func() types.Limiter); ok {
		r0 = rf()
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(types.Limiter)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func() error); ok {
		r1 = rf()
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetRateLimiter_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetRateLimiter'
type Proxy_GetRateLimiter_Call struct {
	*mock.Call
}

// GetRateLimiter is a helper method to define mock.On call
func (_e *Proxy_Expecter) GetRateLimiter() *Proxy_GetRateLimiter_Call {
	return &Proxy_GetRateLimiter_Call{Call: _e.mock.On("GetRateLimiter")}
}

func (_c *Proxy_GetRateLimiter_Call) Run(run func()) *Proxy_GetRateLimiter_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Proxy_GetRateLimiter_Call) Return(_a0 types.Limiter, _a1 error) *Proxy_GetRateLimiter_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetReplicas provides a mock function with given fields: ctx, req
func (_m *Proxy) GetReplicas(ctx context.Context, req *milvuspb.GetReplicasRequest) (*milvuspb.GetReplicasResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.GetReplicasResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.GetReplicasRequest) *milvuspb.GetReplicasResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.GetReplicasResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.GetReplicasRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetReplicas_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetReplicas'
type Proxy_GetReplicas_Call struct {
	*mock.Call
}

// GetReplicas is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.GetReplicasRequest
func (_e *Proxy_Expecter) GetReplicas(ctx interface{}, req interface{}) *Proxy_GetReplicas_Call {
	return &Proxy_GetReplicas_Call{Call: _e.mock.On("GetReplicas", ctx, req)}
}

func (_c *Proxy_GetReplicas_Call) Run(run func(ctx context.Context, req *milvuspb.GetReplicasRequest)) *Proxy_GetReplicas_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.GetReplicasRequest))
	})
	return _c
}

func (_c *Proxy_GetReplicas_Call) Return(_a0 *milvuspb.GetReplicasResponse, _a1 error) *Proxy_GetReplicas_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// GetStatisticsChannel provides a mock function with given fields: ctx
func (_m *Proxy) GetStatisticsChannel(ctx context.Context) (*milvuspb.StringResponse, error) {
	ret := _m.Called(ctx)

	var r0 *milvuspb.StringResponse
	if rf, ok := ret.Get(0).(func(context.Context) *milvuspb.StringResponse); ok {
		r0 = rf(ctx)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.StringResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context) error); ok {
		r1 = rf(ctx)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_GetStatisticsChannel_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'GetStatisticsChannel'
type Proxy_GetStatisticsChannel_Call struct {
	*mock.Call
}

// GetStatisticsChannel is a helper method to define mock.On call
//  - ctx context.Context
func (_e *Proxy_Expecter) GetStatisticsChannel(ctx interface{}) *Proxy_GetStatisticsChannel_Call {
	return &Proxy_GetStatisticsChannel_Call{Call: _e.mock.On("GetStatisticsChannel", ctx)}
}

func (_c *Proxy_GetStatisticsChannel_Call) Run(run func(ctx context.Context)) *Proxy_GetStatisticsChannel_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context))
	})
	return _c
}

func (_c *Proxy_GetStatisticsChannel_Call) Return(_a0 *milvuspb.StringResponse, _a1 error) *Proxy_GetStatisticsChannel_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// HasCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) HasCollection(ctx context.Context, request *milvuspb.HasCollectionRequest) (*milvuspb.BoolResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.BoolResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasCollectionRequest) *milvuspb.BoolResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.BoolResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.HasCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_HasCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'HasCollection'
type Proxy_HasCollection_Call struct {
	*mock.Call
}

// HasCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.HasCollectionRequest
func (_e *Proxy_Expecter) HasCollection(ctx interface{}, request interface{}) *Proxy_HasCollection_Call {
	return &Proxy_HasCollection_Call{Call: _e.mock.On("HasCollection", ctx, request)}
}

func (_c *Proxy_HasCollection_Call) Run(run func(ctx context.Context, request *milvuspb.HasCollectionRequest)) *Proxy_HasCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.HasCollectionRequest))
	})
	return _c
}

func (_c *Proxy_HasCollection_Call) Return(_a0 *milvuspb.BoolResponse, _a1 error) *Proxy_HasCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// HasPartition provides a mock function with given fields: ctx, request
func (_m *Proxy) HasPartition(ctx context.Context, request *milvuspb.HasPartitionRequest) (*milvuspb.BoolResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.BoolResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.HasPartitionRequest) *milvuspb.BoolResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.BoolResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.HasPartitionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_HasPartition_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'HasPartition'
type Proxy_HasPartition_Call struct {
	*mock.Call
}

// HasPartition is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.HasPartitionRequest
func (_e *Proxy_Expecter) HasPartition(ctx interface{}, request interface{}) *Proxy_HasPartition_Call {
	return &Proxy_HasPartition_Call{Call: _e.mock.On("HasPartition", ctx, request)}
}

func (_c *Proxy_HasPartition_Call) Run(run func(ctx context.Context, request *milvuspb.HasPartitionRequest)) *Proxy_HasPartition_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.HasPartitionRequest))
	})
	return _c
}

func (_c *Proxy_HasPartition_Call) Return(_a0 *milvuspb.BoolResponse, _a1 error) *Proxy_HasPartition_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Import provides a mock function with given fields: ctx, req
func (_m *Proxy) Import(ctx context.Context, req *milvuspb.ImportRequest) (*milvuspb.ImportResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.ImportResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ImportRequest) *milvuspb.ImportResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ImportResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ImportRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Import_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Import'
type Proxy_Import_Call struct {
	*mock.Call
}

// Import is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.ImportRequest
func (_e *Proxy_Expecter) Import(ctx interface{}, req interface{}) *Proxy_Import_Call {
	return &Proxy_Import_Call{Call: _e.mock.On("Import", ctx, req)}
}

func (_c *Proxy_Import_Call) Run(run func(ctx context.Context, req *milvuspb.ImportRequest)) *Proxy_Import_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ImportRequest))
	})
	return _c
}

func (_c *Proxy_Import_Call) Return(_a0 *milvuspb.ImportResponse, _a1 error) *Proxy_Import_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Init provides a mock function with given fields:
func (_m *Proxy) Init() 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
}

// Proxy_Init_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Init'
type Proxy_Init_Call struct {
	*mock.Call
}

// Init is a helper method to define mock.On call
func (_e *Proxy_Expecter) Init() *Proxy_Init_Call {
	return &Proxy_Init_Call{Call: _e.mock.On("Init")}
}

func (_c *Proxy_Init_Call) Run(run func()) *Proxy_Init_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Proxy_Init_Call) Return(_a0 error) *Proxy_Init_Call {
	_c.Call.Return(_a0)
	return _c
}

// Insert provides a mock function with given fields: ctx, request
func (_m *Proxy) Insert(ctx context.Context, request *milvuspb.InsertRequest) (*milvuspb.MutationResult, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.MutationResult
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.InsertRequest) *milvuspb.MutationResult); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.MutationResult)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.InsertRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Insert_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Insert'
type Proxy_Insert_Call struct {
	*mock.Call
}

// Insert is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.InsertRequest
func (_e *Proxy_Expecter) Insert(ctx interface{}, request interface{}) *Proxy_Insert_Call {
	return &Proxy_Insert_Call{Call: _e.mock.On("Insert", ctx, request)}
}

func (_c *Proxy_Insert_Call) Run(run func(ctx context.Context, request *milvuspb.InsertRequest)) *Proxy_Insert_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.InsertRequest))
	})
	return _c
}

func (_c *Proxy_Insert_Call) Return(_a0 *milvuspb.MutationResult, _a1 error) *Proxy_Insert_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// InvalidateCollectionMetaCache provides a mock function with given fields: ctx, request
func (_m *Proxy) InvalidateCollectionMetaCache(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *proxypb.InvalidateCollMetaCacheRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_InvalidateCollectionMetaCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InvalidateCollectionMetaCache'
type Proxy_InvalidateCollectionMetaCache_Call struct {
	*mock.Call
}

// InvalidateCollectionMetaCache is a helper method to define mock.On call
//  - ctx context.Context
//  - request *proxypb.InvalidateCollMetaCacheRequest
func (_e *Proxy_Expecter) InvalidateCollectionMetaCache(ctx interface{}, request interface{}) *Proxy_InvalidateCollectionMetaCache_Call {
	return &Proxy_InvalidateCollectionMetaCache_Call{Call: _e.mock.On("InvalidateCollectionMetaCache", ctx, request)}
}

func (_c *Proxy_InvalidateCollectionMetaCache_Call) Run(run func(ctx context.Context, request *proxypb.InvalidateCollMetaCacheRequest)) *Proxy_InvalidateCollectionMetaCache_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*proxypb.InvalidateCollMetaCacheRequest))
	})
	return _c
}

func (_c *Proxy_InvalidateCollectionMetaCache_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_InvalidateCollectionMetaCache_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// InvalidateCredentialCache provides a mock function with given fields: ctx, request
func (_m *Proxy) InvalidateCredentialCache(ctx context.Context, request *proxypb.InvalidateCredCacheRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.InvalidateCredCacheRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *proxypb.InvalidateCredCacheRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_InvalidateCredentialCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'InvalidateCredentialCache'
type Proxy_InvalidateCredentialCache_Call struct {
	*mock.Call
}

// InvalidateCredentialCache is a helper method to define mock.On call
//  - ctx context.Context
//  - request *proxypb.InvalidateCredCacheRequest
func (_e *Proxy_Expecter) InvalidateCredentialCache(ctx interface{}, request interface{}) *Proxy_InvalidateCredentialCache_Call {
	return &Proxy_InvalidateCredentialCache_Call{Call: _e.mock.On("InvalidateCredentialCache", ctx, request)}
}

func (_c *Proxy_InvalidateCredentialCache_Call) Run(run func(ctx context.Context, request *proxypb.InvalidateCredCacheRequest)) *Proxy_InvalidateCredentialCache_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*proxypb.InvalidateCredCacheRequest))
	})
	return _c
}

func (_c *Proxy_InvalidateCredentialCache_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_InvalidateCredentialCache_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ListClientInfos provides a mock function with given fields: ctx, req
func (_m *Proxy) ListClientInfos(ctx context.Context, req *proxypb.ListClientInfosRequest) (*proxypb.ListClientInfosResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *proxypb.ListClientInfosResponse
	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.ListClientInfosRequest) *proxypb.ListClientInfosResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*proxypb.ListClientInfosResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *proxypb.ListClientInfosRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ListClientInfos_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListClientInfos'
type Proxy_ListClientInfos_Call struct {
	*mock.Call
}

// ListClientInfos is a helper method to define mock.On call
//  - ctx context.Context
//  - req *proxypb.ListClientInfosRequest
func (_e *Proxy_Expecter) ListClientInfos(ctx interface{}, req interface{}) *Proxy_ListClientInfos_Call {
	return &Proxy_ListClientInfos_Call{Call: _e.mock.On("ListClientInfos", ctx, req)}
}

func (_c *Proxy_ListClientInfos_Call) Run(run func(ctx context.Context, req *proxypb.ListClientInfosRequest)) *Proxy_ListClientInfos_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*proxypb.ListClientInfosRequest))
	})
	return _c
}

func (_c *Proxy_ListClientInfos_Call) Return(_a0 *proxypb.ListClientInfosResponse, _a1 error) *Proxy_ListClientInfos_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ListCredUsers provides a mock function with given fields: ctx, req
func (_m *Proxy) ListCredUsers(ctx context.Context, req *milvuspb.ListCredUsersRequest) (*milvuspb.ListCredUsersResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.ListCredUsersResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListCredUsersRequest) *milvuspb.ListCredUsersResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ListCredUsersResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListCredUsersRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ListCredUsers_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListCredUsers'
type Proxy_ListCredUsers_Call struct {
	*mock.Call
}

// ListCredUsers is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.ListCredUsersRequest
func (_e *Proxy_Expecter) ListCredUsers(ctx interface{}, req interface{}) *Proxy_ListCredUsers_Call {
	return &Proxy_ListCredUsers_Call{Call: _e.mock.On("ListCredUsers", ctx, req)}
}

func (_c *Proxy_ListCredUsers_Call) Run(run func(ctx context.Context, req *milvuspb.ListCredUsersRequest)) *Proxy_ListCredUsers_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ListCredUsersRequest))
	})
	return _c
}

func (_c *Proxy_ListCredUsers_Call) Return(_a0 *milvuspb.ListCredUsersResponse, _a1 error) *Proxy_ListCredUsers_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ListImportTasks provides a mock function with given fields: ctx, req
func (_m *Proxy) ListImportTasks(ctx context.Context, req *milvuspb.ListImportTasksRequest) (*milvuspb.ListImportTasksResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.ListImportTasksResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListImportTasksRequest) *milvuspb.ListImportTasksResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ListImportTasksResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListImportTasksRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ListImportTasks_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListImportTasks'
type Proxy_ListImportTasks_Call struct {
	*mock.Call
}

// ListImportTasks is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.ListImportTasksRequest
func (_e *Proxy_Expecter) ListImportTasks(ctx interface{}, req interface{}) *Proxy_ListImportTasks_Call {
	return &Proxy_ListImportTasks_Call{Call: _e.mock.On("ListImportTasks", ctx, req)}
}

func (_c *Proxy_ListImportTasks_Call) Run(run func(ctx context.Context, req *milvuspb.ListImportTasksRequest)) *Proxy_ListImportTasks_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ListImportTasksRequest))
	})
	return _c
}

func (_c *Proxy_ListImportTasks_Call) Return(_a0 *milvuspb.ListImportTasksResponse, _a1 error) *Proxy_ListImportTasks_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ListResourceGroups provides a mock function with given fields: ctx, req
func (_m *Proxy) ListResourceGroups(ctx context.Context, req *milvuspb.ListResourceGroupsRequest) (*milvuspb.ListResourceGroupsResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.ListResourceGroupsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ListResourceGroupsRequest) *milvuspb.ListResourceGroupsResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ListResourceGroupsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ListResourceGroupsRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ListResourceGroups_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ListResourceGroups'
type Proxy_ListResourceGroups_Call struct {
	*mock.Call
}

// ListResourceGroups is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.ListResourceGroupsRequest
func (_e *Proxy_Expecter) ListResourceGroups(ctx interface{}, req interface{}) *Proxy_ListResourceGroups_Call {
	return &Proxy_ListResourceGroups_Call{Call: _e.mock.On("ListResourceGroups", ctx, req)}
}

func (_c *Proxy_ListResourceGroups_Call) Run(run func(ctx context.Context, req *milvuspb.ListResourceGroupsRequest)) *Proxy_ListResourceGroups_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ListResourceGroupsRequest))
	})
	return _c
}

func (_c *Proxy_ListResourceGroups_Call) Return(_a0 *milvuspb.ListResourceGroupsResponse, _a1 error) *Proxy_ListResourceGroups_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// LoadBalance provides a mock function with given fields: ctx, request
func (_m *Proxy) LoadBalance(ctx context.Context, request *milvuspb.LoadBalanceRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadBalanceRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadBalanceRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_LoadBalance_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LoadBalance'
type Proxy_LoadBalance_Call struct {
	*mock.Call
}

// LoadBalance is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.LoadBalanceRequest
func (_e *Proxy_Expecter) LoadBalance(ctx interface{}, request interface{}) *Proxy_LoadBalance_Call {
	return &Proxy_LoadBalance_Call{Call: _e.mock.On("LoadBalance", ctx, request)}
}

func (_c *Proxy_LoadBalance_Call) Run(run func(ctx context.Context, request *milvuspb.LoadBalanceRequest)) *Proxy_LoadBalance_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.LoadBalanceRequest))
	})
	return _c
}

func (_c *Proxy_LoadBalance_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_LoadBalance_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// LoadCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) LoadCollection(ctx context.Context, request *milvuspb.LoadCollectionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadCollectionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_LoadCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LoadCollection'
type Proxy_LoadCollection_Call struct {
	*mock.Call
}

// LoadCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.LoadCollectionRequest
func (_e *Proxy_Expecter) LoadCollection(ctx interface{}, request interface{}) *Proxy_LoadCollection_Call {
	return &Proxy_LoadCollection_Call{Call: _e.mock.On("LoadCollection", ctx, request)}
}

func (_c *Proxy_LoadCollection_Call) Run(run func(ctx context.Context, request *milvuspb.LoadCollectionRequest)) *Proxy_LoadCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.LoadCollectionRequest))
	})
	return _c
}

func (_c *Proxy_LoadCollection_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_LoadCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// LoadPartitions provides a mock function with given fields: ctx, request
func (_m *Proxy) LoadPartitions(ctx context.Context, request *milvuspb.LoadPartitionsRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.LoadPartitionsRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.LoadPartitionsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_LoadPartitions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'LoadPartitions'
type Proxy_LoadPartitions_Call struct {
	*mock.Call
}

// LoadPartitions is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.LoadPartitionsRequest
func (_e *Proxy_Expecter) LoadPartitions(ctx interface{}, request interface{}) *Proxy_LoadPartitions_Call {
	return &Proxy_LoadPartitions_Call{Call: _e.mock.On("LoadPartitions", ctx, request)}
}

func (_c *Proxy_LoadPartitions_Call) Run(run func(ctx context.Context, request *milvuspb.LoadPartitionsRequest)) *Proxy_LoadPartitions_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.LoadPartitionsRequest))
	})
	return _c
}

func (_c *Proxy_LoadPartitions_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_LoadPartitions_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ManualCompaction provides a mock function with given fields: ctx, req
func (_m *Proxy) ManualCompaction(ctx context.Context, req *milvuspb.ManualCompactionRequest) (*milvuspb.ManualCompactionResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.ManualCompactionResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ManualCompactionRequest) *milvuspb.ManualCompactionResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ManualCompactionResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ManualCompactionRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ManualCompaction_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ManualCompaction'
type Proxy_ManualCompaction_Call struct {
	*mock.Call
}

// ManualCompaction is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.ManualCompactionRequest
func (_e *Proxy_Expecter) ManualCompaction(ctx interface{}, req interface{}) *Proxy_ManualCompaction_Call {
	return &Proxy_ManualCompaction_Call{Call: _e.mock.On("ManualCompaction", ctx, req)}
}

func (_c *Proxy_ManualCompaction_Call) Run(run func(ctx context.Context, req *milvuspb.ManualCompactionRequest)) *Proxy_ManualCompaction_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ManualCompactionRequest))
	})
	return _c
}

func (_c *Proxy_ManualCompaction_Call) Return(_a0 *milvuspb.ManualCompactionResponse, _a1 error) *Proxy_ManualCompaction_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// OperatePrivilege provides a mock function with given fields: ctx, req
func (_m *Proxy) OperatePrivilege(ctx context.Context, req *milvuspb.OperatePrivilegeRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperatePrivilegeRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.OperatePrivilegeRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_OperatePrivilege_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'OperatePrivilege'
type Proxy_OperatePrivilege_Call struct {
	*mock.Call
}

// OperatePrivilege is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.OperatePrivilegeRequest
func (_e *Proxy_Expecter) OperatePrivilege(ctx interface{}, req interface{}) *Proxy_OperatePrivilege_Call {
	return &Proxy_OperatePrivilege_Call{Call: _e.mock.On("OperatePrivilege", ctx, req)}
}

func (_c *Proxy_OperatePrivilege_Call) Run(run func(ctx context.Context, req *milvuspb.OperatePrivilegeRequest)) *Proxy_OperatePrivilege_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.OperatePrivilegeRequest))
	})
	return _c
}

func (_c *Proxy_OperatePrivilege_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_OperatePrivilege_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// OperateUserRole provides a mock function with given fields: ctx, req
func (_m *Proxy) OperateUserRole(ctx context.Context, req *milvuspb.OperateUserRoleRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.OperateUserRoleRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.OperateUserRoleRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_OperateUserRole_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'OperateUserRole'
type Proxy_OperateUserRole_Call struct {
	*mock.Call
}

// OperateUserRole is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.OperateUserRoleRequest
func (_e *Proxy_Expecter) OperateUserRole(ctx interface{}, req interface{}) *Proxy_OperateUserRole_Call {
	return &Proxy_OperateUserRole_Call{Call: _e.mock.On("OperateUserRole", ctx, req)}
}

func (_c *Proxy_OperateUserRole_Call) Run(run func(ctx context.Context, req *milvuspb.OperateUserRoleRequest)) *Proxy_OperateUserRole_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.OperateUserRoleRequest))
	})
	return _c
}

func (_c *Proxy_OperateUserRole_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_OperateUserRole_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Query provides a mock function with given fields: ctx, request
func (_m *Proxy) Query(ctx context.Context, request *milvuspb.QueryRequest) (*milvuspb.QueryResults, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.QueryResults
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.QueryRequest) *milvuspb.QueryResults); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.QueryResults)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.QueryRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Query_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Query'
type Proxy_Query_Call struct {
	*mock.Call
}

// Query is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.QueryRequest
func (_e *Proxy_Expecter) Query(ctx interface{}, request interface{}) *Proxy_Query_Call {
	return &Proxy_Query_Call{Call: _e.mock.On("Query", ctx, request)}
}

func (_c *Proxy_Query_Call) Run(run func(ctx context.Context, request *milvuspb.QueryRequest)) *Proxy_Query_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.QueryRequest))
	})
	return _c
}

func (_c *Proxy_Query_Call) Return(_a0 *milvuspb.QueryResults, _a1 error) *Proxy_Query_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// RefreshPolicyInfoCache provides a mock function with given fields: ctx, req
func (_m *Proxy) RefreshPolicyInfoCache(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *proxypb.RefreshPolicyInfoCacheRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_RefreshPolicyInfoCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RefreshPolicyInfoCache'
type Proxy_RefreshPolicyInfoCache_Call struct {
	*mock.Call
}

// RefreshPolicyInfoCache is a helper method to define mock.On call
//  - ctx context.Context
//  - req *proxypb.RefreshPolicyInfoCacheRequest
func (_e *Proxy_Expecter) RefreshPolicyInfoCache(ctx interface{}, req interface{}) *Proxy_RefreshPolicyInfoCache_Call {
	return &Proxy_RefreshPolicyInfoCache_Call{Call: _e.mock.On("RefreshPolicyInfoCache", ctx, req)}
}

func (_c *Proxy_RefreshPolicyInfoCache_Call) Run(run func(ctx context.Context, req *proxypb.RefreshPolicyInfoCacheRequest)) *Proxy_RefreshPolicyInfoCache_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*proxypb.RefreshPolicyInfoCacheRequest))
	})
	return _c
}

func (_c *Proxy_RefreshPolicyInfoCache_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_RefreshPolicyInfoCache_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Register provides a mock function with given fields:
func (_m *Proxy) Register() 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
}

// Proxy_Register_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Register'
type Proxy_Register_Call struct {
	*mock.Call
}

// Register is a helper method to define mock.On call
func (_e *Proxy_Expecter) Register() *Proxy_Register_Call {
	return &Proxy_Register_Call{Call: _e.mock.On("Register")}
}

func (_c *Proxy_Register_Call) Run(run func()) *Proxy_Register_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Proxy_Register_Call) Return(_a0 error) *Proxy_Register_Call {
	_c.Call.Return(_a0)
	return _c
}

// RegisterLink provides a mock function with given fields: ctx, request
func (_m *Proxy) RegisterLink(ctx context.Context, request *milvuspb.RegisterLinkRequest) (*milvuspb.RegisterLinkResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.RegisterLinkResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RegisterLinkRequest) *milvuspb.RegisterLinkResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.RegisterLinkResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.RegisterLinkRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_RegisterLink_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RegisterLink'
type Proxy_RegisterLink_Call struct {
	*mock.Call
}

// RegisterLink is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.RegisterLinkRequest
func (_e *Proxy_Expecter) RegisterLink(ctx interface{}, request interface{}) *Proxy_RegisterLink_Call {
	return &Proxy_RegisterLink_Call{Call: _e.mock.On("RegisterLink", ctx, request)}
}

func (_c *Proxy_RegisterLink_Call) Run(run func(ctx context.Context, request *milvuspb.RegisterLinkRequest)) *Proxy_RegisterLink_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.RegisterLinkRequest))
	})
	return _c
}

func (_c *Proxy_RegisterLink_Call) Return(_a0 *milvuspb.RegisterLinkResponse, _a1 error) *Proxy_RegisterLink_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ReleaseCollection provides a mock function with given fields: ctx, request
func (_m *Proxy) ReleaseCollection(ctx context.Context, request *milvuspb.ReleaseCollectionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleaseCollectionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ReleaseCollectionRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ReleaseCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ReleaseCollection'
type Proxy_ReleaseCollection_Call struct {
	*mock.Call
}

// ReleaseCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.ReleaseCollectionRequest
func (_e *Proxy_Expecter) ReleaseCollection(ctx interface{}, request interface{}) *Proxy_ReleaseCollection_Call {
	return &Proxy_ReleaseCollection_Call{Call: _e.mock.On("ReleaseCollection", ctx, request)}
}

func (_c *Proxy_ReleaseCollection_Call) Run(run func(ctx context.Context, request *milvuspb.ReleaseCollectionRequest)) *Proxy_ReleaseCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ReleaseCollectionRequest))
	})
	return _c
}

func (_c *Proxy_ReleaseCollection_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_ReleaseCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ReleasePartitions provides a mock function with given fields: ctx, request
func (_m *Proxy) ReleasePartitions(ctx context.Context, request *milvuspb.ReleasePartitionsRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ReleasePartitionsRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ReleasePartitionsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ReleasePartitions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ReleasePartitions'
type Proxy_ReleasePartitions_Call struct {
	*mock.Call
}

// ReleasePartitions is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.ReleasePartitionsRequest
func (_e *Proxy_Expecter) ReleasePartitions(ctx interface{}, request interface{}) *Proxy_ReleasePartitions_Call {
	return &Proxy_ReleasePartitions_Call{Call: _e.mock.On("ReleasePartitions", ctx, request)}
}

func (_c *Proxy_ReleasePartitions_Call) Run(run func(ctx context.Context, request *milvuspb.ReleasePartitionsRequest)) *Proxy_ReleasePartitions_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ReleasePartitionsRequest))
	})
	return _c
}

func (_c *Proxy_ReleasePartitions_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_ReleasePartitions_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// RenameCollection provides a mock function with given fields: ctx, req
func (_m *Proxy) RenameCollection(ctx context.Context, req *milvuspb.RenameCollectionRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.RenameCollectionRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.RenameCollectionRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_RenameCollection_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'RenameCollection'
type Proxy_RenameCollection_Call struct {
	*mock.Call
}

// RenameCollection is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.RenameCollectionRequest
func (_e *Proxy_Expecter) RenameCollection(ctx interface{}, req interface{}) *Proxy_RenameCollection_Call {
	return &Proxy_RenameCollection_Call{Call: _e.mock.On("RenameCollection", ctx, req)}
}

func (_c *Proxy_RenameCollection_Call) Run(run func(ctx context.Context, req *milvuspb.RenameCollectionRequest)) *Proxy_RenameCollection_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.RenameCollectionRequest))
	})
	return _c
}

func (_c *Proxy_RenameCollection_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_RenameCollection_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Search provides a mock function with given fields: ctx, request
func (_m *Proxy) Search(ctx context.Context, request *milvuspb.SearchRequest) (*milvuspb.SearchResults, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.SearchResults
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SearchRequest) *milvuspb.SearchResults); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.SearchResults)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SearchRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Search_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Search'
type Proxy_Search_Call struct {
	*mock.Call
}

// Search is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.SearchRequest
func (_e *Proxy_Expecter) Search(ctx interface{}, request interface{}) *Proxy_Search_Call {
	return &Proxy_Search_Call{Call: _e.mock.On("Search", ctx, request)}
}

func (_c *Proxy_Search_Call) Run(run func(ctx context.Context, request *milvuspb.SearchRequest)) *Proxy_Search_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.SearchRequest))
	})
	return _c
}

func (_c *Proxy_Search_Call) Return(_a0 *milvuspb.SearchResults, _a1 error) *Proxy_Search_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// SelectGrant provides a mock function with given fields: ctx, req
func (_m *Proxy) SelectGrant(ctx context.Context, req *milvuspb.SelectGrantRequest) (*milvuspb.SelectGrantResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.SelectGrantResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectGrantRequest) *milvuspb.SelectGrantResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.SelectGrantResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectGrantRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_SelectGrant_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SelectGrant'
type Proxy_SelectGrant_Call struct {
	*mock.Call
}

// SelectGrant is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.SelectGrantRequest
func (_e *Proxy_Expecter) SelectGrant(ctx interface{}, req interface{}) *Proxy_SelectGrant_Call {
	return &Proxy_SelectGrant_Call{Call: _e.mock.On("SelectGrant", ctx, req)}
}

func (_c *Proxy_SelectGrant_Call) Run(run func(ctx context.Context, req *milvuspb.SelectGrantRequest)) *Proxy_SelectGrant_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.SelectGrantRequest))
	})
	return _c
}

func (_c *Proxy_SelectGrant_Call) Return(_a0 *milvuspb.SelectGrantResponse, _a1 error) *Proxy_SelectGrant_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// SelectRole provides a mock function with given fields: ctx, req
func (_m *Proxy) SelectRole(ctx context.Context, req *milvuspb.SelectRoleRequest) (*milvuspb.SelectRoleResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.SelectRoleResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectRoleRequest) *milvuspb.SelectRoleResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.SelectRoleResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectRoleRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_SelectRole_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SelectRole'
type Proxy_SelectRole_Call struct {
	*mock.Call
}

// SelectRole is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.SelectRoleRequest
func (_e *Proxy_Expecter) SelectRole(ctx interface{}, req interface{}) *Proxy_SelectRole_Call {
	return &Proxy_SelectRole_Call{Call: _e.mock.On("SelectRole", ctx, req)}
}

func (_c *Proxy_SelectRole_Call) Run(run func(ctx context.Context, req *milvuspb.SelectRoleRequest)) *Proxy_SelectRole_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.SelectRoleRequest))
	})
	return _c
}

func (_c *Proxy_SelectRole_Call) Return(_a0 *milvuspb.SelectRoleResponse, _a1 error) *Proxy_SelectRole_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// SelectUser provides a mock function with given fields: ctx, req
func (_m *Proxy) SelectUser(ctx context.Context, req *milvuspb.SelectUserRequest) (*milvuspb.SelectUserResponse, error) {
	ret := _m.Called(ctx, req)

	var r0 *milvuspb.SelectUserResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.SelectUserRequest) *milvuspb.SelectUserResponse); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.SelectUserResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.SelectUserRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_SelectUser_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SelectUser'
type Proxy_SelectUser_Call struct {
	*mock.Call
}

// SelectUser is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.SelectUserRequest
func (_e *Proxy_Expecter) SelectUser(ctx interface{}, req interface{}) *Proxy_SelectUser_Call {
	return &Proxy_SelectUser_Call{Call: _e.mock.On("SelectUser", ctx, req)}
}

func (_c *Proxy_SelectUser_Call) Run(run func(ctx context.Context, req *milvuspb.SelectUserRequest)) *Proxy_SelectUser_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.SelectUserRequest))
	})
	return _c
}

func (_c *Proxy_SelectUser_Call) Return(_a0 *milvuspb.SelectUserResponse, _a1 error) *Proxy_SelectUser_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// SetAddress provides a mock function with given fields: address
func (_m *Proxy) SetAddress(address string) {
	_m.Called(address)
}

// Proxy_SetAddress_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetAddress'
type Proxy_SetAddress_Call struct {
	*mock.Call
}

// SetAddress is a helper method to define mock.On call
//  - address string
func (_e *Proxy_Expecter) SetAddress(address interface{}) *Proxy_SetAddress_Call {
	return &Proxy_SetAddress_Call{Call: _e.mock.On("SetAddress", address)}
}

func (_c *Proxy_SetAddress_Call) Run(run func(address string)) *Proxy_SetAddress_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(string))
	})
	return _c
}

func (_c *Proxy_SetAddress_Call) Return() *Proxy_SetAddress_Call {
	_c.Call.Return()
	return _c
}

// SetDataCoordClient provides a mock function with given fields: dataCoord
func (_m *Proxy) SetDataCoordClient(dataCoord types.DataCoord) {
	_m.Called(dataCoord)
}

// Proxy_SetDataCoordClient_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetDataCoordClient'
type Proxy_SetDataCoordClient_Call struct {
	*mock.Call
}

// SetDataCoordClient is a helper method to define mock.On call
//  - dataCoord types.DataCoord
func (_e *Proxy_Expecter) SetDataCoordClient(dataCoord interface{}) *Proxy_SetDataCoordClient_Call {
	return &Proxy_SetDataCoordClient_Call{Call: _e.mock.On("SetDataCoordClient", dataCoord)}
}

func (_c *Proxy_SetDataCoordClient_Call) Run(run func(dataCoord types.DataCoord)) *Proxy_SetDataCoordClient_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(types.DataCoord))
	})
	return _c
}

func (_c *Proxy_SetDataCoordClient_Call) Return() *Proxy_SetDataCoordClient_Call {
	_c.Call.Return()
	return _c
}

// SetEtcdClient provides a mock function with given fields: etcdClient
func (_m *Proxy) SetEtcdClient(etcdClient *clientv3.Client) {
	_m.Called(etcdClient)
}

// Proxy_SetEtcdClient_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetEtcdClient'
type Proxy_SetEtcdClient_Call struct {
	*mock.Call
}

// SetEtcdClient is a helper method to define mock.On call
//  - etcdClient *clientv3.Client
func (_e *Proxy_Expecter) SetEtcdClient(etcdClient interface{}) *Proxy_SetEtcdClient_Call {
	return &Proxy_SetEtcdClient_Call{Call: _e.mock.On("SetEtcdClient", etcdClient)}
}

func (_c *Proxy_SetEtcdClient_Call) Run(run func(etcdClient *clientv3.Client)) *Proxy_SetEtcdClient_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(*clientv3.Client))
	})
	return _c
}

func (_c *Proxy_SetEtcdClient_Call) Return() *Proxy_SetEtcdClient_Call {
	_c.Call.Return()
	return _c
}

// SetQueryCoordClient provides a mock function with given fields: queryCoord
func (_m *Proxy) SetQueryCoordClient(queryCoord types.QueryCoord) {
	_m.Called(queryCoord)
}

// Proxy_SetQueryCoordClient_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetQueryCoordClient'
type Proxy_SetQueryCoordClient_Call struct {
	*mock.Call
}

// SetQueryCoordClient is a helper method to define mock.On call
//  - queryCoord types.QueryCoord
func (_e *Proxy_Expecter) SetQueryCoordClient(queryCoord interface{}) *Proxy_SetQueryCoordClient_Call {
	return &Proxy_SetQueryCoordClient_Call{Call: _e.mock.On("SetQueryCoordClient", queryCoord)}
}

func (_c *Proxy_SetQueryCoordClient_Call) Run(run func(queryCoord types.QueryCoord)) *Proxy_SetQueryCoordClient_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(types.QueryCoord))
	})
	return _c
}

func (_c *Proxy_SetQueryCoordClient_Call) Return() *Proxy_SetQueryCoordClient_Call {
	_c.Call.Return()
	return _c
}

// SetQueryNodeCreator provides a mock function with given fields: _a0
func (_m *Proxy) SetQueryNodeCreator(_a0 func(context.Context, string) (types.QueryNode, error)) {
	_m.Called(_a0)
}

// Proxy_SetQueryNodeCreator_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetQueryNodeCreator'
type Proxy_SetQueryNodeCreator_Call struct {
	*mock.Call
}

// SetQueryNodeCreator is a helper method to define mock.On call
//  - _a0 func(context.Context , string)(types.QueryNode , error)
func (_e *Proxy_Expecter) SetQueryNodeCreator(_a0 interface{}) *Proxy_SetQueryNodeCreator_Call {
	return &Proxy_SetQueryNodeCreator_Call{Call: _e.mock.On("SetQueryNodeCreator", _a0)}
}

func (_c *Proxy_SetQueryNodeCreator_Call) Run(run func(_a0 func(context.Context, string) (types.QueryNode, error))) *Proxy_SetQueryNodeCreator_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(func(context.Context, string) (types.QueryNode, error)))
	})
	return _c
}

func (_c *Proxy_SetQueryNodeCreator_Call) Return() *Proxy_SetQueryNodeCreator_Call {
	_c.Call.Return()
	return _c
}

// SetRates provides a mock function with given fields: ctx, req
func (_m *Proxy) SetRates(ctx context.Context, req *proxypb.SetRatesRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.SetRatesRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *proxypb.SetRatesRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_SetRates_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetRates'
type Proxy_SetRates_Call struct {
	*mock.Call
}

// SetRates is a helper method to define mock.On call
//  - ctx context.Context
//  - req *proxypb.SetRatesRequest
func (_e *Proxy_Expecter) SetRates(ctx interface{}, req interface{}) *Proxy_SetRates_Call {
	return &Proxy_SetRates_Call{Call: _e.mock.On("SetRates", ctx, req)}
}

func (_c *Proxy_SetRates_Call) Run(run func(ctx context.Context, req *proxypb.SetRatesRequest)) *Proxy_SetRates_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*proxypb.SetRatesRequest))
	})
	return _c
}

func (_c *Proxy_SetRates_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_SetRates_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// SetRootCoordClient provides a mock function with given fields: rootCoord
func (_m *Proxy) SetRootCoordClient(rootCoord types.RootCoord) {
	_m.Called(rootCoord)
}

// Proxy_SetRootCoordClient_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'SetRootCoordClient'
type Proxy_SetRootCoordClient_Call struct {
	*mock.Call
}

// SetRootCoordClient is a helper method to define mock.On call
//  - rootCoord types.RootCoord
func (_e *Proxy_Expecter) SetRootCoordClient(rootCoord interface{}) *Proxy_SetRootCoordClient_Call {
	return &Proxy_SetRootCoordClient_Call{Call: _e.mock.On("SetRootCoordClient", rootCoord)}
}

func (_c *Proxy_SetRootCoordClient_Call) Run(run func(rootCoord types.RootCoord)) *Proxy_SetRootCoordClient_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(types.RootCoord))
	})
	return _c
}

func (_c *Proxy_SetRootCoordClient_Call) Return() *Proxy_SetRootCoordClient_Call {
	_c.Call.Return()
	return _c
}

// ShowCollections provides a mock function with given fields: ctx, request
func (_m *Proxy) ShowCollections(ctx context.Context, request *milvuspb.ShowCollectionsRequest) (*milvuspb.ShowCollectionsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.ShowCollectionsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowCollectionsRequest) *milvuspb.ShowCollectionsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ShowCollectionsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ShowCollectionsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ShowCollections_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ShowCollections'
type Proxy_ShowCollections_Call struct {
	*mock.Call
}

// ShowCollections is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.ShowCollectionsRequest
func (_e *Proxy_Expecter) ShowCollections(ctx interface{}, request interface{}) *Proxy_ShowCollections_Call {
	return &Proxy_ShowCollections_Call{Call: _e.mock.On("ShowCollections", ctx, request)}
}

func (_c *Proxy_ShowCollections_Call) Run(run func(ctx context.Context, request *milvuspb.ShowCollectionsRequest)) *Proxy_ShowCollections_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ShowCollectionsRequest))
	})
	return _c
}

func (_c *Proxy_ShowCollections_Call) Return(_a0 *milvuspb.ShowCollectionsResponse, _a1 error) *Proxy_ShowCollections_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// ShowPartitions provides a mock function with given fields: ctx, request
func (_m *Proxy) ShowPartitions(ctx context.Context, request *milvuspb.ShowPartitionsRequest) (*milvuspb.ShowPartitionsResponse, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.ShowPartitionsResponse
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.ShowPartitionsRequest) *milvuspb.ShowPartitionsResponse); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.ShowPartitionsResponse)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.ShowPartitionsRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_ShowPartitions_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'ShowPartitions'
type Proxy_ShowPartitions_Call struct {
	*mock.Call
}

// ShowPartitions is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.ShowPartitionsRequest
func (_e *Proxy_Expecter) ShowPartitions(ctx interface{}, request interface{}) *Proxy_ShowPartitions_Call {
	return &Proxy_ShowPartitions_Call{Call: _e.mock.On("ShowPartitions", ctx, request)}
}

func (_c *Proxy_ShowPartitions_Call) Run(run func(ctx context.Context, request *milvuspb.ShowPartitionsRequest)) *Proxy_ShowPartitions_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.ShowPartitionsRequest))
	})
	return _c
}

func (_c *Proxy_ShowPartitions_Call) Return(_a0 *milvuspb.ShowPartitionsResponse, _a1 error) *Proxy_ShowPartitions_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// Start provides a mock function with given fields:
func (_m *Proxy) Start() 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
}

// Proxy_Start_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Start'
type Proxy_Start_Call struct {
	*mock.Call
}

// Start is a helper method to define mock.On call
func (_e *Proxy_Expecter) Start() *Proxy_Start_Call {
	return &Proxy_Start_Call{Call: _e.mock.On("Start")}
}

func (_c *Proxy_Start_Call) Run(run func()) *Proxy_Start_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Proxy_Start_Call) Return(_a0 error) *Proxy_Start_Call {
	_c.Call.Return(_a0)
	return _c
}

// Stop provides a mock function with given fields:
func (_m *Proxy) Stop() 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
}

// Proxy_Stop_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Stop'
type Proxy_Stop_Call struct {
	*mock.Call
}

// Stop is a helper method to define mock.On call
func (_e *Proxy_Expecter) Stop() *Proxy_Stop_Call {
	return &Proxy_Stop_Call{Call: _e.mock.On("Stop")}
}

func (_c *Proxy_Stop_Call) Run(run func()) *Proxy_Stop_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run()
	})
	return _c
}

func (_c *Proxy_Stop_Call) Return(_a0 error) *Proxy_Stop_Call {
	_c.Call.Return(_a0)
	return _c
}

// TransferNode provides a mock function with given fields: ctx, req
func (_m *Proxy) TransferNode(ctx context.Context, req *milvuspb.TransferNodeRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferNodeRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.TransferNodeRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_TransferNode_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'TransferNode'
type Proxy_TransferNode_Call struct {
	*mock.Call
}

// TransferNode is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.TransferNodeRequest
func (_e *Proxy_Expecter) TransferNode(ctx interface{}, req interface{}) *Proxy_TransferNode_Call {
	return &Proxy_TransferNode_Call{Call: _e.mock.On("TransferNode", ctx, req)}
}

func (_c *Proxy_TransferNode_Call) Run(run func(ctx context.Context, req *milvuspb.TransferNodeRequest)) *Proxy_TransferNode_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.TransferNodeRequest))
	})
	return _c
}

func (_c *Proxy_TransferNode_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_TransferNode_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// TransferReplica provides a mock function with given fields: ctx, req
func (_m *Proxy) TransferReplica(ctx context.Context, req *milvuspb.TransferReplicaRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.TransferReplicaRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.TransferReplicaRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_TransferReplica_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'TransferReplica'
type Proxy_TransferReplica_Call struct {
	*mock.Call
}

// TransferReplica is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.TransferReplicaRequest
func (_e *Proxy_Expecter) TransferReplica(ctx interface{}, req interface{}) *Proxy_TransferReplica_Call {
	return &Proxy_TransferReplica_Call{Call: _e.mock.On("TransferReplica", ctx, req)}
}

func (_c *Proxy_TransferReplica_Call) Run(run func(ctx context.Context, req *milvuspb.TransferReplicaRequest)) *Proxy_TransferReplica_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.TransferReplicaRequest))
	})
	return _c
}

func (_c *Proxy_TransferReplica_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_TransferReplica_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// UpdateCredential provides a mock function with given fields: ctx, req
func (_m *Proxy) UpdateCredential(ctx context.Context, req *milvuspb.UpdateCredentialRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, req)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpdateCredentialRequest) *commonpb.Status); ok {
		r0 = rf(ctx, req)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.UpdateCredentialRequest) error); ok {
		r1 = rf(ctx, req)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_UpdateCredential_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpdateCredential'
type Proxy_UpdateCredential_Call struct {
	*mock.Call
}

// UpdateCredential is a helper method to define mock.On call
//  - ctx context.Context
//  - req *milvuspb.UpdateCredentialRequest
func (_e *Proxy_Expecter) UpdateCredential(ctx interface{}, req interface{}) *Proxy_UpdateCredential_Call {
	return &Proxy_UpdateCredential_Call{Call: _e.mock.On("UpdateCredential", ctx, req)}
}

func (_c *Proxy_UpdateCredential_Call) Run(run func(ctx context.Context, req *milvuspb.UpdateCredentialRequest)) *Proxy_UpdateCredential_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.UpdateCredentialRequest))
	})
	return _c
}

func (_c *Proxy_UpdateCredential_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_UpdateCredential_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// UpdateCredentialCache provides a mock function with given fields: ctx, request
func (_m *Proxy) UpdateCredentialCache(ctx context.Context, request *proxypb.UpdateCredCacheRequest) (*commonpb.Status, error) {
	ret := _m.Called(ctx, request)

	var r0 *commonpb.Status
	if rf, ok := ret.Get(0).(func(context.Context, *proxypb.UpdateCredCacheRequest) *commonpb.Status); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*commonpb.Status)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *proxypb.UpdateCredCacheRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_UpdateCredentialCache_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpdateCredentialCache'
type Proxy_UpdateCredentialCache_Call struct {
	*mock.Call
}

// UpdateCredentialCache is a helper method to define mock.On call
//  - ctx context.Context
//  - request *proxypb.UpdateCredCacheRequest
func (_e *Proxy_Expecter) UpdateCredentialCache(ctx interface{}, request interface{}) *Proxy_UpdateCredentialCache_Call {
	return &Proxy_UpdateCredentialCache_Call{Call: _e.mock.On("UpdateCredentialCache", ctx, request)}
}

func (_c *Proxy_UpdateCredentialCache_Call) Run(run func(ctx context.Context, request *proxypb.UpdateCredCacheRequest)) *Proxy_UpdateCredentialCache_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*proxypb.UpdateCredCacheRequest))
	})
	return _c
}

func (_c *Proxy_UpdateCredentialCache_Call) Return(_a0 *commonpb.Status, _a1 error) *Proxy_UpdateCredentialCache_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

// UpdateStateCode provides a mock function with given fields: stateCode
func (_m *Proxy) UpdateStateCode(stateCode commonpb.StateCode) {
	_m.Called(stateCode)
}

// Proxy_UpdateStateCode_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'UpdateStateCode'
type Proxy_UpdateStateCode_Call struct {
	*mock.Call
}

// UpdateStateCode is a helper method to define mock.On call
//  - stateCode commonpb.StateCode
func (_e *Proxy_Expecter) UpdateStateCode(stateCode interface{}) *Proxy_UpdateStateCode_Call {
	return &Proxy_UpdateStateCode_Call{Call: _e.mock.On("UpdateStateCode", stateCode)}
}

func (_c *Proxy_UpdateStateCode_Call) Run(run func(stateCode commonpb.StateCode)) *Proxy_UpdateStateCode_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(commonpb.StateCode))
	})
	return _c
}

func (_c *Proxy_UpdateStateCode_Call) Return() *Proxy_UpdateStateCode_Call {
	_c.Call.Return()
	return _c
}

// Upsert provides a mock function with given fields: ctx, request
func (_m *Proxy) Upsert(ctx context.Context, request *milvuspb.UpsertRequest) (*milvuspb.MutationResult, error) {
	ret := _m.Called(ctx, request)

	var r0 *milvuspb.MutationResult
	if rf, ok := ret.Get(0).(func(context.Context, *milvuspb.UpsertRequest) *milvuspb.MutationResult); ok {
		r0 = rf(ctx, request)
	} else {
		if ret.Get(0) != nil {
			r0 = ret.Get(0).(*milvuspb.MutationResult)
		}
	}

	var r1 error
	if rf, ok := ret.Get(1).(func(context.Context, *milvuspb.UpsertRequest) error); ok {
		r1 = rf(ctx, request)
	} else {
		r1 = ret.Error(1)
	}

	return r0, r1
}

// Proxy_Upsert_Call is a *mock.Call that shadows Run/Return methods with type explicit version for method 'Upsert'
type Proxy_Upsert_Call struct {
	*mock.Call
}

// Upsert is a helper method to define mock.On call
//  - ctx context.Context
//  - request *milvuspb.UpsertRequest
func (_e *Proxy_Expecter) Upsert(ctx interface{}, request interface{}) *Proxy_Upsert_Call {
	return &Proxy_Upsert_Call{Call: _e.mock.On("Upsert", ctx, request)}
}

func (_c *Proxy_Upsert_Call) Run(run func(ctx context.Context, request *milvuspb.UpsertRequest)) *Proxy_Upsert_Call {
	_c.Call.Run(func(args mock.Arguments) {
		run(args[0].(context.Context), args[1].(*milvuspb.UpsertRequest))
	})
	return _c
}

func (_c *Proxy_Upsert_Call) Return(_a0 *milvuspb.MutationResult, _a1 error) *Proxy_Upsert_Call {
	_c.Call.Return(_a0, _a1)
	return _c
}

type mockConstructorTestingTNewProxy interface {
	mock.TestingT
	Cleanup(func())
}

// NewProxy creates a new instance of Proxy. It also registers a testing interface on the mock and a cleanup function to assert the mocks expectations.
func NewProxy(t mockConstructorTestingTNewProxy) *Proxy {
	mock := &Proxy{}
	mock.Mock.Test(t)

	t.Cleanup(func() { mock.AssertExpectations(t) })

	return mock
}