feat: wires up GPUs to mutation handler w/ tests

Author: faiq

pkg/handlers/nutanix/mutation/machinedetails/inject.go

@@ -122,6 +122,24 @@ func (h *nutanixMachineDetailsPatchHandler) Mutate(
 					capxv1.NutanixResourceIdentifier(*nutanixMachineDetailsVar.Project),
 				)
 			}
+			if nutanixMachineDetailsVar.GPUs != nil {
+				spec.GPUs = make(
+					[]capxv1.NutanixGPU,
+					len(nutanixMachineDetailsVar.GPUs),
+				)
+
+				for i, gpu := range nutanixMachineDetailsVar.GPUs {
+					gpuType := capxv1.NutanixGPUIdentifierType(gpu.Type)
+					spec.GPUs[i] = capxv1.NutanixGPU{
+						Type: gpuType,
+					}
+					if gpuType == capxv1.NutanixGPUIdentifierName {
+						spec.GPUs[i].Name = gpu.Name
+						continue
+					}
+					spec.GPUs[i].DeviceID = gpu.DeviceID
+				}
+			}
 
 			obj.Spec.Template.Spec = spec
 			return nil

pkg/handlers/nutanix/mutation/machinedetails/inject_control_plane_test.go

@@ -55,9 +55,33 @@ var (
 			Type: capxv1.NutanixIdentifierName,
 			Name: ptr.To("fake-project"),
 		}),
+		GPUs: []v1alpha1.NutanixGPU{
+			{
+				Type: "name",
+				Name: ptr.To("gpu1"),
+			},
+			{
+				Type:     "deviceID",
+				DeviceID: ptr.To(int64(1)),
+			},
+		},
 	}
 
 	matchersForAllFieldsSet = []capitest.JSONPatchMatcher{
+		{
+			Operation: "add",
+			Path:      "/spec/template/spec/gpus",
+			ValueMatcher: gomega.ContainElements(
+				gomega.SatisfyAll(
+					gomega.HaveKeyWithValue("type", "name"),
+					gomega.HaveKeyWithValue("name", "gpu1"),
+				),
+				gomega.SatisfyAll(
+					gomega.HaveKeyWithValue("type", "deviceID"),
+					gomega.HaveKey("deviceID"),
+				),
+			),
+		},
 		{
 			Operation:    "add",
 			Path:         "/spec/template/spec/bootType",