mirror of https://github.com/k3s-io/k3s
157 lines
6.8 KiB
Go
157 lines
6.8 KiB
Go
// +build !providerless
|
|
|
|
/*
|
|
Copyright 2019 The Kubernetes Authors.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
package mockvmssclient
|
|
|
|
import (
|
|
context "context"
|
|
http "net/http"
|
|
reflect "reflect"
|
|
|
|
compute "github.com/Azure/azure-sdk-for-go/services/compute/mgmt/2019-12-01/compute"
|
|
azure "github.com/Azure/go-autorest/autorest/azure"
|
|
gomock "github.com/golang/mock/gomock"
|
|
retry "k8s.io/legacy-cloud-providers/azure/retry"
|
|
)
|
|
|
|
// MockInterface is a mock of Interface interface
|
|
type MockInterface struct {
|
|
ctrl *gomock.Controller
|
|
recorder *MockInterfaceMockRecorder
|
|
}
|
|
|
|
// MockInterfaceMockRecorder is the mock recorder for MockInterface
|
|
type MockInterfaceMockRecorder struct {
|
|
mock *MockInterface
|
|
}
|
|
|
|
// NewMockInterface creates a new mock instance
|
|
func NewMockInterface(ctrl *gomock.Controller) *MockInterface {
|
|
mock := &MockInterface{ctrl: ctrl}
|
|
mock.recorder = &MockInterfaceMockRecorder{mock}
|
|
return mock
|
|
}
|
|
|
|
// EXPECT returns an object that allows the caller to indicate expected use
|
|
func (m *MockInterface) EXPECT() *MockInterfaceMockRecorder {
|
|
return m.recorder
|
|
}
|
|
|
|
// Get mocks base method
|
|
func (m *MockInterface) Get(ctx context.Context, resourceGroupName, VMScaleSetName string) (compute.VirtualMachineScaleSet, *retry.Error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "Get", ctx, resourceGroupName, VMScaleSetName)
|
|
ret0, _ := ret[0].(compute.VirtualMachineScaleSet)
|
|
ret1, _ := ret[1].(*retry.Error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// Get indicates an expected call of Get
|
|
func (mr *MockInterfaceMockRecorder) Get(ctx, resourceGroupName, VMScaleSetName interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Get", reflect.TypeOf((*MockInterface)(nil).Get), ctx, resourceGroupName, VMScaleSetName)
|
|
}
|
|
|
|
// List mocks base method
|
|
func (m *MockInterface) List(ctx context.Context, resourceGroupName string) ([]compute.VirtualMachineScaleSet, *retry.Error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "List", ctx, resourceGroupName)
|
|
ret0, _ := ret[0].([]compute.VirtualMachineScaleSet)
|
|
ret1, _ := ret[1].(*retry.Error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// List indicates an expected call of List
|
|
func (mr *MockInterfaceMockRecorder) List(ctx, resourceGroupName interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "List", reflect.TypeOf((*MockInterface)(nil).List), ctx, resourceGroupName)
|
|
}
|
|
|
|
// CreateOrUpdate mocks base method
|
|
func (m *MockInterface) CreateOrUpdate(ctx context.Context, resourceGroupName, VMScaleSetName string, parameters compute.VirtualMachineScaleSet) *retry.Error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateOrUpdate", ctx, resourceGroupName, VMScaleSetName, parameters)
|
|
ret0, _ := ret[0].(*retry.Error)
|
|
return ret0
|
|
}
|
|
|
|
// CreateOrUpdate indicates an expected call of CreateOrUpdate
|
|
func (mr *MockInterfaceMockRecorder) CreateOrUpdate(ctx, resourceGroupName, VMScaleSetName, parameters interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrUpdate", reflect.TypeOf((*MockInterface)(nil).CreateOrUpdate), ctx, resourceGroupName, VMScaleSetName, parameters)
|
|
}
|
|
|
|
// CreateOrUpdateAsync mocks base method
|
|
func (m *MockInterface) CreateOrUpdateAsync(ctx context.Context, resourceGroupName, VMScaleSetName string, parameters compute.VirtualMachineScaleSet) (*azure.Future, *retry.Error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "CreateOrUpdateAsync", ctx, resourceGroupName, VMScaleSetName, parameters)
|
|
ret0, _ := ret[0].(*azure.Future)
|
|
ret1, _ := ret[1].(*retry.Error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// CreateOrUpdateAsync indicates an expected call of CreateOrUpdateAsync
|
|
func (mr *MockInterfaceMockRecorder) CreateOrUpdateAsync(ctx, resourceGroupName, VMScaleSetName, parameters interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "CreateOrUpdateAsync", reflect.TypeOf((*MockInterface)(nil).CreateOrUpdateAsync), ctx, resourceGroupName, VMScaleSetName, parameters)
|
|
}
|
|
|
|
// WaitForAsyncOperationResult mocks base method
|
|
func (m *MockInterface) WaitForAsyncOperationResult(ctx context.Context, future *azure.Future) (*http.Response, error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "WaitForAsyncOperationResult", ctx, future)
|
|
ret0, _ := ret[0].(*http.Response)
|
|
ret1, _ := ret[1].(error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// WaitForAsyncOperationResult indicates an expected call of WaitForAsyncOperationResult
|
|
func (mr *MockInterfaceMockRecorder) WaitForAsyncOperationResult(ctx, future interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "WaitForAsyncOperationResult", reflect.TypeOf((*MockInterface)(nil).WaitForAsyncOperationResult), ctx, future)
|
|
}
|
|
|
|
// DeleteInstances mocks base method
|
|
func (m *MockInterface) DeleteInstances(ctx context.Context, resourceGroupName, vmScaleSetName string, vmInstanceIDs compute.VirtualMachineScaleSetVMInstanceRequiredIDs) *retry.Error {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeleteInstances", ctx, resourceGroupName, vmScaleSetName, vmInstanceIDs)
|
|
ret0, _ := ret[0].(*retry.Error)
|
|
return ret0
|
|
}
|
|
|
|
// DeleteInstances indicates an expected call of DeleteInstances
|
|
func (mr *MockInterfaceMockRecorder) DeleteInstances(ctx, resourceGroupName, vmScaleSetName, vmInstanceIDs interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteInstances", reflect.TypeOf((*MockInterface)(nil).DeleteInstances), ctx, resourceGroupName, vmScaleSetName, vmInstanceIDs)
|
|
}
|
|
|
|
// DeleteInstancesAsync mocks base method
|
|
func (m *MockInterface) DeleteInstancesAsync(ctx context.Context, resourceGroupName, VMScaleSetName string, vmInstanceIDs compute.VirtualMachineScaleSetVMInstanceRequiredIDs) (*azure.Future, *retry.Error) {
|
|
m.ctrl.T.Helper()
|
|
ret := m.ctrl.Call(m, "DeleteInstancesAsync", ctx, resourceGroupName, VMScaleSetName, vmInstanceIDs)
|
|
ret0, _ := ret[0].(*azure.Future)
|
|
ret1, _ := ret[1].(*retry.Error)
|
|
return ret0, ret1
|
|
}
|
|
|
|
// DeleteInstancesAsync indicates an expected call of DeleteInstancesAsync
|
|
func (mr *MockInterfaceMockRecorder) DeleteInstancesAsync(ctx, resourceGroupName, VMScaleSetName, vmInstanceIDs interface{}) *gomock.Call {
|
|
mr.mock.ctrl.T.Helper()
|
|
return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteInstancesAsync", reflect.TypeOf((*MockInterface)(nil).DeleteInstancesAsync), ctx, resourceGroupName, VMScaleSetName, vmInstanceIDs)
|
|
}
|