| /* |
| * Licensed to the Apache Software Foundation (ASF) under one or more |
| * contributor license agreements. See the NOTICE file distributed with |
| * this work for additional information regarding copyright ownership. |
| * The ASF licenses this file to You under the Apache License, Version 2.0 |
| * (the "License"); you may not use this file except in compliance with |
| * the License. You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package controllers |
| |
| import ( |
| "context" |
| "path/filepath" |
| "testing" |
| |
| . "github.com/onsi/ginkgo" |
| . "github.com/onsi/gomega" |
| |
| "k8s.io/apimachinery/pkg/runtime" |
| clientgoscheme "k8s.io/client-go/kubernetes/scheme" |
| "k8s.io/client-go/rest" |
| "sigs.k8s.io/controller-runtime/pkg/client" |
| "sigs.k8s.io/controller-runtime/pkg/envtest" |
| "sigs.k8s.io/controller-runtime/pkg/envtest/printer" |
| logf "sigs.k8s.io/controller-runtime/pkg/log" |
| "sigs.k8s.io/controller-runtime/pkg/log/zap" |
| |
| istioscheme "istio.io/client-go/pkg/clientset/versioned/scheme" |
| |
| submarineapacheorgv1 "github.com/apache/submarine/submarine-cloud-v3/api/v1" |
| //+kubebuilder:scaffold:imports |
| ) |
| |
| // 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 |
| var k8sClient client.Client |
| var testEnv *envtest.Environment |
| var scheme *runtime.Scheme |
| var ctx context.Context |
| var cancel context.CancelFunc |
| |
| func TestAPIs(t *testing.T) { |
| RegisterFailHandler(Fail) |
| |
| RunSpecsWithDefaultAndCustomReporters(t, |
| "Controller Suite", |
| []Reporter{printer.NewlineReporter{}}) |
| } |
| |
| var _ = BeforeSuite(func() { |
| // Setup ctx and cancel, which are used in BeforeSuite and AfterSuite first, |
| // in case BeforeSuite fails |
| ctx, cancel = context.WithCancel(context.TODO()) |
| |
| logf.SetLogger(zap.New(zap.WriteTo(GinkgoWriter), zap.UseDevMode(true))) |
| |
| By("bootstrapping test environment") |
| useExistingCluster := true |
| testEnv = &envtest.Environment{ |
| CRDDirectoryPaths: []string{filepath.Join("..", "config", "crd", "bases")}, |
| ErrorIfCRDPathMissing: true, |
| UseExistingCluster: &useExistingCluster, |
| } |
| |
| cfg, err := testEnv.Start() |
| Expect(err).NotTo(HaveOccurred()) |
| Expect(cfg).NotTo(BeNil()) |
| |
| By("adding required schemes") |
| scheme = runtime.NewScheme() |
| err = clientgoscheme.AddToScheme(scheme) |
| Expect(err).NotTo(HaveOccurred()) |
| err = istioscheme.AddToScheme(scheme) |
| Expect(err).NotTo(HaveOccurred()) |
| err = submarineapacheorgv1.AddToScheme(scheme) |
| Expect(err).NotTo(HaveOccurred()) |
| |
| //+kubebuilder:scaffold:scheme |
| |
| By("creating k8s client") |
| k8sClient, err = client.New(cfg, client.Options{Scheme: scheme}) |
| Expect(err).NotTo(HaveOccurred()) |
| Expect(k8sClient).NotTo(BeNil()) |
| }, 60) // timeout, seconds |
| |
| var _ = AfterSuite(func() { |
| cancel() |
| |
| By("tearing down the test environment") |
| err := testEnv.Stop() |
| Expect(err).NotTo(HaveOccurred()) |
| }) |