github.com/pingcap/tiflow@v0.0.0-20240520035814-5bf52d54e205/engine/enginepb/mock/broker_mock.go (about)

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/pingcap/tiflow/engine/enginepb (interfaces: BrokerServiceClient)
     3  
     4  // Package mock is a generated GoMock package.
     5  package mock
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	gomock "github.com/golang/mock/gomock"
    12  	enginepb "github.com/pingcap/tiflow/engine/enginepb"
    13  	grpc "google.golang.org/grpc"
    14  )
    15  
    16  // MockBrokerServiceClient is a mock of BrokerServiceClient interface.
    17  type MockBrokerServiceClient struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockBrokerServiceClientMockRecorder
    20  }
    21  
    22  // MockBrokerServiceClientMockRecorder is the mock recorder for MockBrokerServiceClient.
    23  type MockBrokerServiceClientMockRecorder struct {
    24  	mock *MockBrokerServiceClient
    25  }
    26  
    27  // NewMockBrokerServiceClient creates a new mock instance.
    28  func NewMockBrokerServiceClient(ctrl *gomock.Controller) *MockBrokerServiceClient {
    29  	mock := &MockBrokerServiceClient{ctrl: ctrl}
    30  	mock.recorder = &MockBrokerServiceClientMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use.
    35  func (m *MockBrokerServiceClient) EXPECT() *MockBrokerServiceClientMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // RemoveResource mocks base method.
    40  func (m *MockBrokerServiceClient) RemoveResource(arg0 context.Context, arg1 *enginepb.RemoveLocalResourceRequest, arg2 ...grpc.CallOption) (*enginepb.RemoveLocalResourceResponse, error) {
    41  	m.ctrl.T.Helper()
    42  	varargs := []interface{}{arg0, arg1}
    43  	for _, a := range arg2 {
    44  		varargs = append(varargs, a)
    45  	}
    46  	ret := m.ctrl.Call(m, "RemoveResource", varargs...)
    47  	ret0, _ := ret[0].(*enginepb.RemoveLocalResourceResponse)
    48  	ret1, _ := ret[1].(error)
    49  	return ret0, ret1
    50  }
    51  
    52  // RemoveResource indicates an expected call of RemoveResource.
    53  func (mr *MockBrokerServiceClientMockRecorder) RemoveResource(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    54  	mr.mock.ctrl.T.Helper()
    55  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    56  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveResource", reflect.TypeOf((*MockBrokerServiceClient)(nil).RemoveResource), varargs...)
    57  }