kubernetes_client_mock.go

// Code generated by MockGen. DO NOT EDIT.
// Source: kubernetes_client.go

// Package main is a generated GoMock package.
package main

import (
	context "context"
	gomock "github.com/golang/mock/gomock"
	v1 "k8s.io/api/core/v1"
	reflect "reflect"
)

// MockKubernetesClient is a mock of KubernetesClient interface
type MockKubernetesClient struct {
	ctrl     *gomock.Controller
	recorder *MockKubernetesClientMockRecorder
}

// MockKubernetesClientMockRecorder is the mock recorder for MockKubernetesClient
type MockKubernetesClientMockRecorder struct {
	mock *MockKubernetesClient
}

// NewMockKubernetesClient creates a new mock instance
func NewMockKubernetesClient(ctrl *gomock.Controller) *MockKubernetesClient {
	mock := &MockKubernetesClient{ctrl: ctrl}
	mock.recorder = &MockKubernetesClientMockRecorder{mock}
	return mock
}

// EXPECT returns an object that allows the caller to indicate expected use
func (m *MockKubernetesClient) EXPECT() *MockKubernetesClientMockRecorder {
	return m.recorder
}

// DrainNode mocks base method
func (m *MockKubernetesClient) DrainNode(ctx context.Context, nodeName string, drainTimeout int) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DrainNode", ctx, nodeName, drainTimeout)
	ret0, _ := ret[0].(error)
	return ret0
}

// DrainNode indicates an expected call of DrainNode
func (mr *MockKubernetesClientMockRecorder) DrainNode(ctx, nodeName, drainTimeout interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DrainNode", reflect.TypeOf((*MockKubernetesClient)(nil).DrainNode), ctx, nodeName, drainTimeout)
}

// DrainKubeDNSFromNode mocks base method
func (m *MockKubernetesClient) DrainKubeDNSFromNode(ctx context.Context, nodeName string, drainTimeout int) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DrainKubeDNSFromNode", ctx, nodeName, drainTimeout)
	ret0, _ := ret[0].(error)
	return ret0
}

// DrainKubeDNSFromNode indicates an expected call of DrainKubeDNSFromNode
func (mr *MockKubernetesClientMockRecorder) DrainKubeDNSFromNode(ctx, nodeName, drainTimeout interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DrainKubeDNSFromNode", reflect.TypeOf((*MockKubernetesClient)(nil).DrainKubeDNSFromNode), ctx, nodeName, drainTimeout)
}

// GetNode mocks base method
func (m *MockKubernetesClient) GetNode(ctx context.Context, nodeName string) (*v1.Node, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetNode", ctx, nodeName)
	ret0, _ := ret[0].(*v1.Node)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetNode indicates an expected call of GetNode
func (mr *MockKubernetesClientMockRecorder) GetNode(ctx, nodeName interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetNode", reflect.TypeOf((*MockKubernetesClient)(nil).GetNode), ctx, nodeName)
}

// DeleteNode mocks base method
func (m *MockKubernetesClient) DeleteNode(ctx context.Context, nodeName string) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "DeleteNode", ctx, nodeName)
	ret0, _ := ret[0].(error)
	return ret0
}

// DeleteNode indicates an expected call of DeleteNode
func (mr *MockKubernetesClientMockRecorder) DeleteNode(ctx, nodeName interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "DeleteNode", reflect.TypeOf((*MockKubernetesClient)(nil).DeleteNode), ctx, nodeName)
}

// GetPreemptibleNodes mocks base method
func (m *MockKubernetesClient) GetPreemptibleNodes(ctx context.Context, filters map[string]string) (*v1.NodeList, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPreemptibleNodes", ctx, filters)
	ret0, _ := ret[0].(*v1.NodeList)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPreemptibleNodes indicates an expected call of GetPreemptibleNodes
func (mr *MockKubernetesClientMockRecorder) GetPreemptibleNodes(ctx, filters interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPreemptibleNodes", reflect.TypeOf((*MockKubernetesClient)(nil).GetPreemptibleNodes), ctx, filters)
}

// GetProjectIdAndZoneFromNode mocks base method
func (m *MockKubernetesClient) GetProjectIdAndZoneFromNode(ctx context.Context, nodeName string) (string, string, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetProjectIdAndZoneFromNode", ctx, nodeName)
	ret0, _ := ret[0].(string)
	ret1, _ := ret[1].(string)
	ret2, _ := ret[2].(error)
	return ret0, ret1, ret2
}

// GetProjectIdAndZoneFromNode indicates an expected call of GetProjectIdAndZoneFromNode
func (mr *MockKubernetesClientMockRecorder) GetProjectIdAndZoneFromNode(ctx, nodeName interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetProjectIdAndZoneFromNode", reflect.TypeOf((*MockKubernetesClient)(nil).GetProjectIdAndZoneFromNode), ctx, nodeName)
}

// SetNodeAnnotation mocks base method
func (m *MockKubernetesClient) SetNodeAnnotation(ctx context.Context, nodeName, key, value string) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "SetNodeAnnotation", ctx, nodeName, key, value)
	ret0, _ := ret[0].(error)
	return ret0
}

// SetNodeAnnotation indicates an expected call of SetNodeAnnotation
func (mr *MockKubernetesClientMockRecorder) SetNodeAnnotation(ctx, nodeName, key, value interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetNodeAnnotation", reflect.TypeOf((*MockKubernetesClient)(nil).SetNodeAnnotation), ctx, nodeName, key, value)
}

// SetUnschedulableState mocks base method
func (m *MockKubernetesClient) SetUnschedulableState(ctx context.Context, nodeName string, unschedulable bool) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "SetUnschedulableState", ctx, nodeName, unschedulable)
	ret0, _ := ret[0].(error)
	return ret0
}

// SetUnschedulableState indicates an expected call of SetUnschedulableState
func (mr *MockKubernetesClientMockRecorder) SetUnschedulableState(ctx, nodeName, unschedulable interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SetUnschedulableState", reflect.TypeOf((*MockKubernetesClient)(nil).SetUnschedulableState), ctx, nodeName, unschedulable)
}