Function TestTransformVGPUDeviceManager

controllers/transforms_test.go:2213–2281

(t *testing.T)

}

func TestTransformVGPUDeviceManager(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"}}

	testCases := []struct {
		description       string
		daemonset         Daemonset
		clusterPolicySpec *gpuv1.ClusterPolicySpec
		expectedDaemonset Daemonset
	}{
		{
			description: "transform vgpu device manager",
			daemonset: NewDaemonset().
				WithContainer(corev1.Container{Name: "nvidia-vgpu-device-manager"}).
				WithContainer(corev1.Container{Name: "sidecar"}),
			clusterPolicySpec: &gpuv1.ClusterPolicySpec{
				VGPUDeviceManager: gpuv1.VGPUDeviceManagerSpec{
					Repository:       "nvcr.io/nvidia/cloud-native",
					Image:            "vgpu-device-manager",
					Version:          "v1.0.0",
					ImagePullPolicy:  "IfNotPresent",
					ImagePullSecrets: []string{secret},
					Resources:        &gpuv1.ResourceRequirements{Limits: resources.Limits, Requests: resources.Requests},
					Args:             []string{"--test-flag"},
					Env:              mockEnv,
					Config: &gpuv1.VGPUDevicesConfigSpec{
						Name:    "custom-vgpu-config",
						Default: "perf",
					},
				},
			},
			expectedDaemonset: NewDaemonset().
				WithContainer(corev1.Container{
					Name:            "nvidia-vgpu-device-manager",
					Image:           "nvcr.io/nvidia/cloud-native/vgpu-device-manager:v1.0.0",
					ImagePullPolicy: corev1.PullIfNotPresent,
					Args:            []string{"--test-flag"},
					Env: []corev1.EnvVar{
						{Name: "foo", Value: "bar"},
						{Name: "DEFAULT_VGPU_CONFIG", Value: "perf"},
					},
					Resources: resources,
				}).
				WithContainer(corev1.Container{
					Name:      "sidecar",
					Resources: resources,
				}).
				WithPullSecret(secret),