Add namespace- mappping E2E test

Signed-off-by: danfengl <danfengl@vmware.com>
2026-09-03 22:57:04 +00:00 · 2022-08-11 13:24:51 +00:00
parent fb445b3c0d
commit c8544ea212
10 changed files with 163 additions and 25 deletions
@@ -0,0 +1,101 @@
+package basic
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+
+	. "github.com/onsi/ginkgo"
+	. "github.com/onsi/gomega"
+
+	. "github.com/vmware-tanzu/velero/test/e2e"
+	. "github.com/vmware-tanzu/velero/test/e2e/test"
+	. "github.com/vmware-tanzu/velero/test/e2e/util/k8s"
+	. "github.com/vmware-tanzu/velero/test/e2e/util/kibishii"
+)
+
+type NamespaceMapping struct {
+	TestCase
+	MappedNamespaceList []string
+	kibishiiData        *KibishiiData
+}
+
+var OneNamespaceMappingTest func() = TestFunc(&NamespaceMapping{TestCase: TestCase{NSBaseName: "ns", NSIncluded: &[]string{"ns1"}}})
+var MultiNamespacesMappingTest func() = TestFunc(&NamespaceMapping{TestCase: TestCase{NSBaseName: "ns", NSIncluded: &[]string{"ns1", "ns2"}}})
+
+func (n *NamespaceMapping) Init() error {
+	n.Client = TestClientInstance
+	n.kibishiiData = &KibishiiData{2, 10, 10, 1024, 1024, 0, 2}
+
+	n.TestMsg = &TestMSG{
+		Desc:      "Backup resources with include namespace test",
+		FailedMSG: "Failed to backup with namespace include",
+		Text:      fmt.Sprintf("should backup namespaces %s", *n.NSIncluded),
+	}
+	return nil
+}
+
+func (n *NamespaceMapping) StartRun() error {
+	var mappedNS string
+	var mappedNSList []string
+
+	for index, ns := range *n.NSIncluded {
+		mappedNS = mappedNS + ns + ":" + ns + UUIDgen.String()
+		mappedNSList = append(mappedNSList, ns+UUIDgen.String())
+		if index+1 != len(*n.NSIncluded) {
+			mappedNS = mappedNS + ","
+		}
+		n.BackupName = n.BackupName + ns
+		n.RestoreName = n.RestoreName + ns
+	}
+	n.BackupName = n.BackupName + "backup-ns-mapping-" + UUIDgen.String()
+	n.RestoreName = n.RestoreName + "restore-ns-mapping-" + UUIDgen.String()
+
+	n.MappedNamespaceList = mappedNSList
+	fmt.Println(mappedNSList)
+	n.BackupArgs = []string{
+		"create", "--namespace", VeleroCfg.VeleroNamespace, "backup", n.BackupName,
+		"--include-namespaces", strings.Join(*n.NSIncluded, ","),
+		"--default-volumes-to-restic", "--wait",
+	}
+	n.RestoreArgs = []string{
+		"create", "--namespace", VeleroCfg.VeleroNamespace, "restore", n.RestoreName,
+		"--from-backup", n.BackupName, "--namespace-mappings", mappedNS,
+		"--wait",
+	}
+	return nil
+}
+func (n *NamespaceMapping) CreateResources() error {
+	n.Ctx, _ = context.WithTimeout(context.Background(), 60*time.Minute)
+
+	for index, ns := range *n.NSIncluded {
+		n.kibishiiData.Levels = len(*n.NSIncluded) + index
+		By(fmt.Sprintf("Creating namespaces ...%s\n", ns), func() {
+			Expect(CreateNamespace(n.Ctx, n.Client, ns)).To(Succeed(), fmt.Sprintf("Failed to create namespace %s", ns))
+		})
+		By("Deploy sample workload of Kibishii", func() {
+			Expect(KibishiiPrepareBeforeBackup(n.Ctx, n.Client, VeleroCfg.CloudProvider,
+				ns, VeleroCfg.RegistryCredentialFile, VeleroCfg.Features,
+				VeleroCfg.KibishiiDirectory, false, n.kibishiiData)).To(Succeed())
+		})
+	}
+	return nil
+}
+
+func (n *NamespaceMapping) Verify() error {
+	n.Ctx, _ = context.WithTimeout(context.Background(), 60*time.Minute)
+	for index, ns := range n.MappedNamespaceList {
+		n.kibishiiData.Levels = len(*n.NSIncluded) + index
+		By(fmt.Sprintf("Verify workload %s after restore ", ns), func() {
+			Expect(KibishiiVerifyAfterRestore(n.Client, ns,
+				n.Ctx, n.kibishiiData)).To(Succeed(), "Fail to verify workload after restore")
+		})
+	}
+	for _, ns := range *n.NSIncluded {
+		By(fmt.Sprintf("Verify namespace %s for backup is no longer exist after restore with namespace mapping", ns), func() {
+			Expect(NamespaceShouldNotExist(n.Ctx, n.Client, ns)).To(Succeed())
+		})
+	}
+	return nil
+}