nvidia-container-toolkit/vendor/github.com/NVIDIA/go-nvml/pkg/nvml/mock/gpmsample.go

// Code generated by moq; DO NOT EDIT.
// github.com/matryer/moq

package mock

import (
	"github.com/NVIDIA/go-nvml/pkg/nvml"
	"sync"
)

// Ensure, that GpmSample does implement nvml.GpmSample.
// If this is not the case, regenerate this file with moq.
var _ nvml.GpmSample = &GpmSample{}

// GpmSample is a mock implementation of nvml.GpmSample.
//
//	func TestSomethingThatUsesGpmSample(t *testing.T) {
//
//		// make and configure a mocked nvml.GpmSample
//		mockedGpmSample := &GpmSample{
//			FreeFunc: func() nvml.Return {
//				panic("mock out the Free method")
//			},
//			GetFunc: func(device nvml.Device) nvml.Return {
//				panic("mock out the Get method")
//			},
//			MigGetFunc: func(device nvml.Device, n int) nvml.Return {
//				panic("mock out the MigGet method")
//			},
//		}
//
//		// use mockedGpmSample in code that requires nvml.GpmSample
//		// and then make assertions.
//
//	}
type GpmSample struct {
	// FreeFunc mocks the Free method.
	FreeFunc func() nvml.Return

	// GetFunc mocks the Get method.
	GetFunc func(device nvml.Device) nvml.Return

	// MigGetFunc mocks the MigGet method.
	MigGetFunc func(device nvml.Device, n int) nvml.Return

	// calls tracks calls to the methods.
	calls struct {
		// Free holds details about calls to the Free method.
		Free []struct {
		}
		// Get holds details about calls to the Get method.
		Get []struct {
			// Device is the device argument value.
			Device nvml.Device
		}
		// MigGet holds details about calls to the MigGet method.
		MigGet []struct {
			// Device is the device argument value.
			Device nvml.Device
			// N is the n argument value.
			N int
		}
	}
	lockFree   sync.RWMutex
	lockGet    sync.RWMutex
	lockMigGet sync.RWMutex
}

// Free calls FreeFunc.
func (mock *GpmSample) Free() nvml.Return {
	if mock.FreeFunc == nil {
		panic("GpmSample.FreeFunc: method is nil but GpmSample.Free was just called")
	}
	callInfo := struct {
	}{}
	mock.lockFree.Lock()
	mock.calls.Free = append(mock.calls.Free, callInfo)
	mock.lockFree.Unlock()
	return mock.FreeFunc()
}

// FreeCalls gets all the calls that were made to Free.
// Check the length with:
//
//	len(mockedGpmSample.FreeCalls())
func (mock *GpmSample) FreeCalls() []struct {
} {
	var calls []struct {
	}
	mock.lockFree.RLock()
	calls = mock.calls.Free
	mock.lockFree.RUnlock()
	return calls
}

// Get calls GetFunc.
func (mock *GpmSample) Get(device nvml.Device) nvml.Return {
	if mock.GetFunc == nil {
		panic("GpmSample.GetFunc: method is nil but GpmSample.Get was just called")
	}
	callInfo := struct {
		Device nvml.Device
	}{
		Device: device,
	}
	mock.lockGet.Lock()
	mock.calls.Get = append(mock.calls.Get, callInfo)
	mock.lockGet.Unlock()
	return mock.GetFunc(device)
}

// GetCalls gets all the calls that were made to Get.
// Check the length with:
//
//	len(mockedGpmSample.GetCalls())
func (mock *GpmSample) GetCalls() []struct {
	Device nvml.Device
} {
	var calls []struct {
		Device nvml.Device
	}
	mock.lockGet.RLock()
	calls = mock.calls.Get
	mock.lockGet.RUnlock()
	return calls
}

// MigGet calls MigGetFunc.
func (mock *GpmSample) MigGet(device nvml.Device, n int) nvml.Return {
	if mock.MigGetFunc == nil {
		panic("GpmSample.MigGetFunc: method is nil but GpmSample.MigGet was just called")
	}
	callInfo := struct {
		Device nvml.Device
		N      int
	}{
		Device: device,
		N:      n,
	}
	mock.lockMigGet.Lock()
	mock.calls.MigGet = append(mock.calls.MigGet, callInfo)
	mock.lockMigGet.Unlock()
	return mock.MigGetFunc(device, n)
}

// MigGetCalls gets all the calls that were made to MigGet.
// Check the length with:
//
//	len(mockedGpmSample.MigGetCalls())
func (mock *GpmSample) MigGetCalls() []struct {
	Device nvml.Device
	N      int
} {
	var calls []struct {
		Device nvml.Device
		N      int
	}
	mock.lockMigGet.RLock()
	calls = mock.calls.MigGet
	mock.lockMigGet.RUnlock()
	return calls
}