Function TestTransformValidationInitContainer

controllers/transforms_test.go:2283–2338

(t *testing.T)

}

func TestTransformValidationInitContainer(t *testing.T) {
	testCases := []struct {
		description string
		ds          Daemonset
		cpSpec      *gpuv1.ClusterPolicySpec
		expectedDs  Daemonset
	}{
		{
			description: "transform both driver and toolkit validation initContainers",
			ds: NewDaemonset().
				WithInitContainer(corev1.Container{Name: "driver-validation"}).
				WithInitContainer(corev1.Container{Name: "toolkit-validation"}).
				WithInitContainer(corev1.Container{Name: "dummy"}),
			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-secret"},
					Driver: gpuv1.DriverValidatorSpec{
						Env: []gpuv1.EnvVar{{Name: "foo", Value: "bar"}},
					},
					Toolkit: gpuv1.ToolkitValidatorSpec{
						Env: []gpuv1.EnvVar{{Name: "foo", Value: "bar"}},
					},
				},
			},
			expectedDs: NewDaemonset().WithInitContainer(corev1.Container{
				Name:            "driver-validation",
				Image:           "nvcr.io/nvidia/cloud-native/gpu-operator-validator:v1.0.0",
				ImagePullPolicy: corev1.PullIfNotPresent,
				Env:             []corev1.EnvVar{{Name: "foo", Value: "bar"}},
				SecurityContext: &corev1.SecurityContext{
					RunAsUser: rootUID,
				},
			}).WithInitContainer(corev1.Container{
				Name:            "toolkit-validation",
				Image:           "nvcr.io/nvidia/cloud-native/gpu-operator-validator:v1.0.0",
				ImagePullPolicy: corev1.PullIfNotPresent,
				Env:             []corev1.EnvVar{{Name: "foo", Value: "bar"}},
				SecurityContext: &corev1.SecurityContext{
					RunAsUser: rootUID,
				},
			}).WithInitContainer(corev1.Container{Name: "dummy"}).WithPullSecret("pull-secret"),
		},
	}

	for _, tc := range testCases {
		t.Run(tc.description, func(t *testing.T) {
			err := transformValidationInitContainer(tc.ds.DaemonSet, tc.cpSpec)
			require.NoError(t, err)
			require.EqualValues(t, tc.expectedDs, tc.ds)
		})
	}
}

func newBoolPtr(b bool) *bool {