Function TestTransformDriverRDMA

controllers/transforms_test.go:3751–3838

(t *testing.T)

}

func TestTransformDriverRDMA(t *testing.T) {
	node := &corev1.Node{
		ObjectMeta: metav1.ObjectMeta{
			Name: "test-node",
			Labels: map[string]string{
				nfdOSReleaseIDLabelKey: "ubuntu",
				nfdOSVersionIDLabelKey: "20.04",
				nfdKernelLabelKey:      "6.8.0-60-generic",
				commonGPULabelKey:      "true",
			},
		},
	}
	mockClient := fake.NewFakeClient(node)
	ds := NewDaemonset().WithContainer(corev1.Container{Name: "nvidia-driver-ctr"}).
		WithContainer(corev1.Container{Name: "nvidia-fs"}).
		WithContainer(corev1.Container{Name: "nvidia-gdrcopy"}).
		WithContainer(corev1.Container{Name: "nvidia-peermem"}).
		WithInitContainer(corev1.Container{Name: "k8s-driver-manager"})
	cpSpec := &gpuv1.ClusterPolicySpec{
		Driver: gpuv1.DriverSpec{
			Repository: "nvcr.io/nvidia",
			Image:      "driver",
			Version:    "570.172.08",
			Manager: gpuv1.DriverManagerSpec{
				Repository: "nvcr.io/nvidia/cloud-native",
				Image:      "k8s-driver-manager",
				Version:    "v0.8.0",
			},
			GPUDirectRDMA: &gpuv1.GPUDirectRDMASpec{
				Enabled:      newBoolPtr(true),
				UseHostMOFED: newBoolPtr(true),
			},
		},
	}

	expectedDs := NewDaemonset().WithContainer(corev1.Container{
		Name:            "nvidia-driver-ctr",
		Image:           "nvcr.io/nvidia/driver:570.172.08-ubuntu20.04",
		ImagePullPolicy: corev1.PullIfNotPresent,
		Env: []corev1.EnvVar{
			{
				Name:  "GPU_DIRECT_RDMA_ENABLED",
				Value: "true",
			},
			{
				Name:  "USE_HOST_MOFED",
				Value: "true",
			},
			{
				Name:  "DRIVER_CONFIG_DIGEST",
				Value: "785503300",
			},
		},
	}).WithInitContainer(corev1.Container{
		Name:  "k8s-driver-manager",
		Image: "nvcr.io/nvidia/cloud-native/k8s-driver-manager:v0.8.0",
		Env: []corev1.EnvVar{
			{