// Code generated by MockGen. DO NOT EDIT.
// Source: edge-infra.dev/pkg/sds/devices/k8s/device-plugins (interfaces: Plugin)

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

import (
	context "context"
	reflect "reflect"

	v1 "edge-infra.dev/pkg/sds/devices/k8s/apis/v1"
	gomock "github.com/golang/mock/gomock"
	v1beta1 "k8s.io/kubelet/pkg/apis/deviceplugin/v1beta1"
)

// MockPlugin is a mock of Plugin interface.
type MockPlugin struct {
	ctrl     *gomock.Controller
	recorder *MockPluginMockRecorder
}

// MockPluginMockRecorder is the mock recorder for MockPlugin.
type MockPluginMockRecorder struct {
	mock *MockPlugin
}

// NewMockPlugin creates a new mock instance.
func NewMockPlugin(ctrl *gomock.Controller) *MockPlugin {
	mock := &MockPlugin{ctrl: ctrl}
	mock.recorder = &MockPluginMockRecorder{mock}
	return mock
}

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

// Allocate mocks base method.
func (m *MockPlugin) Allocate(arg0 context.Context, arg1 *v1beta1.AllocateRequest) (*v1beta1.AllocateResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "Allocate", arg0, arg1)
	ret0, _ := ret[0].(*v1beta1.AllocateResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// Allocate indicates an expected call of Allocate.
func (mr *MockPluginMockRecorder) Allocate(arg0, arg1 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Allocate", reflect.TypeOf((*MockPlugin)(nil).Allocate), arg0, arg1)
}

// GetDevicePluginOptions mocks base method.
func (m *MockPlugin) GetDevicePluginOptions(arg0 context.Context, arg1 *v1beta1.Empty) (*v1beta1.DevicePluginOptions, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetDevicePluginOptions", arg0, arg1)
	ret0, _ := ret[0].(*v1beta1.DevicePluginOptions)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetDevicePluginOptions indicates an expected call of GetDevicePluginOptions.
func (mr *MockPluginMockRecorder) GetDevicePluginOptions(arg0, arg1 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetDevicePluginOptions", reflect.TypeOf((*MockPlugin)(nil).GetDevicePluginOptions), arg0, arg1)
}

// GetPreferredAllocation mocks base method.
func (m *MockPlugin) GetPreferredAllocation(arg0 context.Context, arg1 *v1beta1.PreferredAllocationRequest) (*v1beta1.PreferredAllocationResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "GetPreferredAllocation", arg0, arg1)
	ret0, _ := ret[0].(*v1beta1.PreferredAllocationResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// GetPreferredAllocation indicates an expected call of GetPreferredAllocation.
func (mr *MockPluginMockRecorder) GetPreferredAllocation(arg0, arg1 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "GetPreferredAllocation", reflect.TypeOf((*MockPlugin)(nil).GetPreferredAllocation), arg0, arg1)
}

// IsRunning mocks base method.
func (m *MockPlugin) IsRunning() bool {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "IsRunning")
	ret0, _ := ret[0].(bool)
	return ret0
}

// IsRunning indicates an expected call of IsRunning.
func (mr *MockPluginMockRecorder) IsRunning() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "IsRunning", reflect.TypeOf((*MockPlugin)(nil).IsRunning))
}

// ListAndWatch mocks base method.
func (m *MockPlugin) ListAndWatch(arg0 *v1beta1.Empty, arg1 v1beta1.DevicePlugin_ListAndWatchServer) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ListAndWatch", arg0, arg1)
	ret0, _ := ret[0].(error)
	return ret0
}

// ListAndWatch indicates an expected call of ListAndWatch.
func (mr *MockPluginMockRecorder) ListAndWatch(arg0, arg1 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ListAndWatch", reflect.TypeOf((*MockPlugin)(nil).ListAndWatch), arg0, arg1)
}

// PreStartContainer mocks base method.
func (m *MockPlugin) PreStartContainer(arg0 context.Context, arg1 *v1beta1.PreStartContainerRequest) (*v1beta1.PreStartContainerResponse, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "PreStartContainer", arg0, arg1)
	ret0, _ := ret[0].(*v1beta1.PreStartContainerResponse)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// PreStartContainer indicates an expected call of PreStartContainer.
func (mr *MockPluginMockRecorder) PreStartContainer(arg0, arg1 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "PreStartContainer", reflect.TypeOf((*MockPlugin)(nil).PreStartContainer), arg0, arg1)
}

// Remove mocks base method.
func (m *MockPlugin) Remove() {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "Remove")
}

// Remove indicates an expected call of Remove.
func (mr *MockPluginMockRecorder) Remove() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Remove", reflect.TypeOf((*MockPlugin)(nil).Remove))
}

// Run mocks base method.
func (m *MockPlugin) Run(arg0 context.Context) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "Run", arg0)
}

// Run indicates an expected call of Run.
func (mr *MockPluginMockRecorder) Run(arg0 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Run", reflect.TypeOf((*MockPlugin)(nil).Run), arg0)
}

// Stop mocks base method.
func (m *MockPlugin) Stop() {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "Stop")
}

// Stop indicates an expected call of Stop.
func (mr *MockPluginMockRecorder) Stop() *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Stop", reflect.TypeOf((*MockPlugin)(nil).Stop))
}

// Update mocks base method.
func (m *MockPlugin) Update(arg0 *v1.DeviceClass) {
	m.ctrl.T.Helper()
	m.ctrl.Call(m, "Update", arg0)
}

// Update indicates an expected call of Update.
func (mr *MockPluginMockRecorder) Update(arg0 interface{}) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Update", reflect.TypeOf((*MockPlugin)(nil).Update), arg0)
}