Stop accessing SetupDiGetDeviceInfoListDetail() output on error

The data returned by SetupDiGetDeviceInfoListDetail() is nil on error
which will cause the test to crash should the function fail.

Signed-off-by: Simon Rozman <simon@rozman.si>

diff --git a/setupapi/setupapi_windows_test.go b/setupapi/setupapi_windows_test.go
index 062b6bd3e9874e3c9269b111b76369edc77a6c80..17d6d8e36ff04b427d3c64f741ef881b29d6f57b 100644 (file)
--- a/setupapi/setupapi_windows_test.go
+++ b/setupapi/setupapi_windows_test.go
@@ -99,18 +99,18 @@ func TestSetupDiGetDeviceInfoListDetailLocal(t *testing.T) {
        data, err := SetupDiGetDeviceInfoListDetail(devInfoList)
        if err != nil {
                t.Errorf("Error calling SetupDiGetDeviceInfoListDetail: %s", err.Error())
-       }
-
-       if data.ClassGUID != deviceClassNetGUID {
-               t.Error("SetupDiGetDeviceInfoListDetail returned different class GUID")
-       }
+       } else {
+               if data.ClassGUID != deviceClassNetGUID {
+                       t.Error("SetupDiGetDeviceInfoListDetail returned different class GUID")
+               }
 
-       if data.RemoteMachineHandle != windows.Handle(0) {
-               t.Error("SetupDiGetDeviceInfoListDetail returned non-NULL remote machine handle")
-       }
+               if data.RemoteMachineHandle != windows.Handle(0) {
+                       t.Error("SetupDiGetDeviceInfoListDetail returned non-NULL remote machine handle")
+               }
 
-       if data.RemoteMachineName != "" {
-               t.Error("SetupDiGetDeviceInfoListDetail returned non-NULL remote machine name")
+               if data.RemoteMachineName != "" {
+                       t.Error("SetupDiGetDeviceInfoListDetail returned non-NULL remote machine name")
+               }
        }
 }
 
@@ -124,18 +124,18 @@ func TestSetupDiGetDeviceInfoListDetailRemote(t *testing.T) {
        data, err := SetupDiGetDeviceInfoListDetail(devInfoList)
        if err != nil {
                t.Errorf("Error calling SetupDiGetDeviceInfoListDetail: %s", err.Error())
-       }
-
-       if data.ClassGUID != deviceClassNetGUID {
-               t.Error("SetupDiGetDeviceInfoListDetail returned different class GUID")
-       }
+       } else {
+               if data.ClassGUID != deviceClassNetGUID {
+                       t.Error("SetupDiGetDeviceInfoListDetail returned different class GUID")
+               }
 
-       if data.RemoteMachineHandle == windows.Handle(0) {
-               t.Error("SetupDiGetDeviceInfoListDetail returned NULL remote machine handle")
-       }
+               if data.RemoteMachineHandle == windows.Handle(0) {
+                       t.Error("SetupDiGetDeviceInfoListDetail returned NULL remote machine handle")
+               }
 
-       if data.RemoteMachineName != computerName {
-               t.Error("SetupDiGetDeviceInfoListDetail returned different remote machine name")
+               if data.RemoteMachineName != computerName {
+                       t.Error("SetupDiGetDeviceInfoListDetail returned different remote machine name")
+               }
        }
 }