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

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: github.com/pingcap/tiflow/engine/enginepb (interfaces: ResourceManagerClient)
     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  // MockResourceManagerClient is a mock of ResourceManagerClient interface.
    17  type MockResourceManagerClient struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockResourceManagerClientMockRecorder
    20  }
    21  
    22  // MockResourceManagerClientMockRecorder is the mock recorder for MockResourceManagerClient.
    23  type MockResourceManagerClientMockRecorder struct {
    24  	mock *MockResourceManagerClient
    25  }
    26  
    27  // NewMockResourceManagerClient creates a new mock instance.
    28  func NewMockResourceManagerClient(ctrl *gomock.Controller) *MockResourceManagerClient {
    29  	mock := &MockResourceManagerClient{ctrl: ctrl}
    30  	mock.recorder = &MockResourceManagerClientMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use.
    35  func (m *MockResourceManagerClient) EXPECT() *MockResourceManagerClientMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // CreateResource mocks base method.
    40  func (m *MockResourceManagerClient) CreateResource(arg0 context.Context, arg1 *enginepb.CreateResourceRequest, arg2 ...grpc.CallOption) (*enginepb.CreateResourceResponse, 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, "CreateResource", varargs...)
    47  	ret0, _ := ret[0].(*enginepb.CreateResourceResponse)
    48  	ret1, _ := ret[1].(error)
    49  	return ret0, ret1
    50  }
    51  
    52  // CreateResource indicates an expected call of CreateResource.
    53  func (mr *MockResourceManagerClientMockRecorder) CreateResource(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, "CreateResource", reflect.TypeOf((*MockResourceManagerClient)(nil).CreateResource), varargs...)
    57  }
    58  
    59  // QueryResource mocks base method.
    60  func (m *MockResourceManagerClient) QueryResource(arg0 context.Context, arg1 *enginepb.QueryResourceRequest, arg2 ...grpc.CallOption) (*enginepb.QueryResourceResponse, error) {
    61  	m.ctrl.T.Helper()
    62  	varargs := []interface{}{arg0, arg1}
    63  	for _, a := range arg2 {
    64  		varargs = append(varargs, a)
    65  	}
    66  	ret := m.ctrl.Call(m, "QueryResource", varargs...)
    67  	ret0, _ := ret[0].(*enginepb.QueryResourceResponse)
    68  	ret1, _ := ret[1].(error)
    69  	return ret0, ret1
    70  }
    71  
    72  // QueryResource indicates an expected call of QueryResource.
    73  func (mr *MockResourceManagerClientMockRecorder) QueryResource(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    74  	mr.mock.ctrl.T.Helper()
    75  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    76  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "QueryResource", reflect.TypeOf((*MockResourceManagerClient)(nil).QueryResource), varargs...)
    77  }
    78  
    79  // RemoveResource mocks base method.
    80  func (m *MockResourceManagerClient) RemoveResource(arg0 context.Context, arg1 *enginepb.RemoveResourceRequest, arg2 ...grpc.CallOption) (*enginepb.RemoveResourceResponse, error) {
    81  	m.ctrl.T.Helper()
    82  	varargs := []interface{}{arg0, arg1}
    83  	for _, a := range arg2 {
    84  		varargs = append(varargs, a)
    85  	}
    86  	ret := m.ctrl.Call(m, "RemoveResource", varargs...)
    87  	ret0, _ := ret[0].(*enginepb.RemoveResourceResponse)
    88  	ret1, _ := ret[1].(error)
    89  	return ret0, ret1
    90  }
    91  
    92  // RemoveResource indicates an expected call of RemoveResource.
    93  func (mr *MockResourceManagerClientMockRecorder) RemoveResource(arg0, arg1 interface{}, arg2 ...interface{}) *gomock.Call {
    94  	mr.mock.ctrl.T.Helper()
    95  	varargs := append([]interface{}{arg0, arg1}, arg2...)
    96  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemoveResource", reflect.TypeOf((*MockResourceManagerClient)(nil).RemoveResource), varargs...)
    97  }