Function TestTransformMPSControlDaemon

controllers/transforms_test.go:1187–1242

(t *testing.T)

}

func TestTransformMPSControlDaemon(t *testing.T) {
	testCases := []struct {
		description       string
		daemonset         Daemonset
		clusterPolicySpec *gpuv1.ClusterPolicySpec
		expectedDaemonset Daemonset
	}{
		{
			description: "transform mps control daemon",
			daemonset: NewDaemonset().
				WithInitContainer(corev1.Container{Name: "mps-control-daemon-mounts"}).
				WithContainer(corev1.Container{Name: "mps-control-daemon-ctr"}).
				WithHostPathVolume("mps-root", "/run/nvidia/mps", ptr.To(corev1.HostPathDirectoryOrCreate)).
				WithHostPathVolume("mps-shm", "/run/nvidia/mps/shm", ptr.To(corev1.HostPathDirectoryOrCreate)),
			clusterPolicySpec: &gpuv1.ClusterPolicySpec{
				DevicePlugin: gpuv1.DevicePluginSpec{
					Repository:       "nvcr.io",
					Image:            "mps",
					Version:          "latest",
					ImagePullPolicy:  string(corev1.PullAlways),
					ImagePullSecrets: []string{"secret"},
					MPS:              &gpuv1.MPSConfig{Root: "/var/mps"},
				},
			},
			expectedDaemonset: NewDaemonset().
				WithInitContainer(corev1.Container{
					Name:            "mps-control-daemon-mounts",
					Image:           "nvcr.io/mps:latest",
					ImagePullPolicy: corev1.PullAlways,
				}).
				WithContainer(corev1.Container{
					Name:            "mps-control-daemon-ctr",
					Image:           "nvcr.io/mps:latest",
					ImagePullPolicy: corev1.PullAlways,
					Env: []corev1.EnvVar{
						{Name: "NVIDIA_MIG_MONITOR_DEVICES", Value: "all"},
					},
				}).
				WithHostPathVolume("mps-root", "/var/mps", ptr.To(corev1.HostPathDirectoryOrCreate)).
				WithHostPathVolume("mps-shm", "/var/mps/shm", ptr.To(corev1.HostPathDirectoryOrCreate)).
				WithPullSecret("secret").
				WithRuntimeClassName("nvidia"),
		},
	}

	for _, tc := range testCases {
		t.Run(tc.description, func(t *testing.T) {
			err := TransformMPSControlDaemon(tc.daemonset.DaemonSet, tc.clusterPolicySpec, ClusterPolicyController{
				runtime: gpuv1.Containerd,
				logger:  ctrl.Log.WithName("test"),
			})
			require.NoError(t, err)
			require.EqualValues(t, tc.expectedDaemonset.DaemonSet, tc.daemonset.DaemonSet)
		})
	}
}

func TestTransformDCGMExporter(t *testing.T) {