resource_reservation_mock.go

// Code generated by MockGen. DO NOT EDIT.
// Source: pkg/binder/binding/resourcereservation/resource_reservation.go
//
// Generated by this command:
//
//	mockgen -source=pkg/binder/binding/resourcereservation/resource_reservation.go -destination=pkg/binder/binding/resourcereservation/mock/resource_reservation_mock.go -package=mock_resourcereservation
//

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

import (
	context "context"
	reflect "reflect"

	v1alpha2 "github.com/kai-scheduler/KAI-scheduler/pkg/apis/scheduling/v1alpha2"
	gomock "go.uber.org/mock/gomock"
	v1 "k8s.io/api/core/v1"
)

// MockInterface is a mock of Interface interface.
type MockInterface struct {
	ctrl     *gomock.Controller
	recorder *MockInterfaceMockRecorder
	isgomock struct{}
}

// 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
}

// RemovePodGpuGroupsConnection mocks base method.
func (m *MockInterface) RemovePodGpuGroupsConnection(ctx context.Context, pod *v1.Pod) error {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "RemovePodGpuGroupsConnection", ctx, pod)
	ret0, _ := ret[0].(error)
	return ret0
}

// RemovePodGpuGroupsConnection indicates an expected call of RemovePodGpuGroupsConnection.
func (mr *MockInterfaceMockRecorder) RemovePodGpuGroupsConnection(ctx, pod any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "RemovePodGpuGroupsConnection", reflect.TypeOf((*MockInterface)(nil).RemovePodGpuGroupsConnection), ctx, pod)
}

// ReserveGpuDevice mocks base method.
func (m *MockInterface) ReserveGpuDevice(ctx context.Context, pod *v1.Pod, nodeName string, fractionalGpuGroup v1alpha2.FractionalGpuGroup) (string, error) {
	m.ctrl.T.Helper()
	ret := m.ctrl.Call(m, "ReserveGpuDevice", ctx, pod, nodeName, fractionalGpuGroup)
	ret0, _ := ret[0].(string)
	ret1, _ := ret[1].(error)
	return ret0, ret1
}

// ReserveGpuDevice indicates an expected call of ReserveGpuDevice.
func (mr *MockInterfaceMockRecorder) ReserveGpuDevice(ctx, pod, nodeName, fractionalGpuGroup any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "ReserveGpuDevice", reflect.TypeOf((*MockInterface)(nil).ReserveGpuDevice), ctx, pod, nodeName, fractionalGpuGroup)
}

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

// Sync indicates an expected call of Sync.
func (mr *MockInterfaceMockRecorder) Sync(ctx any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "Sync", reflect.TypeOf((*MockInterface)(nil).Sync), ctx)
}

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

// SyncForGpuGroup indicates an expected call of SyncForGpuGroup.
func (mr *MockInterfaceMockRecorder) SyncForGpuGroup(ctx, gpuGroup any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncForGpuGroup", reflect.TypeOf((*MockInterface)(nil).SyncForGpuGroup), ctx, gpuGroup)
}

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

// SyncForNode indicates an expected call of SyncForNode.
func (mr *MockInterfaceMockRecorder) SyncForNode(ctx, nodeName any) *gomock.Call {
	mr.mock.ctrl.T.Helper()
	return mr.mock.ctrl.RecordCallWithMethodType(mr.mock, "SyncForNode", reflect.TypeOf((*MockInterface)(nil).SyncForNode), ctx, nodeName)
}