Bus type auto-detection of the VMware importer does not work correct

Related to: harvester/harvester#7987

Signed-off-by: Volker Theile <[email protected]>

Author: votdev

pkg/source/vmware/client.go

@@ -492,14 +492,34 @@ func mapNetworkCards(networkCards []networkInfo, mapping []migration.NetworkMapp
 	return retNetwork
 }
 
-// adapterType tries to identify the disk bus type from vmware
-// to attempt and set correct bus types in kubevirt
+// adapterType tries to identify the disk bus type from VMware to attempt and
+// set correct bus types in kubevirt.
+// Examples:
+// .-----------------------------------------------.
+// | Bus  | Device ID                              |
+// |------|----------------------------------------|
+// | SCSI | /vm-13010/ParaVirtualSCSIController0:0 |
+// | SATA | /vm-13767/VirtualAHCIController0:1     |
+// | IDE  | /vm-5678/VirtualIDEController1:0       |
+// | NVMe | /vm-2468/VirtualNVMEController0:0      |
+// | USB  | /vm-54321/VirtualUSBController0:0      |
+// '-----------------------------------------------'
 // default is to switch to SATA to ensure device boots
 func adapterType(deviceID string) kubevirt.DiskBus {
-	if strings.Contains(deviceID, "SCSI") {
+	deviceID = strings.ToLower(deviceID)
+	// https://kubevirt.io/api-reference/v1.1.0/definitions.html#_v1_disktarget
+	switch {
+	case strings.Contains(deviceID, "scsi"):
 		return kubevirt.DiskBusSCSI
+	case strings.Contains(deviceID, "ahci"), strings.Contains(deviceID, "sata"):
+		return kubevirt.DiskBusSATA
+	case strings.Contains(deviceID, "nvme"):
+		return kubevirt.DiskBusVirtio
+	case strings.Contains(deviceID, "usb"):
+		return kubevirt.DiskBusUSB
+	default:
+		return kubevirt.DiskBusVirtio
 	}
-	return kubevirt.DiskBusSATA
 }
 
 // SanitizeVirtualMachineImport is used to sanitize the VirtualMachineImport object.

pkg/source/vmware/client_test.go

@@ -13,6 +13,7 @@ import (
 	"github.com/vmware/govmomi/vim25/mo"
 	corev1 "k8s.io/api/core/v1"
 	metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
+	kubevirt "kubevirt.io/api/core/v1"
 
 	migration "github.com/harvester/vm-import-controller/pkg/apis/migration.harvesterhci.io/v1beta1"
 	"github.com/harvester/vm-import-controller/pkg/server"
@@ -304,3 +305,48 @@ func Test_identifyNetworkCards(t *testing.T) {
 	noMappedInfo := mapNetworkCards(networkInfo, noNetworkMapping)
 	assert.Len(noMappedInfo, 0, "expected to find no item in the mapped networkinfo")
 }
+
+func Test_adapterType(t *testing.T) {
+	assert := require.New(t)
+	testCases := []struct {
+		desc     string
+		deviceID string
+		expected kubevirt.DiskBus
+	}{
+		{
+			desc:     "SCSI disk",
+			deviceID: "/vm-13010/ParaVirtualSCSIController0:0",
+			expected: kubevirt.DiskBusSCSI,
+		},
+		{
+			desc:     "NVMe disk",
+			deviceID: "/vm-2468/VirtualNVMEController0:0",
+			expected: kubevirt.DiskBusVirtio,
+		},
+		{
+			desc:     "USB disk",
+			deviceID: "/vm-54321/VirtualUSBController0:0",
+			expected: kubevirt.DiskBusUSB,
+		},
+		{
+			desc:     "SATA disk",
+			deviceID: "/vm-13767/VirtualAHCIController0:1",
+			expected: kubevirt.DiskBusSATA,
+		},
+		{
+			desc:     "IDE disk",
+			deviceID: "/vm-5678/VirtualIDEController1:0",
+			expected: kubevirt.DiskBusVirtio,
+		},
+		{
+			desc:     "Unknown disk",
+			deviceID: "foo",
+			expected: kubevirt.DiskBusVirtio,
+		},
+	}
+
+	for _, tc := range testCases {
+		busType := adapterType(tc.deviceID)
+		assert.Equal(tc.expected, busType)
+	}
+}