Function TestTransformDriver

controllers/transforms_test.go:3088–3203

(t *testing.T)

}

func TestTransformDriver(t *testing.T) {
	initMockK8sClients()
	testCases := []struct {
		description   string
		ds            Daemonset
		cpSpec        *gpuv1.ClusterPolicySpec
		client        client.Client
		expectedDs    Daemonset
		errorExpected bool
	}{
		{
			description: "driver spec with secret env",
			ds: NewDaemonset().WithContainer(corev1.Container{Name: "nvidia-driver-ctr"}).
				WithContainer(corev1.Container{Name: "nvidia-fs"}).
				WithContainer(corev1.Container{Name: "nvidia-gdrcopy"}).
				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",
					},
					SecretEnv: "test-env-secret",
				},
				GPUDirectStorage: &gpuv1.GPUDirectStorageSpec{
					Enabled:    newBoolPtr(true),
					Repository: "nvcr.io/nvidia/cloud-native",
					Image:      "nvidia-fs",
					Version:    "2.20.5",
				},
				GDRCopy: &gpuv1.GDRCopySpec{
					Enabled:    newBoolPtr(true),
					Repository: "nvcr.io/nvidia/cloud-native",
					Image:      "gdrdrv",
					Version:    "v2.5",
				},
			},
			client: mockClientMap["secret-env-client"],
			expectedDs: NewDaemonset().WithContainer(corev1.Container{
				Name:            "nvidia-driver-ctr",
				Image:           "nvcr.io/nvidia/driver:570.172.08-ubuntu20.04",
				ImagePullPolicy: corev1.PullIfNotPresent,
				EnvFrom: []corev1.EnvFromSource{{
					SecretRef: &corev1.SecretEnvSource{
						LocalObjectReference: corev1.LocalObjectReference{
							Name: "test-env-secret",
						},
					},
				}},
				Env: []corev1.EnvVar{
					{
						Name:  "GDRCOPY_ENABLED",
						Value: "true",
					},