fix

Signed-off-by: Gyuho Lee <gyuhol@nvidia.com>

Author: gyuho

components/accelerator/nvidia/nccl/component_test.go

@@ -174,6 +174,7 @@ func TestCheck(t *testing.T) {
 	t.Run("nvml exists but no product name", func(t *testing.T) {
 		mockNvml := new(mockNVMLInstance)
 		mockNvml.On("NVMLExists").Return(true)
+		mockNvml.On("InitError").Return(nil)
 		mockNvml.On("ProductName").Return("")
 
 		comp := &component{
@@ -201,6 +202,7 @@ func TestCheck(t *testing.T) {
 	t.Run("nil readAllKmsg", func(t *testing.T) {
 		mockNvml := new(mockNVMLInstance)
 		mockNvml.On("NVMLExists").Return(true)
+		mockNvml.On("InitError").Return(nil)
 		mockNvml.On("ProductName").Return("Test GPU")
 		mockNvml.On("Devices").Return(map[string]device.Device{})
 		mockNvml.On("GetMemoryErrorManagementCapabilities").Return(nvidiaproduct.MemoryErrorManagementCapabilities{})
@@ -218,6 +220,7 @@ func TestCheck(t *testing.T) {
 	t.Run("readAllKmsg returns error", func(t *testing.T) {
 		mockNvml := new(mockNVMLInstance)
 		mockNvml.On("NVMLExists").Return(true)
+		mockNvml.On("InitError").Return(nil)
 		mockNvml.On("ProductName").Return("Test GPU")
 		mockNvml.On("Devices").Return(map[string]device.Device{})
 		mockNvml.On("GetMemoryErrorManagementCapabilities").Return(nvidiaproduct.MemoryErrorManagementCapabilities{})
@@ -240,6 +243,7 @@ func TestCheck(t *testing.T) {
 	t.Run("no matching messages", func(t *testing.T) {
 		mockNvml := new(mockNVMLInstance)
 		mockNvml.On("NVMLExists").Return(true)
+		mockNvml.On("InitError").Return(nil)
 		mockNvml.On("ProductName").Return("Test GPU")
 		mockNvml.On("Devices").Return(map[string]device.Device{})
 		mockNvml.On("GetMemoryErrorManagementCapabilities").Return(nvidiaproduct.MemoryErrorManagementCapabilities{})
@@ -266,6 +270,7 @@ func TestCheck(t *testing.T) {
 	t.Run("with matching messages", func(t *testing.T) {
 		mockNvml := new(mockNVMLInstance)
 		mockNvml.On("NVMLExists").Return(true)
+		mockNvml.On("InitError").Return(nil)
 		mockNvml.On("ProductName").Return("Test GPU")
 		mockNvml.On("Devices").Return(map[string]device.Device{})
 		mockNvml.On("GetMemoryErrorManagementCapabilities").Return(nvidiaproduct.MemoryErrorManagementCapabilities{})
@@ -577,6 +582,7 @@ func TestCheck_NVML_NoProductName(t *testing.T) {
 
 	mockNvml := new(mockNVMLInstance)
 	mockNvml.On("NVMLExists").Return(true)
+	mockNvml.On("InitError").Return(nil)
 	mockNvml.On("ProductName").Return("")
 
 	comp := &component{