Add test for missing toolkit container

Signed-off-by: Evan Lezar <elezar@nvidia.com>

Author: elezar

controllers/transforms_test.go

@@ -17,6 +17,7 @@
 package controllers
 
 import (
+	"errors"
 	"path/filepath"
 	"testing"
 
@@ -777,11 +778,12 @@ func TestApplyCommonDaemonsetMetadata(t *testing.T) {
 
 func TestTransformToolkit(t *testing.T) {
 	testCases := []struct {
-		description string
-		ds          Daemonset                // Input DaemonSet
-		cpSpec      *gpuv1.ClusterPolicySpec // Input configuration
-		runtime     gpuv1.Runtime
-		expectedDs  Daemonset // Expected output DaemonSet
+		description   string
+		ds            Daemonset                // Input DaemonSet
+		cpSpec        *gpuv1.ClusterPolicySpec // Input configuration
+		runtime       gpuv1.Runtime
+		expectedError error
+		expectedDs    Daemonset // Expected output DaemonSet
 	}{
 		{
 			description: "transform nvidia-container-toolkit-ctr container",
@@ -1032,6 +1034,12 @@ func TestTransformToolkit(t *testing.T) {
 					VolumeMounts: nil,
 				}),
 		},
+		{
+			description:   "no nvidia-container-toolkit-ctr container",
+			ds:            NewDaemonset(),
+			expectedError: errors.New(`failed to find toolkit container "nvidia-container-toolkit-ctr"`),
+			expectedDs:    NewDaemonset(),
+		},
 	}
 
 	for _, tc := range testCases {
@@ -1042,7 +1050,7 @@ func TestTransformToolkit(t *testing.T) {
 			}
 
 			err := TransformToolkit(tc.ds.DaemonSet, tc.cpSpec, controller)
-			require.NoError(t, err)
+			require.EqualValues(t, tc.expectedError, err)
 			require.EqualValues(t, tc.expectedDs, tc.ds)
 		})
 	}