mock_deviceplugin.go

Documentation: edge-infra.dev/pkg/sds/devices/agent/mocks

     1  // Code generated by MockGen. DO NOT EDIT.
     2  // Source: edge-infra.dev/pkg/sds/devices/k8s/device-plugins (interfaces: Plugin)
     3  
     4  // Package mocks is a generated GoMock package.
     5  package mocks
     6  
     7  import (
     8  	context "context"
     9  	reflect "reflect"
    10  
    11  	v1 "edge-infra.dev/pkg/sds/devices/k8s/apis/v1"
    12  	gomock "github.com/golang/mock/gomock"
    13  	v1beta1 "k8s.io/kubelet/pkg/apis/deviceplugin/v1beta1"
    14  )
    15  
    16  // MockPlugin is a mock of Plugin interface.
    17  type MockPlugin struct {
    18  	ctrl     *gomock.Controller
    19  	recorder *MockPluginMockRecorder
    20  }
    21  
    22  // MockPluginMockRecorder is the mock recorder for MockPlugin.
    23  type MockPluginMockRecorder struct {
    24  	mock *MockPlugin
    25  }
    26  
    27  // NewMockPlugin creates a new mock instance.
    28  func NewMockPlugin(ctrl *gomock.Controller) *MockPlugin {
    29  	mock := &MockPlugin{ctrl: ctrl}
    30  	mock.recorder = &MockPluginMockRecorder{mock}
    31  	return mock
    32  }
    33  
    34  // EXPECT returns an object that allows the caller to indicate expected use.
    35  func (m *MockPlugin) EXPECT() *MockPluginMockRecorder {
    36  	return m.recorder
    37  }
    38  
    39  // Allocate mocks base method.
    40  func (m *MockPlugin) Allocate(arg0 context.Context, arg1 *v1beta1.AllocateRequest) (*v1beta1.AllocateResponse, error) {
    41  	m.ctrl.T.Helper()
    42  	ret := m.ctrl.Call(m, "Allocate", arg0, arg1)
    43  	ret0, _ := ret[0].(*v1beta1.AllocateResponse)
    44  	ret1, _ := ret[1].(error)
    45  	return ret0, ret1
    46  }
    47  
    48  // Allocate indicates an expected call of Allocate.
    49  func (mr *MockPluginMockRecorder) Allocate(arg0, arg1 interface{}) *gomock.Call {
    50  	mr.mock.ctrl.T.Helper()
    51  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Allocate", reflect.TypeOf((*MockPlugin)(nil).Allocate), arg0, arg1)
    52  }
    53  
    54  // GetDevicePluginOptions mocks base method.
    55  func (m *MockPlugin) GetDevicePluginOptions(arg0 context.Context, arg1 *v1beta1.Empty) (*v1beta1.DevicePluginOptions, error) {
    56  	m.ctrl.T.Helper()
    57  	ret := m.ctrl.Call(m, "GetDevicePluginOptions", arg0, arg1)
    58  	ret0, _ := ret[0].(*v1beta1.DevicePluginOptions)
    59  	ret1, _ := ret[1].(error)
    60  	return ret0, ret1
    61  }
    62  
    63  // GetDevicePluginOptions indicates an expected call of GetDevicePluginOptions.
    64  func (mr *MockPluginMockRecorder) GetDevicePluginOptions(arg0, arg1 interface{}) *gomock.Call {
    65  	mr.mock.ctrl.T.Helper()
    66  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDevicePluginOptions", reflect.TypeOf((*MockPlugin)(nil).GetDevicePluginOptions), arg0, arg1)
    67  }
    68  
    69  // GetPreferredAllocation mocks base method.
    70  func (m *MockPlugin) GetPreferredAllocation(arg0 context.Context, arg1 *v1beta1.PreferredAllocationRequest) (*v1beta1.PreferredAllocationResponse, error) {
    71  	m.ctrl.T.Helper()
    72  	ret := m.ctrl.Call(m, "GetPreferredAllocation", arg0, arg1)
    73  	ret0, _ := ret[0].(*v1beta1.PreferredAllocationResponse)
    74  	ret1, _ := ret[1].(error)
    75  	return ret0, ret1
    76  }
    77  
    78  // GetPreferredAllocation indicates an expected call of GetPreferredAllocation.
    79  func (mr *MockPluginMockRecorder) GetPreferredAllocation(arg0, arg1 interface{}) *gomock.Call {
    80  	mr.mock.ctrl.T.Helper()
    81  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPreferredAllocation", reflect.TypeOf((*MockPlugin)(nil).GetPreferredAllocation), arg0, arg1)
    82  }
    83  
    84  // IsRunning mocks base method.
    85  func (m *MockPlugin) IsRunning() bool {
    86  	m.ctrl.T.Helper()
    87  	ret := m.ctrl.Call(m, "IsRunning")
    88  	ret0, _ := ret[0].(bool)
    89  	return ret0
    90  }
    91  
    92  // IsRunning indicates an expected call of IsRunning.
    93  func (mr *MockPluginMockRecorder) IsRunning() *gomock.Call {
    94  	mr.mock.ctrl.T.Helper()
    95  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsRunning", reflect.TypeOf((*MockPlugin)(nil).IsRunning))
    96  }
    97  
    98  // ListAndWatch mocks base method.
    99  func (m *MockPlugin) ListAndWatch(arg0 *v1beta1.Empty, arg1 v1beta1.DevicePlugin_ListAndWatchServer) error {
   100  	m.ctrl.T.Helper()
   101  	ret := m.ctrl.Call(m, "ListAndWatch", arg0, arg1)
   102  	ret0, _ := ret[0].(error)
   103  	return ret0
   104  }
   105  
   106  // ListAndWatch indicates an expected call of ListAndWatch.
   107  func (mr *MockPluginMockRecorder) ListAndWatch(arg0, arg1 interface{}) *gomock.Call {
   108  	mr.mock.ctrl.T.Helper()
   109  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAndWatch", reflect.TypeOf((*MockPlugin)(nil).ListAndWatch), arg0, arg1)
   110  }
   111  
   112  // PreStartContainer mocks base method.
   113  func (m *MockPlugin) PreStartContainer(arg0 context.Context, arg1 *v1beta1.PreStartContainerRequest) (*v1beta1.PreStartContainerResponse, error) {
   114  	m.ctrl.T.Helper()
   115  	ret := m.ctrl.Call(m, "PreStartContainer", arg0, arg1)
   116  	ret0, _ := ret[0].(*v1beta1.PreStartContainerResponse)
   117  	ret1, _ := ret[1].(error)
   118  	return ret0, ret1
   119  }
   120  
   121  // PreStartContainer indicates an expected call of PreStartContainer.
   122  func (mr *MockPluginMockRecorder) PreStartContainer(arg0, arg1 interface{}) *gomock.Call {
   123  	mr.mock.ctrl.T.Helper()
   124  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreStartContainer", reflect.TypeOf((*MockPlugin)(nil).PreStartContainer), arg0, arg1)
   125  }
   126  
   127  // Remove mocks base method.
   128  func (m *MockPlugin) Remove() {
   129  	m.ctrl.T.Helper()
   130  	m.ctrl.Call(m, "Remove")
   131  }
   132  
   133  // Remove indicates an expected call of Remove.
   134  func (mr *MockPluginMockRecorder) Remove() *gomock.Call {
   135  	mr.mock.ctrl.T.Helper()
   136  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Remove", reflect.TypeOf((*MockPlugin)(nil).Remove))
   137  }
   138  
   139  // Run mocks base method.
   140  func (m *MockPlugin) Run(arg0 context.Context) {
   141  	m.ctrl.T.Helper()
   142  	m.ctrl.Call(m, "Run", arg0)
   143  }
   144  
   145  // Run indicates an expected call of Run.
   146  func (mr *MockPluginMockRecorder) Run(arg0 interface{}) *gomock.Call {
   147  	mr.mock.ctrl.T.Helper()
   148  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Run", reflect.TypeOf((*MockPlugin)(nil).Run), arg0)
   149  }
   150  
   151  // Stop mocks base method.
   152  func (m *MockPlugin) Stop() {
   153  	m.ctrl.T.Helper()
   154  	m.ctrl.Call(m, "Stop")
   155  }
   156  
   157  // Stop indicates an expected call of Stop.
   158  func (mr *MockPluginMockRecorder) Stop() *gomock.Call {
   159  	mr.mock.ctrl.T.Helper()
   160  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockPlugin)(nil).Stop))
   161  }
   162  
   163  // Update mocks base method.
   164  func (m *MockPlugin) Update(arg0 *v1.DeviceClass) {
   165  	m.ctrl.T.Helper()
   166  	m.ctrl.Call(m, "Update", arg0)
   167  }
   168  
   169  // Update indicates an expected call of Update.
   170  func (mr *MockPluginMockRecorder) Update(arg0 interface{}) *gomock.Call {
   171  	mr.mock.ctrl.T.Helper()
   172  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockPlugin)(nil).Update), arg0)
   173  }
   174
View as plain text