provider_mock.go

Documentation: k8s.io/kubernetes/pkg/kubelet/apis/podresources/testing

     1  /*
     2  Copyright The Kubernetes Authors.
     3  
     4  Licensed under the Apache License, Version 2.0 (the "License");
     5  you may not use this file except in compliance with the License.
     6  You may obtain a copy of the License at
     7  
     8      http://www.apache.org/licenses/LICENSE-2.0
     9  
    10  Unless required by applicable law or agreed to in writing, software
    11  distributed under the License is distributed on an "AS IS" BASIS,
    12  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    13  See the License for the specific language governing permissions and
    14  limitations under the License.
    15  */
    16  
    17  // Code generated by MockGen. DO NOT EDIT.
    18  // Source: types.go
    19  
    20  // Package testing is a generated GoMock package.
    21  package testing
    22  
    23  import (
    24  	reflect "reflect"
    25  
    26  	gomock "github.com/golang/mock/gomock"
    27  	v1 "k8s.io/api/core/v1"
    28  	v10 "k8s.io/kubelet/pkg/apis/podresources/v1"
    29  )
    30  
    31  // MockDevicesProvider is a mock of DevicesProvider interface.
    32  type MockDevicesProvider struct {
    33  	ctrl     *gomock.Controller
    34  	recorder *MockDevicesProviderMockRecorder
    35  }
    36  
    37  // MockDevicesProviderMockRecorder is the mock recorder for MockDevicesProvider.
    38  type MockDevicesProviderMockRecorder struct {
    39  	mock *MockDevicesProvider
    40  }
    41  
    42  // NewMockDevicesProvider creates a new mock instance.
    43  func NewMockDevicesProvider(ctrl *gomock.Controller) *MockDevicesProvider {
    44  	mock := &MockDevicesProvider{ctrl: ctrl}
    45  	mock.recorder = &MockDevicesProviderMockRecorder{mock}
    46  	return mock
    47  }
    48  
    49  // EXPECT returns an object that allows the caller to indicate expected use.
    50  func (m *MockDevicesProvider) EXPECT() *MockDevicesProviderMockRecorder {
    51  	return m.recorder
    52  }
    53  
    54  // GetAllocatableDevices mocks base method.
    55  func (m *MockDevicesProvider) GetAllocatableDevices() []*v10.ContainerDevices {
    56  	m.ctrl.T.Helper()
    57  	ret := m.ctrl.Call(m, "GetAllocatableDevices")
    58  	ret0, _ := ret[0].([]*v10.ContainerDevices)
    59  	return ret0
    60  }
    61  
    62  // GetAllocatableDevices indicates an expected call of GetAllocatableDevices.
    63  func (mr *MockDevicesProviderMockRecorder) GetAllocatableDevices() *gomock.Call {
    64  	mr.mock.ctrl.T.Helper()
    65  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllocatableDevices", reflect.TypeOf((*MockDevicesProvider)(nil).GetAllocatableDevices))
    66  }
    67  
    68  // GetDevices mocks base method.
    69  func (m *MockDevicesProvider) GetDevices(podUID, containerName string) []*v10.ContainerDevices {
    70  	m.ctrl.T.Helper()
    71  	ret := m.ctrl.Call(m, "GetDevices", podUID, containerName)
    72  	ret0, _ := ret[0].([]*v10.ContainerDevices)
    73  	return ret0
    74  }
    75  
    76  // GetDevices indicates an expected call of GetDevices.
    77  func (mr *MockDevicesProviderMockRecorder) GetDevices(podUID, containerName interface{}) *gomock.Call {
    78  	mr.mock.ctrl.T.Helper()
    79  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDevices", reflect.TypeOf((*MockDevicesProvider)(nil).GetDevices), podUID, containerName)
    80  }
    81  
    82  // UpdateAllocatedDevices mocks base method.
    83  func (m *MockDevicesProvider) UpdateAllocatedDevices() {
    84  	m.ctrl.T.Helper()
    85  	m.ctrl.Call(m, "UpdateAllocatedDevices")
    86  }
    87  
    88  // UpdateAllocatedDevices indicates an expected call of UpdateAllocatedDevices.
    89  func (mr *MockDevicesProviderMockRecorder) UpdateAllocatedDevices() *gomock.Call {
    90  	mr.mock.ctrl.T.Helper()
    91  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "UpdateAllocatedDevices", reflect.TypeOf((*MockDevicesProvider)(nil).UpdateAllocatedDevices))
    92  }
    93  
    94  // MockPodsProvider is a mock of PodsProvider interface.
    95  type MockPodsProvider struct {
    96  	ctrl     *gomock.Controller
    97  	recorder *MockPodsProviderMockRecorder
    98  }
    99  
   100  // MockPodsProviderMockRecorder is the mock recorder for MockPodsProvider.
   101  type MockPodsProviderMockRecorder struct {
   102  	mock *MockPodsProvider
   103  }
   104  
   105  // NewMockPodsProvider creates a new mock instance.
   106  func NewMockPodsProvider(ctrl *gomock.Controller) *MockPodsProvider {
   107  	mock := &MockPodsProvider{ctrl: ctrl}
   108  	mock.recorder = &MockPodsProviderMockRecorder{mock}
   109  	return mock
   110  }
   111  
   112  // EXPECT returns an object that allows the caller to indicate expected use.
   113  func (m *MockPodsProvider) EXPECT() *MockPodsProviderMockRecorder {
   114  	return m.recorder
   115  }
   116  
   117  // GetPodByName mocks base method.
   118  func (m *MockPodsProvider) GetPodByName(namespace, name string) (*v1.Pod, bool) {
   119  	m.ctrl.T.Helper()
   120  	ret := m.ctrl.Call(m, "GetPodByName", namespace, name)
   121  	ret0, _ := ret[0].(*v1.Pod)
   122  	ret1, _ := ret[1].(bool)
   123  	return ret0, ret1
   124  }
   125  
   126  // GetPodByName indicates an expected call of GetPodByName.
   127  func (mr *MockPodsProviderMockRecorder) GetPodByName(namespace, name interface{}) *gomock.Call {
   128  	mr.mock.ctrl.T.Helper()
   129  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPodByName", reflect.TypeOf((*MockPodsProvider)(nil).GetPodByName), namespace, name)
   130  }
   131  
   132  // GetPods mocks base method.
   133  func (m *MockPodsProvider) GetPods() []*v1.Pod {
   134  	m.ctrl.T.Helper()
   135  	ret := m.ctrl.Call(m, "GetPods")
   136  	ret0, _ := ret[0].([]*v1.Pod)
   137  	return ret0
   138  }
   139  
   140  // GetPods indicates an expected call of GetPods.
   141  func (mr *MockPodsProviderMockRecorder) GetPods() *gomock.Call {
   142  	mr.mock.ctrl.T.Helper()
   143  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPods", reflect.TypeOf((*MockPodsProvider)(nil).GetPods))
   144  }
   145  
   146  // MockCPUsProvider is a mock of CPUsProvider interface.
   147  type MockCPUsProvider struct {
   148  	ctrl     *gomock.Controller
   149  	recorder *MockCPUsProviderMockRecorder
   150  }
   151  
   152  // MockCPUsProviderMockRecorder is the mock recorder for MockCPUsProvider.
   153  type MockCPUsProviderMockRecorder struct {
   154  	mock *MockCPUsProvider
   155  }
   156  
   157  // NewMockCPUsProvider creates a new mock instance.
   158  func NewMockCPUsProvider(ctrl *gomock.Controller) *MockCPUsProvider {
   159  	mock := &MockCPUsProvider{ctrl: ctrl}
   160  	mock.recorder = &MockCPUsProviderMockRecorder{mock}
   161  	return mock
   162  }
   163  
   164  // EXPECT returns an object that allows the caller to indicate expected use.
   165  func (m *MockCPUsProvider) EXPECT() *MockCPUsProviderMockRecorder {
   166  	return m.recorder
   167  }
   168  
   169  // GetAllocatableCPUs mocks base method.
   170  func (m *MockCPUsProvider) GetAllocatableCPUs() []int64 {
   171  	m.ctrl.T.Helper()
   172  	ret := m.ctrl.Call(m, "GetAllocatableCPUs")
   173  	ret0, _ := ret[0].([]int64)
   174  	return ret0
   175  }
   176  
   177  // GetAllocatableCPUs indicates an expected call of GetAllocatableCPUs.
   178  func (mr *MockCPUsProviderMockRecorder) GetAllocatableCPUs() *gomock.Call {
   179  	mr.mock.ctrl.T.Helper()
   180  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllocatableCPUs", reflect.TypeOf((*MockCPUsProvider)(nil).GetAllocatableCPUs))
   181  }
   182  
   183  // GetCPUs mocks base method.
   184  func (m *MockCPUsProvider) GetCPUs(podUID, containerName string) []int64 {
   185  	m.ctrl.T.Helper()
   186  	ret := m.ctrl.Call(m, "GetCPUs", podUID, containerName)
   187  	ret0, _ := ret[0].([]int64)
   188  	return ret0
   189  }
   190  
   191  // GetCPUs indicates an expected call of GetCPUs.
   192  func (mr *MockCPUsProviderMockRecorder) GetCPUs(podUID, containerName interface{}) *gomock.Call {
   193  	mr.mock.ctrl.T.Helper()
   194  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCPUs", reflect.TypeOf((*MockCPUsProvider)(nil).GetCPUs), podUID, containerName)
   195  }
   196  
   197  // MockMemoryProvider is a mock of MemoryProvider interface.
   198  type MockMemoryProvider struct {
   199  	ctrl     *gomock.Controller
   200  	recorder *MockMemoryProviderMockRecorder
   201  }
   202  
   203  // MockMemoryProviderMockRecorder is the mock recorder for MockMemoryProvider.
   204  type MockMemoryProviderMockRecorder struct {
   205  	mock *MockMemoryProvider
   206  }
   207  
   208  // NewMockMemoryProvider creates a new mock instance.
   209  func NewMockMemoryProvider(ctrl *gomock.Controller) *MockMemoryProvider {
   210  	mock := &MockMemoryProvider{ctrl: ctrl}
   211  	mock.recorder = &MockMemoryProviderMockRecorder{mock}
   212  	return mock
   213  }
   214  
   215  // EXPECT returns an object that allows the caller to indicate expected use.
   216  func (m *MockMemoryProvider) EXPECT() *MockMemoryProviderMockRecorder {
   217  	return m.recorder
   218  }
   219  
   220  // GetAllocatableMemory mocks base method.
   221  func (m *MockMemoryProvider) GetAllocatableMemory() []*v10.ContainerMemory {
   222  	m.ctrl.T.Helper()
   223  	ret := m.ctrl.Call(m, "GetAllocatableMemory")
   224  	ret0, _ := ret[0].([]*v10.ContainerMemory)
   225  	return ret0
   226  }
   227  
   228  // GetAllocatableMemory indicates an expected call of GetAllocatableMemory.
   229  func (mr *MockMemoryProviderMockRecorder) GetAllocatableMemory() *gomock.Call {
   230  	mr.mock.ctrl.T.Helper()
   231  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetAllocatableMemory", reflect.TypeOf((*MockMemoryProvider)(nil).GetAllocatableMemory))
   232  }
   233  
   234  // GetMemory mocks base method.
   235  func (m *MockMemoryProvider) GetMemory(podUID, containerName string) []*v10.ContainerMemory {
   236  	m.ctrl.T.Helper()
   237  	ret := m.ctrl.Call(m, "GetMemory", podUID, containerName)
   238  	ret0, _ := ret[0].([]*v10.ContainerMemory)
   239  	return ret0
   240  }
   241  
   242  // GetMemory indicates an expected call of GetMemory.
   243  func (mr *MockMemoryProviderMockRecorder) GetMemory(podUID, containerName interface{}) *gomock.Call {
   244  	mr.mock.ctrl.T.Helper()
   245  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetMemory", reflect.TypeOf((*MockMemoryProvider)(nil).GetMemory), podUID, containerName)
   246  }
   247  
   248  // MockDynamicResourcesProvider is a mock of DynamicResourcesProvider interface.
   249  type MockDynamicResourcesProvider struct {
   250  	ctrl     *gomock.Controller
   251  	recorder *MockDynamicResourcesProviderMockRecorder
   252  }
   253  
   254  // MockDynamicResourcesProviderMockRecorder is the mock recorder for MockDynamicResourcesProvider.
   255  type MockDynamicResourcesProviderMockRecorder struct {
   256  	mock *MockDynamicResourcesProvider
   257  }
   258  
   259  // NewMockDynamicResourcesProvider creates a new mock instance.
   260  func NewMockDynamicResourcesProvider(ctrl *gomock.Controller) *MockDynamicResourcesProvider {
   261  	mock := &MockDynamicResourcesProvider{ctrl: ctrl}
   262  	mock.recorder = &MockDynamicResourcesProviderMockRecorder{mock}
   263  	return mock
   264  }
   265  
   266  // EXPECT returns an object that allows the caller to indicate expected use.
   267  func (m *MockDynamicResourcesProvider) EXPECT() *MockDynamicResourcesProviderMockRecorder {
   268  	return m.recorder
   269  }
   270  
   271  // GetDynamicResources mocks base method.
   272  func (m *MockDynamicResourcesProvider) GetDynamicResources(pod *v1.Pod, container *v1.Container) []*v10.DynamicResource {
   273  	m.ctrl.T.Helper()
   274  	ret := m.ctrl.Call(m, "GetDynamicResources", pod, container)
   275  	ret0, _ := ret[0].([]*v10.DynamicResource)
   276  	return ret0
   277  }
   278  
   279  // GetDynamicResources indicates an expected call of GetDynamicResources.
   280  func (mr *MockDynamicResourcesProviderMockRecorder) GetDynamicResources(pod, container interface{}) *gomock.Call {
   281  	mr.mock.ctrl.T.Helper()
   282  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDynamicResources", reflect.TypeOf((*MockDynamicResourcesProvider)(nil).GetDynamicResources), pod, container)
   283  }
   284
View as plain text