| // 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 v1alpha1 |
| |
| import ( |
| "context" |
| "crypto/tls" |
| "fmt" |
| "net" |
| "path/filepath" |
| "testing" |
| "time" |
| |
| . "github.com/onsi/ginkgo" |
| . "github.com/onsi/gomega" |
| |
| admissionv1beta1 "k8s.io/api/admission/v1beta1" |
| //+kubebuilder:scaffold:imports |
| "k8s.io/apimachinery/pkg/runtime" |
| "k8s.io/client-go/rest" |
| ctrl "sigs.k8s.io/controller-runtime" |
| "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" |
| ) |
| |
| // 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 ctx context.Context |
| var cancel context.CancelFunc |
| |
| func TestAPIs(t *testing.T) { |
| RegisterFailHandler(Fail) |
| |
| RunSpecsWithDefaultAndCustomReporters(t, |
| "Webhook Suite", |
| []Reporter{printer.NewlineReporter{}}) |
| } |
| |
| var _ = BeforeSuite(func() { |
| logf.SetLogger(zap.New(zap.WriteTo(GinkgoWriter), zap.UseDevMode(true))) |
| |
| ctx, cancel = context.WithCancel(context.TODO()) |
| |
| By("bootstrapping test environment") |
| testEnv = &envtest.Environment{ |
| CRDDirectoryPaths: []string{filepath.Join("..", "..", "config", "crd", "bases")}, |
| ErrorIfCRDPathMissing: false, |
| WebhookInstallOptions: envtest.WebhookInstallOptions{ |
| Paths: []string{filepath.Join("..", "..", "config", "webhook")}, |
| }, |
| } |
| |
| cfg, err := testEnv.Start() |
| Expect(err).NotTo(HaveOccurred()) |
| Expect(cfg).NotTo(BeNil()) |
| |
| scheme := runtime.NewScheme() |
| err = AddToScheme(scheme) |
| Expect(err).NotTo(HaveOccurred()) |
| |
| err = admissionv1beta1.AddToScheme(scheme) |
| Expect(err).NotTo(HaveOccurred()) |
| |
| //+kubebuilder:scaffold:scheme |
| |
| k8sClient, err = client.New(cfg, client.Options{Scheme: scheme}) |
| Expect(err).NotTo(HaveOccurred()) |
| Expect(k8sClient).NotTo(BeNil()) |
| |
| // start webhook server using Manager |
| webhookInstallOptions := &testEnv.WebhookInstallOptions |
| mgr, err := ctrl.NewManager(cfg, ctrl.Options{ |
| Scheme: scheme, |
| Host: webhookInstallOptions.LocalServingHost, |
| Port: webhookInstallOptions.LocalServingPort, |
| CertDir: webhookInstallOptions.LocalServingCertDir, |
| LeaderElection: false, |
| MetricsBindAddress: "0", |
| }) |
| Expect(err).NotTo(HaveOccurred()) |
| |
| err = (&DSMaster{}).SetupWebhookWithManager(mgr) |
| Expect(err).NotTo(HaveOccurred()) |
| |
| //+kubebuilder:scaffold:webhook |
| |
| go func() { |
| defer GinkgoRecover() |
| err = mgr.Start(ctx) |
| Expect(err).NotTo(HaveOccurred()) |
| }() |
| |
| // wait for the webhook server to get ready |
| dialer := &net.Dialer{Timeout: time.Second} |
| addrPort := fmt.Sprintf("%s:%d", webhookInstallOptions.LocalServingHost, webhookInstallOptions.LocalServingPort) |
| Eventually(func() error { |
| conn, err := tls.DialWithDialer(dialer, "tcp", addrPort, &tls.Config{InsecureSkipVerify: true}) |
| if err != nil { |
| return err |
| } |
| conn.Close() |
| return nil |
| }).Should(Succeed()) |
| |
| }, 60) |
| |
| var _ = AfterSuite(func() { |
| cancel() |
| By("tearing down the test environment") |
| err := testEnv.Stop() |
| Expect(err).NotTo(HaveOccurred()) |
| }) |