@@ -614,6 +614,10 @@ func TestTransformToolkit(t *testing.T) {
614614 },
615615 },
616616 Env : []corev1.EnvVar {
617+ {Name : CDIEnabledEnvName , Value : "true" },
618+ {Name : CrioConfigModeEnvName , Value : "config" },
619+ {Name : NvidiaRuntimeSetAsDefaultEnvName , Value : "false" },
620+ {Name : NvidiaCtrRuntimeModeEnvName , Value : "cdi" },
617621 {Name : "RUNTIME" , Value : "containerd" },
618622 {Name : "CONTAINERD_RUNTIME_CLASS" , Value : "nvidia" },
619623 {Name : "RUNTIME_CONFIG" , Value : "/runtime/config-dir/config.toml" },
@@ -671,6 +675,10 @@ func TestTransformDevicePlugin(t *testing.T) {
671675 {Name : "foo" , Value : "bar" },
672676 },
673677 },
678+ Toolkit : gpuv1.ToolkitSpec {
679+ Enabled : newBoolPtr (true ),
680+ InstallDir : "/path/to/install" ,
681+ },
674682 },
675683 expectedDs : NewDaemonset ().WithContainer (corev1.Container {
676684 Name : "nvidia-device-plugin-ctr" ,
@@ -679,6 +687,10 @@ func TestTransformDevicePlugin(t *testing.T) {
679687 Args : []string {"--fail-on-init-error=false" },
680688 Env : []corev1.EnvVar {
681689 {Name : "NVIDIA_MIG_MONITOR_DEVICES" , Value : "all" },
690+ {Name : CDIEnabledEnvName , Value : "true" },
691+ {Name : DeviceListStrategyEnvName , Value : "cdi-annotations,cdi-cri" },
692+ {Name : CDIAnnotationPrefixEnvName , Value : "cdi.k8s.io/" },
693+ {Name : NvidiaCDIHookPathEnvName , Value : "/path/to/install/toolkit/nvidia-cdi-hook" },
682694 {Name : "foo" , Value : "bar" },
683695 },
684696 }).WithContainer (corev1.Container {Name : "dummy" }).WithPullSecret ("pull-secret" ).WithRuntimeClassName ("nvidia" ),
@@ -768,13 +780,19 @@ func TestTransformMigManager(t *testing.T) {
768780 {Name : "foo" , Value : "bar" },
769781 },
770782 },
783+ Toolkit : gpuv1.ToolkitSpec {
784+ Enabled : newBoolPtr (true ),
785+ InstallDir : "/path/to/install" ,
786+ },
771787 },
772788 expectedDs : NewDaemonset ().WithContainer (corev1.Container {
773789 Name : "mig-manager" ,
774790 Image : "nvcr.io/nvidia/cloud-native/mig-manager:v1.0.0" ,
775791 ImagePullPolicy : corev1 .PullIfNotPresent ,
776792 Args : []string {"--test-flag" },
777793 Env : []corev1.EnvVar {
794+ {Name : CDIEnabledEnvName , Value : "true" },
795+ {Name : NvidiaCDIHookPathEnvName , Value : "/path/to/install/toolkit/nvidia-cdi-hook" },
778796 {Name : "foo" , Value : "bar" },
779797 },
780798 }).WithPullSecret ("pull-secret" ).WithRuntimeClassName ("nvidia" ),
0 commit comments