Function TestTransformValidatorComponent

controllers/transforms_test.go:2459–2764

(t *testing.T)

}

func TestTransformValidatorComponent(t *testing.T) {
	testCases := []struct {
		description   string
		pod           Pod
		cpSpec        *gpuv1.ClusterPolicySpec
		component     string
		expectedPod   Pod
		errorExpected bool
	}{
		{
			description: "no validation init container is a no-op",
			pod:         NewPod(),
			cpSpec:      nil,
			component:   "driver",
			expectedPod: NewPod(),
		},
		{
			description: "invalid component",
			pod:         NewPod().WithInitContainer(corev1.Container{Name: "invalid-validation"}),
			cpSpec: &gpuv1.ClusterPolicySpec{
				Validator: gpuv1.ValidatorSpec{},
			},
			component:     "invalid",
			expectedPod:   NewPod(),
			errorExpected: true,
		},
		{
			description: "cuda validation",
			pod: NewPod().
				WithInitContainer(corev1.Container{Name: "cuda-validation"}).
				WithRuntimeClassName("nvidia"),
			cpSpec: &gpuv1.ClusterPolicySpec{
				Validator: gpuv1.ValidatorSpec{
					Repository:       "nvcr.io/nvidia/cloud-native",
					Image:            "gpu-operator-validator",
					Version:          "v1.0.0",
					ImagePullPolicy:  "IfNotPresent",
					ImagePullSecrets: []string{"pull-secret1", "pull-secret2"},
					CUDA: gpuv1.CUDAValidatorSpec{
						Env: []gpuv1.EnvVar{{Name: "foo", Value: "bar"}},
					},
				},
			},
			component: "cuda",
			expectedPod: NewPod().WithInitContainer(corev1.Container{
				Name:            "cuda-validation",
				Image:           "nvcr.io/nvidia/cloud-native/gpu-operator-validator:v1.0.0",
				ImagePullPolicy: corev1.PullIfNotPresent,
				Env: []corev1.EnvVar{
					{Name: ValidatorImageEnvName, Value: "nvcr.io/nvidia/cloud-native/gpu-operator-validator:v1.0.0"},
					{Name: ValidatorImagePullPolicyEnvName, Value: "IfNotPresent"},
					{Name: ValidatorImagePullSecretsEnvName, Value: "pull-secret1,pull-secret2"},
					{Name: ValidatorRuntimeClassEnvName, Value: "nvidia"},
					{Name: "foo", Value: "bar"},
				},
				SecurityContext: &corev1.SecurityContext{
					RunAsUser: rootUID,
				},