stack/ee/agent/tests/main_test.go

package tests

import (
	"path/filepath"
	osRuntime "runtime"
	"testing"
	"time"

	"github.com/formancehq/stack/components/agent/internal"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
	"k8s.io/apimachinery/pkg/api/meta"

	"github.com/formancehq/operator/api/formance.com/v1beta1"
	"k8s.io/client-go/kubernetes/scheme"
	"k8s.io/client-go/rest"
	"sigs.k8s.io/controller-runtime/pkg/envtest"
)

var (
	testEnv    *envtest.Environment
	restConfig *rest.Config
	k8sClient  *rest.RESTClient
	mapper     meta.RESTMapper
)

var _ = BeforeSuite(func() {
	SetDefaultEventuallyTimeout(5 * time.Second)

	_, filename, _, _ := osRuntime.Caller(0)

	apiServer := envtest.APIServer{}
	apiServer.Configure().
		Set("service-cluster-ip-range", "10.0.0.0/20")

	Expect(v1beta1.AddToScheme(scheme.Scheme)).To(Succeed())

	testEnv = &envtest.Environment{
		CRDDirectoryPaths: []string{
			filepath.Join(filepath.Dir(filename), "..", "..", "..", "components", "operator",
				"config", "crd", "bases"),
		},
		ErrorIfCRDPathMissing: true,
		ControlPlane: envtest.ControlPlane{
			APIServer: &apiServer,
		},
		Scheme: scheme.Scheme,
	}

	var err error
	restConfig, err = testEnv.Start()
	Expect(err).NotTo(HaveOccurred())
	Expect(restConfig).NotTo(BeNil())

	restConfig.GroupVersion = &v1beta1.GroupVersion
	restConfig.NegotiatedSerializer = scheme.Codecs.WithoutConversion()
	restConfig.APIPath = "/apis"

	k8sClient, err = rest.RESTClientFor(restConfig)
	Expect(err).NotTo(HaveOccurred())

	mapper, err = internal.CreateRestMapper(restConfig)
	Expect(err).NotTo(HaveOccurred())

})

var _ = AfterSuite(func() {
	if testEnv != nil {
		Expect(testEnv.Stop()).To(BeNil())
	}
})

func TestAgent(t *testing.T) {
	RegisterFailHandler(Fail)
	RunSpecs(t, "Agent Suite")
}