Function TestTransformToolkit

controllers/transforms_test.go:848–1089

(t *testing.T)

}

func TestTransformToolkit(t *testing.T) {
	testCases := []struct {
		description string
		ds          Daemonset                // Input DaemonSet
		cpSpec      *gpuv1.ClusterPolicySpec // Input configuration
		runtime     gpuv1.Runtime
		expectedDs  Daemonset // Expected output DaemonSet
	}{
		{
			description: "transform nvidia-container-toolkit-ctr container",
			ds: NewDaemonset().
				WithContainer(corev1.Container{Name: "nvidia-container-toolkit-ctr"}),
			runtime: gpuv1.Containerd,
			cpSpec: &gpuv1.ClusterPolicySpec{
				Toolkit: gpuv1.ToolkitSpec{
					Repository:       "nvcr.io/nvidia/cloud-native",
					Image:            "nvidia-container-toolkit",
					Version:          "v1.0.0",
					ImagePullPolicy:  "IfNotPresent",
					ImagePullSecrets: []string{"pull-secret"},
					Resources: &gpuv1.ResourceRequirements{
						Limits: corev1.ResourceList{
							corev1.ResourceCPU:    resource.MustParse("100m"),
							corev1.ResourceMemory: resource.MustParse("100Mi"),
						},
						Requests: corev1.ResourceList{
							corev1.ResourceCPU:    resource.MustParse("50m"),
							corev1.ResourceMemory: resource.MustParse("50Mi"),
						},
					},
					Env: []gpuv1.EnvVar{
						{Name: "foo", Value: "bar"},
					},
				},
			},
			expectedDs: NewDaemonset().
				WithContainer(corev1.Container{
					Name:            "nvidia-container-toolkit-ctr",
					Image:           "nvcr.io/nvidia/cloud-native/nvidia-container-toolkit:v1.0.0",
					ImagePullPolicy: corev1.PullIfNotPresent,
					Resources: corev1.ResourceRequirements{
						Limits: corev1.ResourceList{
							corev1.ResourceCPU:    resource.MustParse("100m"),
							corev1.ResourceMemory: resource.MustParse("100Mi"),
						},
						Requests: corev1.ResourceList{
							corev1.ResourceCPU:    resource.MustParse("50m"),
							corev1.ResourceMemory: resource.MustParse("50Mi"),
						},
					},
					Env: []corev1.EnvVar{
						{Name: CDIEnabledEnvName, Value: "true"},
						{Name: NvidiaRuntimeSetAsDefaultEnvName, Value: "false"},
						{Name: NvidiaCtrRuntimeModeEnvName, Value: "cdi"},
						{Name: CRIOConfigModeEnvName, Value: "config"},
						{Name: "foo", Value: "bar"},
						{Name: "RUNTIME", Value: "containerd"},
						{Name: "CONTAINERD_RUNTIME_CLASS", Value: "nvidia"},