mirror of
https://github.com/rancher/k3k.git
synced 2026-08-23 22:36:17 +00:00
Enrico Candino
bc3f906280
* fix status update * fix schema and default image * removed retry in controller * removed fmt
97 lines
2.3 KiB
Go
97 lines
2.3 KiB
Go
package cluster_test
|
|
|
|
import (
|
|
"context"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/go-logr/zapr"
|
|
"github.com/rancher/k3k/pkg/apis/k3k.io/v1alpha1"
|
|
"github.com/rancher/k3k/pkg/controller/cluster"
|
|
|
|
"go.uber.org/zap"
|
|
appsv1 "k8s.io/api/apps/v1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
networkingv1 "k8s.io/api/networking/v1"
|
|
rbacv1 "k8s.io/api/rbac/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/client-go/kubernetes"
|
|
ctrl "sigs.k8s.io/controller-runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/envtest"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
func TestController(t *testing.T) {
|
|
RegisterFailHandler(Fail)
|
|
RunSpecs(t, "Cluster Controller Suite")
|
|
}
|
|
|
|
var (
|
|
testEnv *envtest.Environment
|
|
k8s *kubernetes.Clientset
|
|
k8sClient client.Client
|
|
ctx context.Context
|
|
cancel context.CancelFunc
|
|
)
|
|
|
|
var _ = BeforeSuite(func() {
|
|
|
|
By("bootstrapping test environment")
|
|
testEnv = &envtest.Environment{
|
|
CRDDirectoryPaths: []string{filepath.Join("..", "..", "..", "charts", "k3k", "crds")},
|
|
ErrorIfCRDPathMissing: true,
|
|
}
|
|
cfg, err := testEnv.Start()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
k8s, err = kubernetes.NewForConfig(cfg)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
scheme := buildScheme()
|
|
k8sClient, err = client.New(cfg, client.Options{Scheme: scheme})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
ctrl.SetLogger(zapr.NewLogger(zap.NewNop()))
|
|
|
|
mgr, err := ctrl.NewManager(cfg, ctrl.Options{Scheme: scheme})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
ctx, cancel = context.WithCancel(context.Background())
|
|
err = cluster.Add(ctx, mgr, "rancher/k3k-kubelet:latest", "")
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
go func() {
|
|
defer GinkgoRecover()
|
|
err = mgr.Start(ctx)
|
|
Expect(err).NotTo(HaveOccurred(), "failed to run manager")
|
|
}()
|
|
})
|
|
|
|
var _ = AfterSuite(func() {
|
|
cancel()
|
|
|
|
By("tearing down the test environment")
|
|
err := testEnv.Stop()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
})
|
|
|
|
func buildScheme() *runtime.Scheme {
|
|
scheme := runtime.NewScheme()
|
|
|
|
err := corev1.AddToScheme(scheme)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
err = rbacv1.AddToScheme(scheme)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
err = appsv1.AddToScheme(scheme)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
err = networkingv1.AddToScheme(scheme)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
err = v1alpha1.AddToScheme(scheme)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
return scheme
|
|
}
|