mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2024-11-25 13:35:00 +00:00
f2ef7ee661
Signed-off-by: Evan Lezar <elezar@nvidia.com>
1156 lines
38 KiB
Go
1156 lines
38 KiB
Go
// Code generated by moq; DO NOT EDIT.
|
|
// github.com/matryer/moq
|
|
|
|
package nvml
|
|
|
|
import (
|
|
"sync"
|
|
)
|
|
|
|
// Ensure, that DeviceMock does implement Device.
|
|
// If this is not the case, regenerate this file with moq.
|
|
var _ Device = &DeviceMock{}
|
|
|
|
// DeviceMock is a mock implementation of Device.
|
|
//
|
|
// func TestSomethingThatUsesDevice(t *testing.T) {
|
|
//
|
|
// // make and configure a mocked Device
|
|
// mockedDevice := &DeviceMock{
|
|
// CreateGpuInstanceWithPlacementFunc: func(gpuInstanceProfileInfo *GpuInstanceProfileInfo, gpuInstancePlacement *GpuInstancePlacement) (GpuInstance, Return) {
|
|
// panic("mock out the CreateGpuInstanceWithPlacement method")
|
|
// },
|
|
// GetArchitectureFunc: func() (DeviceArchitecture, Return) {
|
|
// panic("mock out the GetArchitecture method")
|
|
// },
|
|
// GetAttributesFunc: func() (DeviceAttributes, Return) {
|
|
// panic("mock out the GetAttributes method")
|
|
// },
|
|
// GetBrandFunc: func() (BrandType, Return) {
|
|
// panic("mock out the GetBrand method")
|
|
// },
|
|
// GetComputeInstanceIdFunc: func() (int, Return) {
|
|
// panic("mock out the GetComputeInstanceId method")
|
|
// },
|
|
// GetCudaComputeCapabilityFunc: func() (int, int, Return) {
|
|
// panic("mock out the GetCudaComputeCapability method")
|
|
// },
|
|
// GetDeviceHandleFromMigDeviceHandleFunc: func() (Device, Return) {
|
|
// panic("mock out the GetDeviceHandleFromMigDeviceHandle method")
|
|
// },
|
|
// GetGpuInstanceByIdFunc: func(ID int) (GpuInstance, Return) {
|
|
// panic("mock out the GetGpuInstanceById method")
|
|
// },
|
|
// GetGpuInstanceIdFunc: func() (int, Return) {
|
|
// panic("mock out the GetGpuInstanceId method")
|
|
// },
|
|
// GetGpuInstancePossiblePlacementsFunc: func(gpuInstanceProfileInfo *GpuInstanceProfileInfo) ([]GpuInstancePlacement, Return) {
|
|
// panic("mock out the GetGpuInstancePossiblePlacements method")
|
|
// },
|
|
// GetGpuInstanceProfileInfoFunc: func(Profile int) (GpuInstanceProfileInfo, Return) {
|
|
// panic("mock out the GetGpuInstanceProfileInfo method")
|
|
// },
|
|
// GetGpuInstancesFunc: func(Info *GpuInstanceProfileInfo) ([]GpuInstance, Return) {
|
|
// panic("mock out the GetGpuInstances method")
|
|
// },
|
|
// GetIndexFunc: func() (int, Return) {
|
|
// panic("mock out the GetIndex method")
|
|
// },
|
|
// GetMaxMigDeviceCountFunc: func() (int, Return) {
|
|
// panic("mock out the GetMaxMigDeviceCount method")
|
|
// },
|
|
// GetMemoryInfoFunc: func() (Memory, Return) {
|
|
// panic("mock out the GetMemoryInfo method")
|
|
// },
|
|
// GetMigDeviceHandleByIndexFunc: func(Index int) (Device, Return) {
|
|
// panic("mock out the GetMigDeviceHandleByIndex method")
|
|
// },
|
|
// GetMigModeFunc: func() (int, int, Return) {
|
|
// panic("mock out the GetMigMode method")
|
|
// },
|
|
// GetMinorNumberFunc: func() (int, Return) {
|
|
// panic("mock out the GetMinorNumber method")
|
|
// },
|
|
// GetNameFunc: func() (string, Return) {
|
|
// panic("mock out the GetName method")
|
|
// },
|
|
// GetNvLinkRemotePciInfoFunc: func(n int) (PciInfo, Return) {
|
|
// panic("mock out the GetNvLinkRemotePciInfo method")
|
|
// },
|
|
// GetNvLinkStateFunc: func(n int) (EnableState, Return) {
|
|
// panic("mock out the GetNvLinkState method")
|
|
// },
|
|
// GetPciInfoFunc: func() (PciInfo, Return) {
|
|
// panic("mock out the GetPciInfo method")
|
|
// },
|
|
// GetSupportedEventTypesFunc: func() (uint64, Return) {
|
|
// panic("mock out the GetSupportedEventTypes method")
|
|
// },
|
|
// GetTopologyCommonAncestorFunc: func(device Device) (GpuTopologyLevel, Return) {
|
|
// panic("mock out the GetTopologyCommonAncestor method")
|
|
// },
|
|
// GetUUIDFunc: func() (string, Return) {
|
|
// panic("mock out the GetUUID method")
|
|
// },
|
|
// IsMigDeviceHandleFunc: func() (bool, Return) {
|
|
// panic("mock out the IsMigDeviceHandle method")
|
|
// },
|
|
// RegisterEventsFunc: func(v uint64, eventSet EventSet) Return {
|
|
// panic("mock out the RegisterEvents method")
|
|
// },
|
|
// SetMigModeFunc: func(Mode int) (Return, Return) {
|
|
// panic("mock out the SetMigMode method")
|
|
// },
|
|
// }
|
|
//
|
|
// // use mockedDevice in code that requires Device
|
|
// // and then make assertions.
|
|
//
|
|
// }
|
|
type DeviceMock struct {
|
|
// CreateGpuInstanceWithPlacementFunc mocks the CreateGpuInstanceWithPlacement method.
|
|
CreateGpuInstanceWithPlacementFunc func(gpuInstanceProfileInfo *GpuInstanceProfileInfo, gpuInstancePlacement *GpuInstancePlacement) (GpuInstance, Return)
|
|
|
|
// GetArchitectureFunc mocks the GetArchitecture method.
|
|
GetArchitectureFunc func() (DeviceArchitecture, Return)
|
|
|
|
// GetAttributesFunc mocks the GetAttributes method.
|
|
GetAttributesFunc func() (DeviceAttributes, Return)
|
|
|
|
// GetBrandFunc mocks the GetBrand method.
|
|
GetBrandFunc func() (BrandType, Return)
|
|
|
|
// GetComputeInstanceIdFunc mocks the GetComputeInstanceId method.
|
|
GetComputeInstanceIdFunc func() (int, Return)
|
|
|
|
// GetCudaComputeCapabilityFunc mocks the GetCudaComputeCapability method.
|
|
GetCudaComputeCapabilityFunc func() (int, int, Return)
|
|
|
|
// GetDeviceHandleFromMigDeviceHandleFunc mocks the GetDeviceHandleFromMigDeviceHandle method.
|
|
GetDeviceHandleFromMigDeviceHandleFunc func() (Device, Return)
|
|
|
|
// GetGpuInstanceByIdFunc mocks the GetGpuInstanceById method.
|
|
GetGpuInstanceByIdFunc func(ID int) (GpuInstance, Return)
|
|
|
|
// GetGpuInstanceIdFunc mocks the GetGpuInstanceId method.
|
|
GetGpuInstanceIdFunc func() (int, Return)
|
|
|
|
// GetGpuInstancePossiblePlacementsFunc mocks the GetGpuInstancePossiblePlacements method.
|
|
GetGpuInstancePossiblePlacementsFunc func(gpuInstanceProfileInfo *GpuInstanceProfileInfo) ([]GpuInstancePlacement, Return)
|
|
|
|
// GetGpuInstanceProfileInfoFunc mocks the GetGpuInstanceProfileInfo method.
|
|
GetGpuInstanceProfileInfoFunc func(Profile int) (GpuInstanceProfileInfo, Return)
|
|
|
|
// GetGpuInstancesFunc mocks the GetGpuInstances method.
|
|
GetGpuInstancesFunc func(Info *GpuInstanceProfileInfo) ([]GpuInstance, Return)
|
|
|
|
// GetIndexFunc mocks the GetIndex method.
|
|
GetIndexFunc func() (int, Return)
|
|
|
|
// GetMaxMigDeviceCountFunc mocks the GetMaxMigDeviceCount method.
|
|
GetMaxMigDeviceCountFunc func() (int, Return)
|
|
|
|
// GetMemoryInfoFunc mocks the GetMemoryInfo method.
|
|
GetMemoryInfoFunc func() (Memory, Return)
|
|
|
|
// GetMigDeviceHandleByIndexFunc mocks the GetMigDeviceHandleByIndex method.
|
|
GetMigDeviceHandleByIndexFunc func(Index int) (Device, Return)
|
|
|
|
// GetMigModeFunc mocks the GetMigMode method.
|
|
GetMigModeFunc func() (int, int, Return)
|
|
|
|
// GetMinorNumberFunc mocks the GetMinorNumber method.
|
|
GetMinorNumberFunc func() (int, Return)
|
|
|
|
// GetNameFunc mocks the GetName method.
|
|
GetNameFunc func() (string, Return)
|
|
|
|
// GetNvLinkRemotePciInfoFunc mocks the GetNvLinkRemotePciInfo method.
|
|
GetNvLinkRemotePciInfoFunc func(n int) (PciInfo, Return)
|
|
|
|
// GetNvLinkStateFunc mocks the GetNvLinkState method.
|
|
GetNvLinkStateFunc func(n int) (EnableState, Return)
|
|
|
|
// GetPciInfoFunc mocks the GetPciInfo method.
|
|
GetPciInfoFunc func() (PciInfo, Return)
|
|
|
|
// GetSupportedEventTypesFunc mocks the GetSupportedEventTypes method.
|
|
GetSupportedEventTypesFunc func() (uint64, Return)
|
|
|
|
// GetTopologyCommonAncestorFunc mocks the GetTopologyCommonAncestor method.
|
|
GetTopologyCommonAncestorFunc func(device Device) (GpuTopologyLevel, Return)
|
|
|
|
// GetUUIDFunc mocks the GetUUID method.
|
|
GetUUIDFunc func() (string, Return)
|
|
|
|
// IsMigDeviceHandleFunc mocks the IsMigDeviceHandle method.
|
|
IsMigDeviceHandleFunc func() (bool, Return)
|
|
|
|
// RegisterEventsFunc mocks the RegisterEvents method.
|
|
RegisterEventsFunc func(v uint64, eventSet EventSet) Return
|
|
|
|
// SetMigModeFunc mocks the SetMigMode method.
|
|
SetMigModeFunc func(Mode int) (Return, Return)
|
|
|
|
// calls tracks calls to the methods.
|
|
calls struct {
|
|
// CreateGpuInstanceWithPlacement holds details about calls to the CreateGpuInstanceWithPlacement method.
|
|
CreateGpuInstanceWithPlacement []struct {
|
|
// GpuInstanceProfileInfo is the gpuInstanceProfileInfo argument value.
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
// GpuInstancePlacement is the gpuInstancePlacement argument value.
|
|
GpuInstancePlacement *GpuInstancePlacement
|
|
}
|
|
// GetArchitecture holds details about calls to the GetArchitecture method.
|
|
GetArchitecture []struct {
|
|
}
|
|
// GetAttributes holds details about calls to the GetAttributes method.
|
|
GetAttributes []struct {
|
|
}
|
|
// GetBrand holds details about calls to the GetBrand method.
|
|
GetBrand []struct {
|
|
}
|
|
// GetComputeInstanceId holds details about calls to the GetComputeInstanceId method.
|
|
GetComputeInstanceId []struct {
|
|
}
|
|
// GetCudaComputeCapability holds details about calls to the GetCudaComputeCapability method.
|
|
GetCudaComputeCapability []struct {
|
|
}
|
|
// GetDeviceHandleFromMigDeviceHandle holds details about calls to the GetDeviceHandleFromMigDeviceHandle method.
|
|
GetDeviceHandleFromMigDeviceHandle []struct {
|
|
}
|
|
// GetGpuInstanceById holds details about calls to the GetGpuInstanceById method.
|
|
GetGpuInstanceById []struct {
|
|
// ID is the ID argument value.
|
|
ID int
|
|
}
|
|
// GetGpuInstanceId holds details about calls to the GetGpuInstanceId method.
|
|
GetGpuInstanceId []struct {
|
|
}
|
|
// GetGpuInstancePossiblePlacements holds details about calls to the GetGpuInstancePossiblePlacements method.
|
|
GetGpuInstancePossiblePlacements []struct {
|
|
// GpuInstanceProfileInfo is the gpuInstanceProfileInfo argument value.
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
}
|
|
// GetGpuInstanceProfileInfo holds details about calls to the GetGpuInstanceProfileInfo method.
|
|
GetGpuInstanceProfileInfo []struct {
|
|
// Profile is the Profile argument value.
|
|
Profile int
|
|
}
|
|
// GetGpuInstances holds details about calls to the GetGpuInstances method.
|
|
GetGpuInstances []struct {
|
|
// Info is the Info argument value.
|
|
Info *GpuInstanceProfileInfo
|
|
}
|
|
// GetIndex holds details about calls to the GetIndex method.
|
|
GetIndex []struct {
|
|
}
|
|
// GetMaxMigDeviceCount holds details about calls to the GetMaxMigDeviceCount method.
|
|
GetMaxMigDeviceCount []struct {
|
|
}
|
|
// GetMemoryInfo holds details about calls to the GetMemoryInfo method.
|
|
GetMemoryInfo []struct {
|
|
}
|
|
// GetMigDeviceHandleByIndex holds details about calls to the GetMigDeviceHandleByIndex method.
|
|
GetMigDeviceHandleByIndex []struct {
|
|
// Index is the Index argument value.
|
|
Index int
|
|
}
|
|
// GetMigMode holds details about calls to the GetMigMode method.
|
|
GetMigMode []struct {
|
|
}
|
|
// GetMinorNumber holds details about calls to the GetMinorNumber method.
|
|
GetMinorNumber []struct {
|
|
}
|
|
// GetName holds details about calls to the GetName method.
|
|
GetName []struct {
|
|
}
|
|
// GetNvLinkRemotePciInfo holds details about calls to the GetNvLinkRemotePciInfo method.
|
|
GetNvLinkRemotePciInfo []struct {
|
|
// N is the n argument value.
|
|
N int
|
|
}
|
|
// GetNvLinkState holds details about calls to the GetNvLinkState method.
|
|
GetNvLinkState []struct {
|
|
// N is the n argument value.
|
|
N int
|
|
}
|
|
// GetPciInfo holds details about calls to the GetPciInfo method.
|
|
GetPciInfo []struct {
|
|
}
|
|
// GetSupportedEventTypes holds details about calls to the GetSupportedEventTypes method.
|
|
GetSupportedEventTypes []struct {
|
|
}
|
|
// GetTopologyCommonAncestor holds details about calls to the GetTopologyCommonAncestor method.
|
|
GetTopologyCommonAncestor []struct {
|
|
// Device is the device argument value.
|
|
Device Device
|
|
}
|
|
// GetUUID holds details about calls to the GetUUID method.
|
|
GetUUID []struct {
|
|
}
|
|
// IsMigDeviceHandle holds details about calls to the IsMigDeviceHandle method.
|
|
IsMigDeviceHandle []struct {
|
|
}
|
|
// RegisterEvents holds details about calls to the RegisterEvents method.
|
|
RegisterEvents []struct {
|
|
// V is the v argument value.
|
|
V uint64
|
|
// EventSet is the eventSet argument value.
|
|
EventSet EventSet
|
|
}
|
|
// SetMigMode holds details about calls to the SetMigMode method.
|
|
SetMigMode []struct {
|
|
// Mode is the Mode argument value.
|
|
Mode int
|
|
}
|
|
}
|
|
lockCreateGpuInstanceWithPlacement sync.RWMutex
|
|
lockGetArchitecture sync.RWMutex
|
|
lockGetAttributes sync.RWMutex
|
|
lockGetBrand sync.RWMutex
|
|
lockGetComputeInstanceId sync.RWMutex
|
|
lockGetCudaComputeCapability sync.RWMutex
|
|
lockGetDeviceHandleFromMigDeviceHandle sync.RWMutex
|
|
lockGetGpuInstanceById sync.RWMutex
|
|
lockGetGpuInstanceId sync.RWMutex
|
|
lockGetGpuInstancePossiblePlacements sync.RWMutex
|
|
lockGetGpuInstanceProfileInfo sync.RWMutex
|
|
lockGetGpuInstances sync.RWMutex
|
|
lockGetIndex sync.RWMutex
|
|
lockGetMaxMigDeviceCount sync.RWMutex
|
|
lockGetMemoryInfo sync.RWMutex
|
|
lockGetMigDeviceHandleByIndex sync.RWMutex
|
|
lockGetMigMode sync.RWMutex
|
|
lockGetMinorNumber sync.RWMutex
|
|
lockGetName sync.RWMutex
|
|
lockGetNvLinkRemotePciInfo sync.RWMutex
|
|
lockGetNvLinkState sync.RWMutex
|
|
lockGetPciInfo sync.RWMutex
|
|
lockGetSupportedEventTypes sync.RWMutex
|
|
lockGetTopologyCommonAncestor sync.RWMutex
|
|
lockGetUUID sync.RWMutex
|
|
lockIsMigDeviceHandle sync.RWMutex
|
|
lockRegisterEvents sync.RWMutex
|
|
lockSetMigMode sync.RWMutex
|
|
}
|
|
|
|
// CreateGpuInstanceWithPlacement calls CreateGpuInstanceWithPlacementFunc.
|
|
func (mock *DeviceMock) CreateGpuInstanceWithPlacement(gpuInstanceProfileInfo *GpuInstanceProfileInfo, gpuInstancePlacement *GpuInstancePlacement) (GpuInstance, Return) {
|
|
if mock.CreateGpuInstanceWithPlacementFunc == nil {
|
|
panic("DeviceMock.CreateGpuInstanceWithPlacementFunc: method is nil but Device.CreateGpuInstanceWithPlacement was just called")
|
|
}
|
|
callInfo := struct {
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
GpuInstancePlacement *GpuInstancePlacement
|
|
}{
|
|
GpuInstanceProfileInfo: gpuInstanceProfileInfo,
|
|
GpuInstancePlacement: gpuInstancePlacement,
|
|
}
|
|
mock.lockCreateGpuInstanceWithPlacement.Lock()
|
|
mock.calls.CreateGpuInstanceWithPlacement = append(mock.calls.CreateGpuInstanceWithPlacement, callInfo)
|
|
mock.lockCreateGpuInstanceWithPlacement.Unlock()
|
|
return mock.CreateGpuInstanceWithPlacementFunc(gpuInstanceProfileInfo, gpuInstancePlacement)
|
|
}
|
|
|
|
// CreateGpuInstanceWithPlacementCalls gets all the calls that were made to CreateGpuInstanceWithPlacement.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.CreateGpuInstanceWithPlacementCalls())
|
|
func (mock *DeviceMock) CreateGpuInstanceWithPlacementCalls() []struct {
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
GpuInstancePlacement *GpuInstancePlacement
|
|
} {
|
|
var calls []struct {
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
GpuInstancePlacement *GpuInstancePlacement
|
|
}
|
|
mock.lockCreateGpuInstanceWithPlacement.RLock()
|
|
calls = mock.calls.CreateGpuInstanceWithPlacement
|
|
mock.lockCreateGpuInstanceWithPlacement.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetArchitecture calls GetArchitectureFunc.
|
|
func (mock *DeviceMock) GetArchitecture() (DeviceArchitecture, Return) {
|
|
if mock.GetArchitectureFunc == nil {
|
|
panic("DeviceMock.GetArchitectureFunc: method is nil but Device.GetArchitecture was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetArchitecture.Lock()
|
|
mock.calls.GetArchitecture = append(mock.calls.GetArchitecture, callInfo)
|
|
mock.lockGetArchitecture.Unlock()
|
|
return mock.GetArchitectureFunc()
|
|
}
|
|
|
|
// GetArchitectureCalls gets all the calls that were made to GetArchitecture.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetArchitectureCalls())
|
|
func (mock *DeviceMock) GetArchitectureCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetArchitecture.RLock()
|
|
calls = mock.calls.GetArchitecture
|
|
mock.lockGetArchitecture.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetAttributes calls GetAttributesFunc.
|
|
func (mock *DeviceMock) GetAttributes() (DeviceAttributes, Return) {
|
|
if mock.GetAttributesFunc == nil {
|
|
panic("DeviceMock.GetAttributesFunc: method is nil but Device.GetAttributes was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetAttributes.Lock()
|
|
mock.calls.GetAttributes = append(mock.calls.GetAttributes, callInfo)
|
|
mock.lockGetAttributes.Unlock()
|
|
return mock.GetAttributesFunc()
|
|
}
|
|
|
|
// GetAttributesCalls gets all the calls that were made to GetAttributes.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetAttributesCalls())
|
|
func (mock *DeviceMock) GetAttributesCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetAttributes.RLock()
|
|
calls = mock.calls.GetAttributes
|
|
mock.lockGetAttributes.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetBrand calls GetBrandFunc.
|
|
func (mock *DeviceMock) GetBrand() (BrandType, Return) {
|
|
if mock.GetBrandFunc == nil {
|
|
panic("DeviceMock.GetBrandFunc: method is nil but Device.GetBrand was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetBrand.Lock()
|
|
mock.calls.GetBrand = append(mock.calls.GetBrand, callInfo)
|
|
mock.lockGetBrand.Unlock()
|
|
return mock.GetBrandFunc()
|
|
}
|
|
|
|
// GetBrandCalls gets all the calls that were made to GetBrand.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetBrandCalls())
|
|
func (mock *DeviceMock) GetBrandCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetBrand.RLock()
|
|
calls = mock.calls.GetBrand
|
|
mock.lockGetBrand.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetComputeInstanceId calls GetComputeInstanceIdFunc.
|
|
func (mock *DeviceMock) GetComputeInstanceId() (int, Return) {
|
|
if mock.GetComputeInstanceIdFunc == nil {
|
|
panic("DeviceMock.GetComputeInstanceIdFunc: method is nil but Device.GetComputeInstanceId was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetComputeInstanceId.Lock()
|
|
mock.calls.GetComputeInstanceId = append(mock.calls.GetComputeInstanceId, callInfo)
|
|
mock.lockGetComputeInstanceId.Unlock()
|
|
return mock.GetComputeInstanceIdFunc()
|
|
}
|
|
|
|
// GetComputeInstanceIdCalls gets all the calls that were made to GetComputeInstanceId.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetComputeInstanceIdCalls())
|
|
func (mock *DeviceMock) GetComputeInstanceIdCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetComputeInstanceId.RLock()
|
|
calls = mock.calls.GetComputeInstanceId
|
|
mock.lockGetComputeInstanceId.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetCudaComputeCapability calls GetCudaComputeCapabilityFunc.
|
|
func (mock *DeviceMock) GetCudaComputeCapability() (int, int, Return) {
|
|
if mock.GetCudaComputeCapabilityFunc == nil {
|
|
panic("DeviceMock.GetCudaComputeCapabilityFunc: method is nil but Device.GetCudaComputeCapability was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetCudaComputeCapability.Lock()
|
|
mock.calls.GetCudaComputeCapability = append(mock.calls.GetCudaComputeCapability, callInfo)
|
|
mock.lockGetCudaComputeCapability.Unlock()
|
|
return mock.GetCudaComputeCapabilityFunc()
|
|
}
|
|
|
|
// GetCudaComputeCapabilityCalls gets all the calls that were made to GetCudaComputeCapability.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetCudaComputeCapabilityCalls())
|
|
func (mock *DeviceMock) GetCudaComputeCapabilityCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetCudaComputeCapability.RLock()
|
|
calls = mock.calls.GetCudaComputeCapability
|
|
mock.lockGetCudaComputeCapability.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetDeviceHandleFromMigDeviceHandle calls GetDeviceHandleFromMigDeviceHandleFunc.
|
|
func (mock *DeviceMock) GetDeviceHandleFromMigDeviceHandle() (Device, Return) {
|
|
if mock.GetDeviceHandleFromMigDeviceHandleFunc == nil {
|
|
panic("DeviceMock.GetDeviceHandleFromMigDeviceHandleFunc: method is nil but Device.GetDeviceHandleFromMigDeviceHandle was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetDeviceHandleFromMigDeviceHandle.Lock()
|
|
mock.calls.GetDeviceHandleFromMigDeviceHandle = append(mock.calls.GetDeviceHandleFromMigDeviceHandle, callInfo)
|
|
mock.lockGetDeviceHandleFromMigDeviceHandle.Unlock()
|
|
return mock.GetDeviceHandleFromMigDeviceHandleFunc()
|
|
}
|
|
|
|
// GetDeviceHandleFromMigDeviceHandleCalls gets all the calls that were made to GetDeviceHandleFromMigDeviceHandle.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetDeviceHandleFromMigDeviceHandleCalls())
|
|
func (mock *DeviceMock) GetDeviceHandleFromMigDeviceHandleCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetDeviceHandleFromMigDeviceHandle.RLock()
|
|
calls = mock.calls.GetDeviceHandleFromMigDeviceHandle
|
|
mock.lockGetDeviceHandleFromMigDeviceHandle.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetGpuInstanceById calls GetGpuInstanceByIdFunc.
|
|
func (mock *DeviceMock) GetGpuInstanceById(ID int) (GpuInstance, Return) {
|
|
if mock.GetGpuInstanceByIdFunc == nil {
|
|
panic("DeviceMock.GetGpuInstanceByIdFunc: method is nil but Device.GetGpuInstanceById was just called")
|
|
}
|
|
callInfo := struct {
|
|
ID int
|
|
}{
|
|
ID: ID,
|
|
}
|
|
mock.lockGetGpuInstanceById.Lock()
|
|
mock.calls.GetGpuInstanceById = append(mock.calls.GetGpuInstanceById, callInfo)
|
|
mock.lockGetGpuInstanceById.Unlock()
|
|
return mock.GetGpuInstanceByIdFunc(ID)
|
|
}
|
|
|
|
// GetGpuInstanceByIdCalls gets all the calls that were made to GetGpuInstanceById.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetGpuInstanceByIdCalls())
|
|
func (mock *DeviceMock) GetGpuInstanceByIdCalls() []struct {
|
|
ID int
|
|
} {
|
|
var calls []struct {
|
|
ID int
|
|
}
|
|
mock.lockGetGpuInstanceById.RLock()
|
|
calls = mock.calls.GetGpuInstanceById
|
|
mock.lockGetGpuInstanceById.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetGpuInstanceId calls GetGpuInstanceIdFunc.
|
|
func (mock *DeviceMock) GetGpuInstanceId() (int, Return) {
|
|
if mock.GetGpuInstanceIdFunc == nil {
|
|
panic("DeviceMock.GetGpuInstanceIdFunc: method is nil but Device.GetGpuInstanceId was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetGpuInstanceId.Lock()
|
|
mock.calls.GetGpuInstanceId = append(mock.calls.GetGpuInstanceId, callInfo)
|
|
mock.lockGetGpuInstanceId.Unlock()
|
|
return mock.GetGpuInstanceIdFunc()
|
|
}
|
|
|
|
// GetGpuInstanceIdCalls gets all the calls that were made to GetGpuInstanceId.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetGpuInstanceIdCalls())
|
|
func (mock *DeviceMock) GetGpuInstanceIdCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetGpuInstanceId.RLock()
|
|
calls = mock.calls.GetGpuInstanceId
|
|
mock.lockGetGpuInstanceId.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetGpuInstancePossiblePlacements calls GetGpuInstancePossiblePlacementsFunc.
|
|
func (mock *DeviceMock) GetGpuInstancePossiblePlacements(gpuInstanceProfileInfo *GpuInstanceProfileInfo) ([]GpuInstancePlacement, Return) {
|
|
if mock.GetGpuInstancePossiblePlacementsFunc == nil {
|
|
panic("DeviceMock.GetGpuInstancePossiblePlacementsFunc: method is nil but Device.GetGpuInstancePossiblePlacements was just called")
|
|
}
|
|
callInfo := struct {
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
}{
|
|
GpuInstanceProfileInfo: gpuInstanceProfileInfo,
|
|
}
|
|
mock.lockGetGpuInstancePossiblePlacements.Lock()
|
|
mock.calls.GetGpuInstancePossiblePlacements = append(mock.calls.GetGpuInstancePossiblePlacements, callInfo)
|
|
mock.lockGetGpuInstancePossiblePlacements.Unlock()
|
|
return mock.GetGpuInstancePossiblePlacementsFunc(gpuInstanceProfileInfo)
|
|
}
|
|
|
|
// GetGpuInstancePossiblePlacementsCalls gets all the calls that were made to GetGpuInstancePossiblePlacements.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetGpuInstancePossiblePlacementsCalls())
|
|
func (mock *DeviceMock) GetGpuInstancePossiblePlacementsCalls() []struct {
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
} {
|
|
var calls []struct {
|
|
GpuInstanceProfileInfo *GpuInstanceProfileInfo
|
|
}
|
|
mock.lockGetGpuInstancePossiblePlacements.RLock()
|
|
calls = mock.calls.GetGpuInstancePossiblePlacements
|
|
mock.lockGetGpuInstancePossiblePlacements.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetGpuInstanceProfileInfo calls GetGpuInstanceProfileInfoFunc.
|
|
func (mock *DeviceMock) GetGpuInstanceProfileInfo(Profile int) (GpuInstanceProfileInfo, Return) {
|
|
if mock.GetGpuInstanceProfileInfoFunc == nil {
|
|
panic("DeviceMock.GetGpuInstanceProfileInfoFunc: method is nil but Device.GetGpuInstanceProfileInfo was just called")
|
|
}
|
|
callInfo := struct {
|
|
Profile int
|
|
}{
|
|
Profile: Profile,
|
|
}
|
|
mock.lockGetGpuInstanceProfileInfo.Lock()
|
|
mock.calls.GetGpuInstanceProfileInfo = append(mock.calls.GetGpuInstanceProfileInfo, callInfo)
|
|
mock.lockGetGpuInstanceProfileInfo.Unlock()
|
|
return mock.GetGpuInstanceProfileInfoFunc(Profile)
|
|
}
|
|
|
|
// GetGpuInstanceProfileInfoCalls gets all the calls that were made to GetGpuInstanceProfileInfo.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetGpuInstanceProfileInfoCalls())
|
|
func (mock *DeviceMock) GetGpuInstanceProfileInfoCalls() []struct {
|
|
Profile int
|
|
} {
|
|
var calls []struct {
|
|
Profile int
|
|
}
|
|
mock.lockGetGpuInstanceProfileInfo.RLock()
|
|
calls = mock.calls.GetGpuInstanceProfileInfo
|
|
mock.lockGetGpuInstanceProfileInfo.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetGpuInstances calls GetGpuInstancesFunc.
|
|
func (mock *DeviceMock) GetGpuInstances(Info *GpuInstanceProfileInfo) ([]GpuInstance, Return) {
|
|
if mock.GetGpuInstancesFunc == nil {
|
|
panic("DeviceMock.GetGpuInstancesFunc: method is nil but Device.GetGpuInstances was just called")
|
|
}
|
|
callInfo := struct {
|
|
Info *GpuInstanceProfileInfo
|
|
}{
|
|
Info: Info,
|
|
}
|
|
mock.lockGetGpuInstances.Lock()
|
|
mock.calls.GetGpuInstances = append(mock.calls.GetGpuInstances, callInfo)
|
|
mock.lockGetGpuInstances.Unlock()
|
|
return mock.GetGpuInstancesFunc(Info)
|
|
}
|
|
|
|
// GetGpuInstancesCalls gets all the calls that were made to GetGpuInstances.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetGpuInstancesCalls())
|
|
func (mock *DeviceMock) GetGpuInstancesCalls() []struct {
|
|
Info *GpuInstanceProfileInfo
|
|
} {
|
|
var calls []struct {
|
|
Info *GpuInstanceProfileInfo
|
|
}
|
|
mock.lockGetGpuInstances.RLock()
|
|
calls = mock.calls.GetGpuInstances
|
|
mock.lockGetGpuInstances.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetIndex calls GetIndexFunc.
|
|
func (mock *DeviceMock) GetIndex() (int, Return) {
|
|
if mock.GetIndexFunc == nil {
|
|
panic("DeviceMock.GetIndexFunc: method is nil but Device.GetIndex was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetIndex.Lock()
|
|
mock.calls.GetIndex = append(mock.calls.GetIndex, callInfo)
|
|
mock.lockGetIndex.Unlock()
|
|
return mock.GetIndexFunc()
|
|
}
|
|
|
|
// GetIndexCalls gets all the calls that were made to GetIndex.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetIndexCalls())
|
|
func (mock *DeviceMock) GetIndexCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetIndex.RLock()
|
|
calls = mock.calls.GetIndex
|
|
mock.lockGetIndex.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetMaxMigDeviceCount calls GetMaxMigDeviceCountFunc.
|
|
func (mock *DeviceMock) GetMaxMigDeviceCount() (int, Return) {
|
|
if mock.GetMaxMigDeviceCountFunc == nil {
|
|
panic("DeviceMock.GetMaxMigDeviceCountFunc: method is nil but Device.GetMaxMigDeviceCount was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetMaxMigDeviceCount.Lock()
|
|
mock.calls.GetMaxMigDeviceCount = append(mock.calls.GetMaxMigDeviceCount, callInfo)
|
|
mock.lockGetMaxMigDeviceCount.Unlock()
|
|
return mock.GetMaxMigDeviceCountFunc()
|
|
}
|
|
|
|
// GetMaxMigDeviceCountCalls gets all the calls that were made to GetMaxMigDeviceCount.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetMaxMigDeviceCountCalls())
|
|
func (mock *DeviceMock) GetMaxMigDeviceCountCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetMaxMigDeviceCount.RLock()
|
|
calls = mock.calls.GetMaxMigDeviceCount
|
|
mock.lockGetMaxMigDeviceCount.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetMemoryInfo calls GetMemoryInfoFunc.
|
|
func (mock *DeviceMock) GetMemoryInfo() (Memory, Return) {
|
|
if mock.GetMemoryInfoFunc == nil {
|
|
panic("DeviceMock.GetMemoryInfoFunc: method is nil but Device.GetMemoryInfo was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetMemoryInfo.Lock()
|
|
mock.calls.GetMemoryInfo = append(mock.calls.GetMemoryInfo, callInfo)
|
|
mock.lockGetMemoryInfo.Unlock()
|
|
return mock.GetMemoryInfoFunc()
|
|
}
|
|
|
|
// GetMemoryInfoCalls gets all the calls that were made to GetMemoryInfo.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetMemoryInfoCalls())
|
|
func (mock *DeviceMock) GetMemoryInfoCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetMemoryInfo.RLock()
|
|
calls = mock.calls.GetMemoryInfo
|
|
mock.lockGetMemoryInfo.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetMigDeviceHandleByIndex calls GetMigDeviceHandleByIndexFunc.
|
|
func (mock *DeviceMock) GetMigDeviceHandleByIndex(Index int) (Device, Return) {
|
|
if mock.GetMigDeviceHandleByIndexFunc == nil {
|
|
panic("DeviceMock.GetMigDeviceHandleByIndexFunc: method is nil but Device.GetMigDeviceHandleByIndex was just called")
|
|
}
|
|
callInfo := struct {
|
|
Index int
|
|
}{
|
|
Index: Index,
|
|
}
|
|
mock.lockGetMigDeviceHandleByIndex.Lock()
|
|
mock.calls.GetMigDeviceHandleByIndex = append(mock.calls.GetMigDeviceHandleByIndex, callInfo)
|
|
mock.lockGetMigDeviceHandleByIndex.Unlock()
|
|
return mock.GetMigDeviceHandleByIndexFunc(Index)
|
|
}
|
|
|
|
// GetMigDeviceHandleByIndexCalls gets all the calls that were made to GetMigDeviceHandleByIndex.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetMigDeviceHandleByIndexCalls())
|
|
func (mock *DeviceMock) GetMigDeviceHandleByIndexCalls() []struct {
|
|
Index int
|
|
} {
|
|
var calls []struct {
|
|
Index int
|
|
}
|
|
mock.lockGetMigDeviceHandleByIndex.RLock()
|
|
calls = mock.calls.GetMigDeviceHandleByIndex
|
|
mock.lockGetMigDeviceHandleByIndex.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetMigMode calls GetMigModeFunc.
|
|
func (mock *DeviceMock) GetMigMode() (int, int, Return) {
|
|
if mock.GetMigModeFunc == nil {
|
|
panic("DeviceMock.GetMigModeFunc: method is nil but Device.GetMigMode was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetMigMode.Lock()
|
|
mock.calls.GetMigMode = append(mock.calls.GetMigMode, callInfo)
|
|
mock.lockGetMigMode.Unlock()
|
|
return mock.GetMigModeFunc()
|
|
}
|
|
|
|
// GetMigModeCalls gets all the calls that were made to GetMigMode.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetMigModeCalls())
|
|
func (mock *DeviceMock) GetMigModeCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetMigMode.RLock()
|
|
calls = mock.calls.GetMigMode
|
|
mock.lockGetMigMode.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetMinorNumber calls GetMinorNumberFunc.
|
|
func (mock *DeviceMock) GetMinorNumber() (int, Return) {
|
|
if mock.GetMinorNumberFunc == nil {
|
|
panic("DeviceMock.GetMinorNumberFunc: method is nil but Device.GetMinorNumber was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetMinorNumber.Lock()
|
|
mock.calls.GetMinorNumber = append(mock.calls.GetMinorNumber, callInfo)
|
|
mock.lockGetMinorNumber.Unlock()
|
|
return mock.GetMinorNumberFunc()
|
|
}
|
|
|
|
// GetMinorNumberCalls gets all the calls that were made to GetMinorNumber.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetMinorNumberCalls())
|
|
func (mock *DeviceMock) GetMinorNumberCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetMinorNumber.RLock()
|
|
calls = mock.calls.GetMinorNumber
|
|
mock.lockGetMinorNumber.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetName calls GetNameFunc.
|
|
func (mock *DeviceMock) GetName() (string, Return) {
|
|
if mock.GetNameFunc == nil {
|
|
panic("DeviceMock.GetNameFunc: method is nil but Device.GetName was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetName.Lock()
|
|
mock.calls.GetName = append(mock.calls.GetName, callInfo)
|
|
mock.lockGetName.Unlock()
|
|
return mock.GetNameFunc()
|
|
}
|
|
|
|
// GetNameCalls gets all the calls that were made to GetName.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetNameCalls())
|
|
func (mock *DeviceMock) GetNameCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetName.RLock()
|
|
calls = mock.calls.GetName
|
|
mock.lockGetName.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetNvLinkRemotePciInfo calls GetNvLinkRemotePciInfoFunc.
|
|
func (mock *DeviceMock) GetNvLinkRemotePciInfo(n int) (PciInfo, Return) {
|
|
if mock.GetNvLinkRemotePciInfoFunc == nil {
|
|
panic("DeviceMock.GetNvLinkRemotePciInfoFunc: method is nil but Device.GetNvLinkRemotePciInfo was just called")
|
|
}
|
|
callInfo := struct {
|
|
N int
|
|
}{
|
|
N: n,
|
|
}
|
|
mock.lockGetNvLinkRemotePciInfo.Lock()
|
|
mock.calls.GetNvLinkRemotePciInfo = append(mock.calls.GetNvLinkRemotePciInfo, callInfo)
|
|
mock.lockGetNvLinkRemotePciInfo.Unlock()
|
|
return mock.GetNvLinkRemotePciInfoFunc(n)
|
|
}
|
|
|
|
// GetNvLinkRemotePciInfoCalls gets all the calls that were made to GetNvLinkRemotePciInfo.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetNvLinkRemotePciInfoCalls())
|
|
func (mock *DeviceMock) GetNvLinkRemotePciInfoCalls() []struct {
|
|
N int
|
|
} {
|
|
var calls []struct {
|
|
N int
|
|
}
|
|
mock.lockGetNvLinkRemotePciInfo.RLock()
|
|
calls = mock.calls.GetNvLinkRemotePciInfo
|
|
mock.lockGetNvLinkRemotePciInfo.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetNvLinkState calls GetNvLinkStateFunc.
|
|
func (mock *DeviceMock) GetNvLinkState(n int) (EnableState, Return) {
|
|
if mock.GetNvLinkStateFunc == nil {
|
|
panic("DeviceMock.GetNvLinkStateFunc: method is nil but Device.GetNvLinkState was just called")
|
|
}
|
|
callInfo := struct {
|
|
N int
|
|
}{
|
|
N: n,
|
|
}
|
|
mock.lockGetNvLinkState.Lock()
|
|
mock.calls.GetNvLinkState = append(mock.calls.GetNvLinkState, callInfo)
|
|
mock.lockGetNvLinkState.Unlock()
|
|
return mock.GetNvLinkStateFunc(n)
|
|
}
|
|
|
|
// GetNvLinkStateCalls gets all the calls that were made to GetNvLinkState.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetNvLinkStateCalls())
|
|
func (mock *DeviceMock) GetNvLinkStateCalls() []struct {
|
|
N int
|
|
} {
|
|
var calls []struct {
|
|
N int
|
|
}
|
|
mock.lockGetNvLinkState.RLock()
|
|
calls = mock.calls.GetNvLinkState
|
|
mock.lockGetNvLinkState.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetPciInfo calls GetPciInfoFunc.
|
|
func (mock *DeviceMock) GetPciInfo() (PciInfo, Return) {
|
|
if mock.GetPciInfoFunc == nil {
|
|
panic("DeviceMock.GetPciInfoFunc: method is nil but Device.GetPciInfo was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetPciInfo.Lock()
|
|
mock.calls.GetPciInfo = append(mock.calls.GetPciInfo, callInfo)
|
|
mock.lockGetPciInfo.Unlock()
|
|
return mock.GetPciInfoFunc()
|
|
}
|
|
|
|
// GetPciInfoCalls gets all the calls that were made to GetPciInfo.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetPciInfoCalls())
|
|
func (mock *DeviceMock) GetPciInfoCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetPciInfo.RLock()
|
|
calls = mock.calls.GetPciInfo
|
|
mock.lockGetPciInfo.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetSupportedEventTypes calls GetSupportedEventTypesFunc.
|
|
func (mock *DeviceMock) GetSupportedEventTypes() (uint64, Return) {
|
|
if mock.GetSupportedEventTypesFunc == nil {
|
|
panic("DeviceMock.GetSupportedEventTypesFunc: method is nil but Device.GetSupportedEventTypes was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetSupportedEventTypes.Lock()
|
|
mock.calls.GetSupportedEventTypes = append(mock.calls.GetSupportedEventTypes, callInfo)
|
|
mock.lockGetSupportedEventTypes.Unlock()
|
|
return mock.GetSupportedEventTypesFunc()
|
|
}
|
|
|
|
// GetSupportedEventTypesCalls gets all the calls that were made to GetSupportedEventTypes.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetSupportedEventTypesCalls())
|
|
func (mock *DeviceMock) GetSupportedEventTypesCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetSupportedEventTypes.RLock()
|
|
calls = mock.calls.GetSupportedEventTypes
|
|
mock.lockGetSupportedEventTypes.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetTopologyCommonAncestor calls GetTopologyCommonAncestorFunc.
|
|
func (mock *DeviceMock) GetTopologyCommonAncestor(device Device) (GpuTopologyLevel, Return) {
|
|
if mock.GetTopologyCommonAncestorFunc == nil {
|
|
panic("DeviceMock.GetTopologyCommonAncestorFunc: method is nil but Device.GetTopologyCommonAncestor was just called")
|
|
}
|
|
callInfo := struct {
|
|
Device Device
|
|
}{
|
|
Device: device,
|
|
}
|
|
mock.lockGetTopologyCommonAncestor.Lock()
|
|
mock.calls.GetTopologyCommonAncestor = append(mock.calls.GetTopologyCommonAncestor, callInfo)
|
|
mock.lockGetTopologyCommonAncestor.Unlock()
|
|
return mock.GetTopologyCommonAncestorFunc(device)
|
|
}
|
|
|
|
// GetTopologyCommonAncestorCalls gets all the calls that were made to GetTopologyCommonAncestor.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetTopologyCommonAncestorCalls())
|
|
func (mock *DeviceMock) GetTopologyCommonAncestorCalls() []struct {
|
|
Device Device
|
|
} {
|
|
var calls []struct {
|
|
Device Device
|
|
}
|
|
mock.lockGetTopologyCommonAncestor.RLock()
|
|
calls = mock.calls.GetTopologyCommonAncestor
|
|
mock.lockGetTopologyCommonAncestor.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// GetUUID calls GetUUIDFunc.
|
|
func (mock *DeviceMock) GetUUID() (string, Return) {
|
|
if mock.GetUUIDFunc == nil {
|
|
panic("DeviceMock.GetUUIDFunc: method is nil but Device.GetUUID was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockGetUUID.Lock()
|
|
mock.calls.GetUUID = append(mock.calls.GetUUID, callInfo)
|
|
mock.lockGetUUID.Unlock()
|
|
return mock.GetUUIDFunc()
|
|
}
|
|
|
|
// GetUUIDCalls gets all the calls that were made to GetUUID.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.GetUUIDCalls())
|
|
func (mock *DeviceMock) GetUUIDCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockGetUUID.RLock()
|
|
calls = mock.calls.GetUUID
|
|
mock.lockGetUUID.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// IsMigDeviceHandle calls IsMigDeviceHandleFunc.
|
|
func (mock *DeviceMock) IsMigDeviceHandle() (bool, Return) {
|
|
if mock.IsMigDeviceHandleFunc == nil {
|
|
panic("DeviceMock.IsMigDeviceHandleFunc: method is nil but Device.IsMigDeviceHandle was just called")
|
|
}
|
|
callInfo := struct {
|
|
}{}
|
|
mock.lockIsMigDeviceHandle.Lock()
|
|
mock.calls.IsMigDeviceHandle = append(mock.calls.IsMigDeviceHandle, callInfo)
|
|
mock.lockIsMigDeviceHandle.Unlock()
|
|
return mock.IsMigDeviceHandleFunc()
|
|
}
|
|
|
|
// IsMigDeviceHandleCalls gets all the calls that were made to IsMigDeviceHandle.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.IsMigDeviceHandleCalls())
|
|
func (mock *DeviceMock) IsMigDeviceHandleCalls() []struct {
|
|
} {
|
|
var calls []struct {
|
|
}
|
|
mock.lockIsMigDeviceHandle.RLock()
|
|
calls = mock.calls.IsMigDeviceHandle
|
|
mock.lockIsMigDeviceHandle.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// RegisterEvents calls RegisterEventsFunc.
|
|
func (mock *DeviceMock) RegisterEvents(v uint64, eventSet EventSet) Return {
|
|
if mock.RegisterEventsFunc == nil {
|
|
panic("DeviceMock.RegisterEventsFunc: method is nil but Device.RegisterEvents was just called")
|
|
}
|
|
callInfo := struct {
|
|
V uint64
|
|
EventSet EventSet
|
|
}{
|
|
V: v,
|
|
EventSet: eventSet,
|
|
}
|
|
mock.lockRegisterEvents.Lock()
|
|
mock.calls.RegisterEvents = append(mock.calls.RegisterEvents, callInfo)
|
|
mock.lockRegisterEvents.Unlock()
|
|
return mock.RegisterEventsFunc(v, eventSet)
|
|
}
|
|
|
|
// RegisterEventsCalls gets all the calls that were made to RegisterEvents.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.RegisterEventsCalls())
|
|
func (mock *DeviceMock) RegisterEventsCalls() []struct {
|
|
V uint64
|
|
EventSet EventSet
|
|
} {
|
|
var calls []struct {
|
|
V uint64
|
|
EventSet EventSet
|
|
}
|
|
mock.lockRegisterEvents.RLock()
|
|
calls = mock.calls.RegisterEvents
|
|
mock.lockRegisterEvents.RUnlock()
|
|
return calls
|
|
}
|
|
|
|
// SetMigMode calls SetMigModeFunc.
|
|
func (mock *DeviceMock) SetMigMode(Mode int) (Return, Return) {
|
|
if mock.SetMigModeFunc == nil {
|
|
panic("DeviceMock.SetMigModeFunc: method is nil but Device.SetMigMode was just called")
|
|
}
|
|
callInfo := struct {
|
|
Mode int
|
|
}{
|
|
Mode: Mode,
|
|
}
|
|
mock.lockSetMigMode.Lock()
|
|
mock.calls.SetMigMode = append(mock.calls.SetMigMode, callInfo)
|
|
mock.lockSetMigMode.Unlock()
|
|
return mock.SetMigModeFunc(Mode)
|
|
}
|
|
|
|
// SetMigModeCalls gets all the calls that were made to SetMigMode.
|
|
// Check the length with:
|
|
//
|
|
// len(mockedDevice.SetMigModeCalls())
|
|
func (mock *DeviceMock) SetMigModeCalls() []struct {
|
|
Mode int
|
|
} {
|
|
var calls []struct {
|
|
Mode int
|
|
}
|
|
mock.lockSetMigMode.RLock()
|
|
calls = mock.calls.SetMigMode
|
|
mock.lockSetMigMode.RUnlock()
|
|
return calls
|
|
}
|