mock_stats_provider.go

Documentation: k8s.io/kubernetes/pkg/kubelet/server/stats/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: handler.go
    19  
    20  // Package testing is a generated GoMock package.
    21  package testing
    22  
    23  import (
    24  	context "context"
    25  	reflect "reflect"
    26  
    27  	gomock "github.com/golang/mock/gomock"
    28  	v1 "github.com/google/cadvisor/info/v1"
    29  	v2 "github.com/google/cadvisor/info/v2"
    30  	v10 "k8s.io/api/core/v1"
    31  	types "k8s.io/apimachinery/pkg/types"
    32  	v1alpha1 "k8s.io/kubelet/pkg/apis/stats/v1alpha1"
    33  	cm "k8s.io/kubernetes/pkg/kubelet/cm"
    34  	volume "k8s.io/kubernetes/pkg/volume"
    35  )
    36  
    37  // MockProvider is a mock of Provider interface.
    38  type MockProvider struct {
    39  	ctrl     *gomock.Controller
    40  	recorder *MockProviderMockRecorder
    41  }
    42  
    43  // MockProviderMockRecorder is the mock recorder for MockProvider.
    44  type MockProviderMockRecorder struct {
    45  	mock *MockProvider
    46  }
    47  
    48  // NewMockProvider creates a new mock instance.
    49  func NewMockProvider(ctrl *gomock.Controller) *MockProvider {
    50  	mock := &MockProvider{ctrl: ctrl}
    51  	mock.recorder = &MockProviderMockRecorder{mock}
    52  	return mock
    53  }
    54  
    55  // EXPECT returns an object that allows the caller to indicate expected use.
    56  func (m *MockProvider) EXPECT() *MockProviderMockRecorder {
    57  	return m.recorder
    58  }
    59  
    60  // GetCgroupCPUAndMemoryStats mocks base method.
    61  func (m *MockProvider) GetCgroupCPUAndMemoryStats(cgroupName string, updateStats bool) (*v1alpha1.ContainerStats, error) {
    62  	m.ctrl.T.Helper()
    63  	ret := m.ctrl.Call(m, "GetCgroupCPUAndMemoryStats", cgroupName, updateStats)
    64  	ret0, _ := ret[0].(*v1alpha1.ContainerStats)
    65  	ret1, _ := ret[1].(error)
    66  	return ret0, ret1
    67  }
    68  
    69  // GetCgroupCPUAndMemoryStats indicates an expected call of GetCgroupCPUAndMemoryStats.
    70  func (mr *MockProviderMockRecorder) GetCgroupCPUAndMemoryStats(cgroupName, updateStats interface{}) *gomock.Call {
    71  	mr.mock.ctrl.T.Helper()
    72  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCgroupCPUAndMemoryStats", reflect.TypeOf((*MockProvider)(nil).GetCgroupCPUAndMemoryStats), cgroupName, updateStats)
    73  }
    74  
    75  // GetCgroupStats mocks base method.
    76  func (m *MockProvider) GetCgroupStats(cgroupName string, updateStats bool) (*v1alpha1.ContainerStats, *v1alpha1.NetworkStats, error) {
    77  	m.ctrl.T.Helper()
    78  	ret := m.ctrl.Call(m, "GetCgroupStats", cgroupName, updateStats)
    79  	ret0, _ := ret[0].(*v1alpha1.ContainerStats)
    80  	ret1, _ := ret[1].(*v1alpha1.NetworkStats)
    81  	ret2, _ := ret[2].(error)
    82  	return ret0, ret1, ret2
    83  }
    84  
    85  // GetCgroupStats indicates an expected call of GetCgroupStats.
    86  func (mr *MockProviderMockRecorder) GetCgroupStats(cgroupName, updateStats interface{}) *gomock.Call {
    87  	mr.mock.ctrl.T.Helper()
    88  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetCgroupStats", reflect.TypeOf((*MockProvider)(nil).GetCgroupStats), cgroupName, updateStats)
    89  }
    90  
    91  // GetNode mocks base method.
    92  func (m *MockProvider) GetNode() (*v10.Node, error) {
    93  	m.ctrl.T.Helper()
    94  	ret := m.ctrl.Call(m, "GetNode")
    95  	ret0, _ := ret[0].(*v10.Node)
    96  	ret1, _ := ret[1].(error)
    97  	return ret0, ret1
    98  }
    99  
   100  // GetNode indicates an expected call of GetNode.
   101  func (mr *MockProviderMockRecorder) GetNode() *gomock.Call {
   102  	mr.mock.ctrl.T.Helper()
   103  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNode", reflect.TypeOf((*MockProvider)(nil).GetNode))
   104  }
   105  
   106  // GetNodeConfig mocks base method.
   107  func (m *MockProvider) GetNodeConfig() cm.NodeConfig {
   108  	m.ctrl.T.Helper()
   109  	ret := m.ctrl.Call(m, "GetNodeConfig")
   110  	ret0, _ := ret[0].(cm.NodeConfig)
   111  	return ret0
   112  }
   113  
   114  // GetNodeConfig indicates an expected call of GetNodeConfig.
   115  func (mr *MockProviderMockRecorder) GetNodeConfig() *gomock.Call {
   116  	mr.mock.ctrl.T.Helper()
   117  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNodeConfig", reflect.TypeOf((*MockProvider)(nil).GetNodeConfig))
   118  }
   119  
   120  // GetPodByCgroupfs mocks base method.
   121  func (m *MockProvider) GetPodByCgroupfs(cgroupfs string) (*v10.Pod, bool) {
   122  	m.ctrl.T.Helper()
   123  	ret := m.ctrl.Call(m, "GetPodByCgroupfs", cgroupfs)
   124  	ret0, _ := ret[0].(*v10.Pod)
   125  	ret1, _ := ret[1].(bool)
   126  	return ret0, ret1
   127  }
   128  
   129  // GetPodByCgroupfs indicates an expected call of GetPodByCgroupfs.
   130  func (mr *MockProviderMockRecorder) GetPodByCgroupfs(cgroupfs interface{}) *gomock.Call {
   131  	mr.mock.ctrl.T.Helper()
   132  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPodByCgroupfs", reflect.TypeOf((*MockProvider)(nil).GetPodByCgroupfs), cgroupfs)
   133  }
   134  
   135  // GetPodByName mocks base method.
   136  func (m *MockProvider) GetPodByName(namespace, name string) (*v10.Pod, bool) {
   137  	m.ctrl.T.Helper()
   138  	ret := m.ctrl.Call(m, "GetPodByName", namespace, name)
   139  	ret0, _ := ret[0].(*v10.Pod)
   140  	ret1, _ := ret[1].(bool)
   141  	return ret0, ret1
   142  }
   143  
   144  // GetPodByName indicates an expected call of GetPodByName.
   145  func (mr *MockProviderMockRecorder) GetPodByName(namespace, name interface{}) *gomock.Call {
   146  	mr.mock.ctrl.T.Helper()
   147  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPodByName", reflect.TypeOf((*MockProvider)(nil).GetPodByName), namespace, name)
   148  }
   149  
   150  // GetPodCgroupRoot mocks base method.
   151  func (m *MockProvider) GetPodCgroupRoot() string {
   152  	m.ctrl.T.Helper()
   153  	ret := m.ctrl.Call(m, "GetPodCgroupRoot")
   154  	ret0, _ := ret[0].(string)
   155  	return ret0
   156  }
   157  
   158  // GetPodCgroupRoot indicates an expected call of GetPodCgroupRoot.
   159  func (mr *MockProviderMockRecorder) GetPodCgroupRoot() *gomock.Call {
   160  	mr.mock.ctrl.T.Helper()
   161  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPodCgroupRoot", reflect.TypeOf((*MockProvider)(nil).GetPodCgroupRoot))
   162  }
   163  
   164  // GetPods mocks base method.
   165  func (m *MockProvider) GetPods() []*v10.Pod {
   166  	m.ctrl.T.Helper()
   167  	ret := m.ctrl.Call(m, "GetPods")
   168  	ret0, _ := ret[0].([]*v10.Pod)
   169  	return ret0
   170  }
   171  
   172  // GetPods indicates an expected call of GetPods.
   173  func (mr *MockProviderMockRecorder) GetPods() *gomock.Call {
   174  	mr.mock.ctrl.T.Helper()
   175  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPods", reflect.TypeOf((*MockProvider)(nil).GetPods))
   176  }
   177  
   178  // GetRequestedContainersInfo mocks base method.
   179  func (m *MockProvider) GetRequestedContainersInfo(containerName string, options v2.RequestOptions) (map[string]*v1.ContainerInfo, error) {
   180  	m.ctrl.T.Helper()
   181  	ret := m.ctrl.Call(m, "GetRequestedContainersInfo", containerName, options)
   182  	ret0, _ := ret[0].(map[string]*v1.ContainerInfo)
   183  	ret1, _ := ret[1].(error)
   184  	return ret0, ret1
   185  }
   186  
   187  // GetRequestedContainersInfo indicates an expected call of GetRequestedContainersInfo.
   188  func (mr *MockProviderMockRecorder) GetRequestedContainersInfo(containerName, options interface{}) *gomock.Call {
   189  	mr.mock.ctrl.T.Helper()
   190  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetRequestedContainersInfo", reflect.TypeOf((*MockProvider)(nil).GetRequestedContainersInfo), containerName, options)
   191  }
   192  
   193  // ImageFsStats mocks base method.
   194  func (m *MockProvider) ImageFsStats(ctx context.Context) (*v1alpha1.FsStats, *v1alpha1.FsStats, error) {
   195  	m.ctrl.T.Helper()
   196  	ret := m.ctrl.Call(m, "ImageFsStats", ctx)
   197  	ret0, _ := ret[0].(*v1alpha1.FsStats)
   198  	ret1, _ := ret[1].(*v1alpha1.FsStats)
   199  	ret2, _ := ret[2].(error)
   200  	return ret0, ret1, ret2
   201  }
   202  
   203  // ImageFsStats indicates an expected call of ImageFsStats.
   204  func (mr *MockProviderMockRecorder) ImageFsStats(ctx interface{}) *gomock.Call {
   205  	mr.mock.ctrl.T.Helper()
   206  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ImageFsStats", reflect.TypeOf((*MockProvider)(nil).ImageFsStats), ctx)
   207  }
   208  
   209  // ListBlockVolumesForPod mocks base method.
   210  func (m *MockProvider) ListBlockVolumesForPod(podUID types.UID) (map[string]volume.BlockVolume, bool) {
   211  	m.ctrl.T.Helper()
   212  	ret := m.ctrl.Call(m, "ListBlockVolumesForPod", podUID)
   213  	ret0, _ := ret[0].(map[string]volume.BlockVolume)
   214  	ret1, _ := ret[1].(bool)
   215  	return ret0, ret1
   216  }
   217  
   218  // ListBlockVolumesForPod indicates an expected call of ListBlockVolumesForPod.
   219  func (mr *MockProviderMockRecorder) ListBlockVolumesForPod(podUID interface{}) *gomock.Call {
   220  	mr.mock.ctrl.T.Helper()
   221  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListBlockVolumesForPod", reflect.TypeOf((*MockProvider)(nil).ListBlockVolumesForPod), podUID)
   222  }
   223  
   224  // ListPodCPUAndMemoryStats mocks base method.
   225  func (m *MockProvider) ListPodCPUAndMemoryStats(ctx context.Context) ([]v1alpha1.PodStats, error) {
   226  	m.ctrl.T.Helper()
   227  	ret := m.ctrl.Call(m, "ListPodCPUAndMemoryStats", ctx)
   228  	ret0, _ := ret[0].([]v1alpha1.PodStats)
   229  	ret1, _ := ret[1].(error)
   230  	return ret0, ret1
   231  }
   232  
   233  // ListPodCPUAndMemoryStats indicates an expected call of ListPodCPUAndMemoryStats.
   234  func (mr *MockProviderMockRecorder) ListPodCPUAndMemoryStats(ctx interface{}) *gomock.Call {
   235  	mr.mock.ctrl.T.Helper()
   236  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPodCPUAndMemoryStats", reflect.TypeOf((*MockProvider)(nil).ListPodCPUAndMemoryStats), ctx)
   237  }
   238  
   239  // ListPodStats mocks base method.
   240  func (m *MockProvider) ListPodStats(ctx context.Context) ([]v1alpha1.PodStats, error) {
   241  	m.ctrl.T.Helper()
   242  	ret := m.ctrl.Call(m, "ListPodStats", ctx)
   243  	ret0, _ := ret[0].([]v1alpha1.PodStats)
   244  	ret1, _ := ret[1].(error)
   245  	return ret0, ret1
   246  }
   247  
   248  // ListPodStats indicates an expected call of ListPodStats.
   249  func (mr *MockProviderMockRecorder) ListPodStats(ctx interface{}) *gomock.Call {
   250  	mr.mock.ctrl.T.Helper()
   251  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPodStats", reflect.TypeOf((*MockProvider)(nil).ListPodStats), ctx)
   252  }
   253  
   254  // ListPodStatsAndUpdateCPUNanoCoreUsage mocks base method.
   255  func (m *MockProvider) ListPodStatsAndUpdateCPUNanoCoreUsage(ctx context.Context) ([]v1alpha1.PodStats, error) {
   256  	m.ctrl.T.Helper()
   257  	ret := m.ctrl.Call(m, "ListPodStatsAndUpdateCPUNanoCoreUsage", ctx)
   258  	ret0, _ := ret[0].([]v1alpha1.PodStats)
   259  	ret1, _ := ret[1].(error)
   260  	return ret0, ret1
   261  }
   262  
   263  // ListPodStatsAndUpdateCPUNanoCoreUsage indicates an expected call of ListPodStatsAndUpdateCPUNanoCoreUsage.
   264  func (mr *MockProviderMockRecorder) ListPodStatsAndUpdateCPUNanoCoreUsage(ctx interface{}) *gomock.Call {
   265  	mr.mock.ctrl.T.Helper()
   266  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListPodStatsAndUpdateCPUNanoCoreUsage", reflect.TypeOf((*MockProvider)(nil).ListPodStatsAndUpdateCPUNanoCoreUsage), ctx)
   267  }
   268  
   269  // ListVolumesForPod mocks base method.
   270  func (m *MockProvider) ListVolumesForPod(podUID types.UID) (map[string]volume.Volume, bool) {
   271  	m.ctrl.T.Helper()
   272  	ret := m.ctrl.Call(m, "ListVolumesForPod", podUID)
   273  	ret0, _ := ret[0].(map[string]volume.Volume)
   274  	ret1, _ := ret[1].(bool)
   275  	return ret0, ret1
   276  }
   277  
   278  // ListVolumesForPod indicates an expected call of ListVolumesForPod.
   279  func (mr *MockProviderMockRecorder) ListVolumesForPod(podUID interface{}) *gomock.Call {
   280  	mr.mock.ctrl.T.Helper()
   281  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListVolumesForPod", reflect.TypeOf((*MockProvider)(nil).ListVolumesForPod), podUID)
   282  }
   283  
   284  // RlimitStats mocks base method.
   285  func (m *MockProvider) RlimitStats() (*v1alpha1.RlimitStats, error) {
   286  	m.ctrl.T.Helper()
   287  	ret := m.ctrl.Call(m, "RlimitStats")
   288  	ret0, _ := ret[0].(*v1alpha1.RlimitStats)
   289  	ret1, _ := ret[1].(error)
   290  	return ret0, ret1
   291  }
   292  
   293  // RlimitStats indicates an expected call of RlimitStats.
   294  func (mr *MockProviderMockRecorder) RlimitStats() *gomock.Call {
   295  	mr.mock.ctrl.T.Helper()
   296  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RlimitStats", reflect.TypeOf((*MockProvider)(nil).RlimitStats))
   297  }
   298  
   299  // RootFsStats mocks base method.
   300  func (m *MockProvider) RootFsStats() (*v1alpha1.FsStats, error) {
   301  	m.ctrl.T.Helper()
   302  	ret := m.ctrl.Call(m, "RootFsStats")
   303  	ret0, _ := ret[0].(*v1alpha1.FsStats)
   304  	ret1, _ := ret[1].(error)
   305  	return ret0, ret1
   306  }
   307  
   308  // RootFsStats indicates an expected call of RootFsStats.
   309  func (mr *MockProviderMockRecorder) RootFsStats() *gomock.Call {
   310  	mr.mock.ctrl.T.Helper()
   311  	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RootFsStats", reflect.TypeOf((*MockProvider)(nil).RootFsStats))
   312  }
   313
View as plain text