Function TestTransformNodeStatusExporter

controllers/transforms_test.go:3033–3086

(t *testing.T)

}

func TestTransformNodeStatusExporter(t *testing.T) {
	testCases := []struct {
		description   string
		ds            Daemonset
		cpSpec        *gpuv1.ClusterPolicySpec
		expectedDs    Daemonset
		errorExpected bool
	}{
		{
			description: "empty node status exporter spec",
			ds: NewDaemonset().
				WithContainer(corev1.Container{Name: "dummy"}),
			cpSpec: &gpuv1.ClusterPolicySpec{
				NodeStatusExporter: gpuv1.NodeStatusExporterSpec{},
			},
			expectedDs:    NewDaemonset(),
			errorExpected: true,
		},
		{
			description: "valid node status exporter spec",
			ds: NewDaemonset().
				WithContainer(corev1.Container{Name: "dummy"}),
			cpSpec: &gpuv1.ClusterPolicySpec{
				NodeStatusExporter: gpuv1.NodeStatusExporterSpec{
					Repository:      "nvcr.io/nvidia/cloud-native",
					Image:           "node-status-exporter",
					Version:         "v1.0.0",
					ImagePullPolicy: "IfNotPresent",
				},
			},
			expectedDs: NewDaemonset().
				WithContainer(corev1.Container{
					Name:            "dummy",
					Image:           "nvcr.io/nvidia/cloud-native/node-status-exporter:v1.0.0",
					ImagePullPolicy: corev1.PullIfNotPresent,
					SecurityContext: &corev1.SecurityContext{
						RunAsUser: rootUID,
					},
				}),
		},
	}

	for _, tc := range testCases {
		t.Run(tc.description, func(t *testing.T) {
			err := TransformNodeStatusExporter(tc.ds.DaemonSet, tc.cpSpec, ClusterPolicyController{runtime: gpuv1.Containerd, logger: ctrl.Log.WithName("test")})
			if tc.errorExpected {
				require.Error(t, err)
				return
			}
			require.NoError(t, err)
			require.EqualValues(t, tc.expectedDs, tc.ds)
		})
	}
}

func TestTransformDriver(t *testing.T) {
	initMockK8sClients()