test(nvml): cover no-op and errored instances

Author: gyuho

pkg/nvidia-query/nvml/instance_test.go

@@ -4,6 +4,7 @@ import (
 	"errors"
 	"testing"
 
+	nvidiaproduct "github.com/leptonai/gpud/pkg/nvidia/product"
 	nvmllib "github.com/leptonai/gpud/pkg/nvidia-query/nvml/lib"
 )
 
@@ -17,3 +18,98 @@ func TestInstanceV2(t *testing.T) {
 	}
 	t.Logf("instance mem cap %+v", inst.GetMemoryErrorManagementCapabilities())
 }
+
+func TestNewNoOpInstance(t *testing.T) {
+	inst := NewNoOp()
+
+	if inst.NVMLExists() {
+		t.Fatalf("expected NVMLExists to be false")
+	}
+	if inst.Library() != nil {
+		t.Fatalf("expected nil library")
+	}
+	if inst.Devices() != nil {
+		t.Fatalf("expected nil devices map")
+	}
+	if inst.ProductName() != "" {
+		t.Fatalf("expected empty product name")
+	}
+	if inst.Architecture() != "" {
+		t.Fatalf("expected empty architecture")
+	}
+	if inst.Brand() != "" {
+		t.Fatalf("expected empty brand")
+	}
+	if inst.DriverVersion() != "" {
+		t.Fatalf("expected empty driver version")
+	}
+	if inst.DriverMajor() != 0 {
+		t.Fatalf("expected driver major 0")
+	}
+	if inst.CUDAVersion() != "" {
+		t.Fatalf("expected empty CUDA version")
+	}
+	if inst.FabricManagerSupported() {
+		t.Fatalf("expected FabricManagerSupported false")
+	}
+	if inst.FabricStateSupported() {
+		t.Fatalf("expected FabricStateSupported false")
+	}
+	if inst.GetMemoryErrorManagementCapabilities() != (nvidiaproduct.MemoryErrorManagementCapabilities{}) {
+		t.Fatalf("expected empty memory error management capabilities")
+	}
+	if err := inst.Shutdown(); err != nil {
+		t.Fatalf("expected Shutdown to return nil, got %v", err)
+	}
+	if inst.InitError() != nil {
+		t.Fatalf("expected InitError nil")
+	}
+}
+
+func TestNewErroredInstance(t *testing.T) {
+	initErr := errors.New("nvml init failed")
+	inst := NewErrored(initErr)
+
+	if !inst.NVMLExists() {
+		t.Fatalf("expected NVMLExists to be true")
+	}
+	if inst.Library() != nil {
+		t.Fatalf("expected nil library")
+	}
+	if inst.Devices() != nil {
+		t.Fatalf("expected nil devices map")
+	}
+	if inst.ProductName() != "" {
+		t.Fatalf("expected empty product name")
+	}
+	if inst.Architecture() != "" {
+		t.Fatalf("expected empty architecture")
+	}
+	if inst.Brand() != "" {
+		t.Fatalf("expected empty brand")
+	}
+	if inst.DriverVersion() != "" {
+		t.Fatalf("expected empty driver version")
+	}
+	if inst.DriverMajor() != 0 {
+		t.Fatalf("expected driver major 0")
+	}
+	if inst.CUDAVersion() != "" {
+		t.Fatalf("expected empty CUDA version")
+	}
+	if inst.FabricManagerSupported() {
+		t.Fatalf("expected FabricManagerSupported false")
+	}
+	if inst.FabricStateSupported() {
+		t.Fatalf("expected FabricStateSupported false")
+	}
+	if inst.GetMemoryErrorManagementCapabilities() != (nvidiaproduct.MemoryErrorManagementCapabilities{}) {
+		t.Fatalf("expected empty memory error management capabilities")
+	}
+	if err := inst.Shutdown(); err != nil {
+		t.Fatalf("expected Shutdown to return nil, got %v", err)
+	}
+	if !errors.Is(inst.InitError(), initErr) {
+		t.Fatalf("expected InitError %v, got %v", initErr, inst.InitError())
+	}
+}