mirror of
https://github.com/projectcapsule/capsule.git
synced 2026-05-26 19:23:21 +00:00
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
//go:build e2e
|
|
|
|
// Copyright 2020-2023 Project Capsule Authors.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package e2e
|
|
|
|
import (
|
|
"testing"
|
|
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/client-go/kubernetes/scheme"
|
|
"k8s.io/client-go/rest"
|
|
"k8s.io/utils/pointer"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/client/config"
|
|
"sigs.k8s.io/controller-runtime/pkg/envtest"
|
|
logf "sigs.k8s.io/controller-runtime/pkg/log"
|
|
"sigs.k8s.io/controller-runtime/pkg/log/zap"
|
|
|
|
capsulev1beta2 "github.com/projectcapsule/capsule/api/v1beta2"
|
|
)
|
|
|
|
// These tests use Ginkgo (BDD-style Go testing framework). Refer to
|
|
// http://onsi.github.io/ginkgo/ to learn more about Ginkgo.
|
|
|
|
var (
|
|
cfg *rest.Config
|
|
k8sClient client.Client
|
|
testEnv *envtest.Environment
|
|
)
|
|
|
|
func TestAPIs(t *testing.T) {
|
|
RegisterFailHandler(Fail)
|
|
|
|
RunSpecs(t, "Controller Suite")
|
|
}
|
|
|
|
var _ = BeforeSuite(func() {
|
|
logf.SetLogger(zap.New(zap.WriteTo(GinkgoWriter)))
|
|
|
|
By("bootstrapping test environment")
|
|
testEnv = &envtest.Environment{
|
|
UseExistingCluster: pointer.Bool(true),
|
|
}
|
|
|
|
var err error
|
|
cfg, err = testEnv.Start()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(cfg).ToNot(BeNil())
|
|
|
|
Expect(capsulev1beta2.AddToScheme(scheme.Scheme)).NotTo(HaveOccurred())
|
|
|
|
ctrlClient, err := client.New(cfg, client.Options{Scheme: scheme.Scheme})
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(ctrlClient).ToNot(BeNil())
|
|
|
|
k8sClient = &e2eClient{Client: ctrlClient}
|
|
})
|
|
|
|
var _ = AfterSuite(func() {
|
|
By("tearing down the test environment")
|
|
Expect(testEnv.Stop()).ToNot(HaveOccurred())
|
|
})
|
|
|
|
func ownerClient(owner capsulev1beta2.OwnerSpec) (cs kubernetes.Interface) {
|
|
c, err := config.GetConfig()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
c.Impersonate.Groups = []string{capsulev1beta2.GroupVersion.Group, owner.Name}
|
|
c.Impersonate.UserName = owner.Name
|
|
cs, err = kubernetes.NewForConfig(c)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
|
|
return cs
|
|
}
|