Function TestTransformCCManager

controllers/transforms_test.go:2145–2211

(t *testing.T)

}

func TestTransformCCManager(t *testing.T) {
	resources := corev1.ResourceRequirements{
		Limits: corev1.ResourceList{
			corev1.ResourceCPU:    resource.MustParse("100m"),
			corev1.ResourceMemory: resource.MustParse("128Mi"),
		},
		Requests: corev1.ResourceList{
			corev1.ResourceCPU:    resource.MustParse("50m"),
			corev1.ResourceMemory: resource.MustParse("64Mi"),
		},
	}
	secret := "pull-secret"
	mockEnv := []gpuv1.EnvVar{{Name: "foo", Value: "bar"}}
	defaultMode := "devtools"

	testCases := []struct {
		description       string
		daemonset         Daemonset
		clusterPolicySpec *gpuv1.ClusterPolicySpec
		expectedDaemonset Daemonset
	}{
		{
			description: "transform cc manager",
			daemonset: NewDaemonset().
				WithContainer(corev1.Container{Name: "nvidia-cc-manager"}).
				WithContainer(corev1.Container{Name: "sidecar"}),
			clusterPolicySpec: &gpuv1.ClusterPolicySpec{
				CCManager: gpuv1.CCManagerSpec{
					Repository:       "nvcr.io/nvidia/cloud-native",
					Image:            "k8s-cc-manager",
					Version:          "v1.0.0",
					ImagePullPolicy:  "IfNotPresent",
					ImagePullSecrets: []string{secret},
					Resources:        &gpuv1.ResourceRequirements{Limits: resources.Limits, Requests: resources.Requests},
					Args:             []string{"--test-flag"},
					DefaultMode:      defaultMode,
					Env:              mockEnv,
				},
			},
			expectedDaemonset: NewDaemonset().
				WithContainer(corev1.Container{
					Name:            "nvidia-cc-manager",
					Image:           "nvcr.io/nvidia/cloud-native/k8s-cc-manager:v1.0.0",
					ImagePullPolicy: corev1.PullIfNotPresent,
					Args:            []string{"--test-flag"},
					Env: []corev1.EnvVar{
						{Name: "DEFAULT_CC_MODE", Value: defaultMode},
						{Name: "foo", Value: "bar"},
					},
					Resources: resources,
				}).
				WithContainer(corev1.Container{
					Name:      "sidecar",
					Resources: resources,
				}).
				WithPullSecret(secret),
		},
	}