constellation/operators/constellation-node-operator/controllers/schemes_test.go
2022-09-22 09:10:19 +02:00

27 lines
719 B
Go

/*
Copyright (c) Edgeless Systems GmbH
SPDX-License-Identifier: AGPL-3.0-only
*/
package controllers
import (
"testing"
"k8s.io/apimachinery/pkg/runtime"
updatev1alpha1 "github.com/edgelesssys/constellation/operators/constellation-node-operator/v2/api/v1alpha1"
nodemaintenancev1beta1 "github.com/medik8s/node-maintenance-operator/api/v1beta1"
"github.com/stretchr/testify/require"
clientgoscheme "k8s.io/client-go/kubernetes/scheme"
)
func getScheme(t *testing.T) *runtime.Scheme {
scheme := runtime.NewScheme()
require.NoError(t, clientgoscheme.AddToScheme(scheme))
require.NoError(t, updatev1alpha1.AddToScheme(scheme))
require.NoError(t, nodemaintenancev1beta1.AddToScheme(scheme))
return scheme
}