Function TestRuntimeClasses

controllers/object_controls_test.go:1644–1795

(t *testing.T)

}

func TestRuntimeClasses(t *testing.T) {
	const (
		testNamespace = "test-namespace"
	)

	// Create scheme with required types
	scheme := runtime.NewScheme()
	require.NoError(t, nodev1.AddToScheme(scheme))
	require.NoError(t, nodev1beta1.AddToScheme(scheme))
	require.NoError(t, apiextensionsv1.AddToScheme(scheme))
	require.NoError(t, gpuv1.AddToScheme(scheme))

	// Create controller with given spec and state
	newController := func(k8s client.Client, scheme *runtime.Scheme, spec gpuv1.ClusterPolicySpec, state string) ClusterPolicyController {
		clusterPolicy := &gpuv1.ClusterPolicy{Spec: spec}
		resources := []Resources{
			{
				RuntimeClasses: []nodev1.RuntimeClass{
					{
						ObjectMeta: metav1.ObjectMeta{
							Name: "nvidia",
						},
						TypeMeta: metav1.TypeMeta{
							Kind:       "RuntimeClass",
							APIVersion: "node.k8s.io/v1",
						},
					},
					{
						ObjectMeta: metav1.ObjectMeta{
							Name: "nvidia-cdi",
						},
						TypeMeta: metav1.TypeMeta{
							Kind:       "RuntimeClass",
							APIVersion: "node.k8s.io/v1",
						},
					},
					{
						ObjectMeta: metav1.ObjectMeta{
							Name: "nvidia-legacy",
						},
						TypeMeta: metav1.TypeMeta{
							Kind:       "RuntimeClass",
							APIVersion: "node.k8s.io/v1",
						},
					},
				},
			},
		}

		return ClusterPolicyController{
			client:            k8s,
			ctx:               context.Background(),
			singleton:         clusterPolicy,
			scheme:            scheme,
			operatorNamespace: testNamespace,
			resources:         resources,
			stateNames:        []string{state},
			idx:               0,