Function TestTransformToolkitCtrForCDI

controllers/transforms_test.go:3205–3261

(t *testing.T)

}

func TestTransformToolkitCtrForCDI(t *testing.T) {
	testCases := []struct {
		description string
		ds          Daemonset
		cpSpec      *gpuv1.ClusterPolicySpec
		expectedDs  Daemonset
	}{
		{
			description: "cdi enabled",
			ds:          NewDaemonset().WithContainer(corev1.Container{Name: "main-ctr"}),
			cpSpec: &gpuv1.ClusterPolicySpec{
				CDI: gpuv1.CDIConfigSpec{
					Enabled: newBoolPtr(true),
				},
			},
			expectedDs: NewDaemonset().WithContainer(
				corev1.Container{
					Name: "main-ctr",
					Env: []corev1.EnvVar{
						{Name: CDIEnabledEnvName, Value: "true"},
						{Name: NvidiaRuntimeSetAsDefaultEnvName, Value: "false"},
						{Name: NvidiaCtrRuntimeModeEnvName, Value: "cdi"},
						{Name: CRIOConfigModeEnvName, Value: "config"},
					},
				}),
		},
		{
			description: "cdi and nri plugin enabled",
			ds:          NewDaemonset().WithContainer(corev1.Container{Name: "main-ctr"}),
			cpSpec: &gpuv1.ClusterPolicySpec{
				CDI: gpuv1.CDIConfigSpec{
					Enabled:          newBoolPtr(true),
					NRIPluginEnabled: newBoolPtr(true),
				},
			},
			expectedDs: NewDaemonset().WithContainer(
				corev1.Container{
					Name: "main-ctr",
					Env: []corev1.EnvVar{
						{Name: CDIEnabledEnvName, Value: "true"},
						{Name: NvidiaRuntimeSetAsDefaultEnvName, Value: "false"},
						{Name: NvidiaCtrRuntimeModeEnvName, Value: "cdi"},
						{Name: CRIOConfigModeEnvName, Value: "config"},
						{Name: CDIEnableNRIPlugin, Value: "true"},
					},
				}),
		},
	}

	for _, tc := range testCases {
		t.Run(tc.description, func(t *testing.T) {
			mainContainer := &tc.ds.Spec.Template.Spec.Containers[0]
			transformToolkitCtrForCDI(mainContainer, tc.cpSpec.CDI.IsNRIPluginEnabled())
			require.EqualValues(t, tc.expectedDs, tc.ds)
		})
	}
}