test/e2e: NamespacedTest helper can create/delete multiple namespaces (…

…#5928) Signed-off-by: satyazzz123 <[email protected]>
projectcontour · Nov 2, 2023 · f661786 · f661786
1 parent ac77b71
commit f661786
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/test/e2e/framework.go b/test/e2e/framework.go
@@ -283,13 +283,17 @@ type NamespacedGatewayTestBody func(ns string, gw types.NamespacedName)
 type NamespacedTestBody func(string)
 type TestBody func()
 
-func (f *Framework) NamespacedTest(namespace string, body NamespacedTestBody) {
+func (f *Framework) NamespacedTest(namespace string, body NamespacedTestBody, additionalNamespaces ...string) {
 	ginkgo.Context("with namespace: "+namespace, func() {
 		ginkgo.BeforeEach(func() {
-			f.CreateNamespace(namespace)
+			for _, ns := range append(additionalNamespaces, namespace) {
+				f.CreateNamespace(ns)
+			}
 		})
 		ginkgo.AfterEach(func() {
-			f.DeleteNamespace(namespace, false)
+			for _, ns := range append(additionalNamespaces, namespace) {
+				f.DeleteNamespace(ns, false)
+			}
 		})
 
 		body(namespace)