123 lines
4.6 KiB
Go
123 lines
4.6 KiB
Go
/*
|
|
Copyright 2021 The Karmada Authors.
|
|
|
|
Licensed 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 framework
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"github.com/onsi/ginkgo/v2"
|
|
"github.com/onsi/gomega"
|
|
corev1 "k8s.io/api/core/v1"
|
|
apierrors "k8s.io/apimachinery/pkg/api/errors"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/client-go/kubernetes"
|
|
"k8s.io/klog/v2"
|
|
)
|
|
|
|
// CreateNamespace create Namespace.
|
|
func CreateNamespace(client kubernetes.Interface, namespace *corev1.Namespace) {
|
|
ginkgo.By(fmt.Sprintf("Creating Namespace(%s)", namespace.Name), func() {
|
|
_, err := client.CoreV1().Namespaces().Create(context.TODO(), namespace, metav1.CreateOptions{})
|
|
gomega.Expect(err).ShouldNot(gomega.HaveOccurred())
|
|
})
|
|
}
|
|
|
|
// RemoveNamespace delete Namespace.
|
|
func RemoveNamespace(client kubernetes.Interface, name string) {
|
|
ginkgo.By(fmt.Sprintf("Removing Namespace(%s)", name), func() {
|
|
err := client.CoreV1().Namespaces().Delete(context.TODO(), name, metav1.DeleteOptions{})
|
|
gomega.Expect(err).ShouldNot(gomega.HaveOccurred())
|
|
})
|
|
}
|
|
|
|
// WaitNamespacePresentOnClusterByClient wait namespace present on cluster until timeout directly by kube client.
|
|
func WaitNamespacePresentOnClusterByClient(client kubernetes.Interface, name string) {
|
|
klog.Infof("Waiting for namespace present on karmada control client")
|
|
gomega.Eventually(func(g gomega.Gomega) (bool, error) {
|
|
_, err := client.CoreV1().Namespaces().Get(context.TODO(), name, metav1.GetOptions{})
|
|
g.Expect(err).NotTo(gomega.HaveOccurred())
|
|
return true, nil
|
|
}, PollTimeout, PollInterval).Should(gomega.Equal(true))
|
|
}
|
|
|
|
// WaitNamespacePresentOnCluster wait namespace present on cluster until timeout.
|
|
func WaitNamespacePresentOnCluster(cluster, name string) {
|
|
clusterClient := GetClusterClient(cluster)
|
|
gomega.Expect(clusterClient).ShouldNot(gomega.BeNil())
|
|
|
|
klog.Infof("Waiting for namespace present on cluster(%s)", cluster)
|
|
gomega.Eventually(func(g gomega.Gomega) (bool, error) {
|
|
_, err := clusterClient.CoreV1().Namespaces().Get(context.TODO(), name, metav1.GetOptions{})
|
|
g.Expect(err).NotTo(gomega.HaveOccurred())
|
|
return true, nil
|
|
}, PollTimeout, PollInterval).Should(gomega.Equal(true))
|
|
}
|
|
|
|
// WaitNamespacePresentOnClusters wait namespace present on clusters until timeout.
|
|
func WaitNamespacePresentOnClusters(clusters []string, name string) {
|
|
ginkgo.By(fmt.Sprintf("Check if namespace(%s) present on member clusters", name), func() {
|
|
for _, clusterName := range clusters {
|
|
WaitNamespacePresentOnCluster(clusterName, name)
|
|
}
|
|
})
|
|
}
|
|
|
|
// WaitNamespaceDisappearOnCluster wait namespace disappear on cluster until timeout.
|
|
func WaitNamespaceDisappearOnCluster(cluster, name string) {
|
|
clusterClient := GetClusterClient(cluster)
|
|
gomega.Expect(clusterClient).ShouldNot(gomega.BeNil())
|
|
|
|
klog.Infof("Waiting for namespace(%s) disappear on cluster(%s)", name, cluster)
|
|
gomega.Eventually(func() bool {
|
|
_, err := clusterClient.CoreV1().Namespaces().Get(context.TODO(), name, metav1.GetOptions{})
|
|
if err == nil {
|
|
return false
|
|
}
|
|
if apierrors.IsNotFound(err) {
|
|
return true
|
|
}
|
|
|
|
klog.Errorf("Failed to get namespace(%s) on cluster(%s), err: %v", name, cluster, err)
|
|
return false
|
|
}, PollTimeout, PollInterval).Should(gomega.Equal(true))
|
|
}
|
|
|
|
// WaitNamespaceDisappearOnClusters wait namespace disappear on clusters until timeout.
|
|
func WaitNamespaceDisappearOnClusters(clusters []string, name string) {
|
|
ginkgo.By(fmt.Sprintf("Check if namespace(%s) disappear on member clusters", name), func() {
|
|
for _, clusterName := range clusters {
|
|
WaitNamespaceDisappearOnCluster(clusterName, name)
|
|
}
|
|
})
|
|
}
|
|
|
|
// UpdateNamespaceLabels update namespace's labels.
|
|
func UpdateNamespaceLabels(client kubernetes.Interface, namespace *corev1.Namespace, labels map[string]string) {
|
|
ginkgo.By(fmt.Sprintf("Updating namespace(%s)'s labels to %v", namespace.Name, labels), func() {
|
|
gomega.Eventually(func() error {
|
|
ns, err := client.CoreV1().Namespaces().Get(context.TODO(), namespace.Name, metav1.GetOptions{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
ns.Labels = labels
|
|
_, err = client.CoreV1().Namespaces().Update(context.TODO(), ns, metav1.UpdateOptions{})
|
|
return err
|
|
}, PollTimeout, PollInterval).ShouldNot(gomega.HaveOccurred())
|
|
})
|
|
}
|