e2e: allow skipping cleanup on failure

Author: timebertt

webhosting-operator/test/e2e/e2e_suite_test.go

@@ -18,6 +18,7 @@ package e2e
 
 import (
 	"context"
+	"flag"
 	"maps"
 	"testing"
 	"time"
@@ -42,6 +43,14 @@ import (
 	webhostingv1alpha1 "github.com/timebertt/kubernetes-controller-sharding/webhosting-operator/pkg/apis/webhosting/v1alpha1"
 )
 
+var skipCleanup bool
+
+func TestMain(m *testing.M) {
+	flag.BoolVar(&skipCleanup, "skip-cleanup", false, "Skip cleanup after test failure")
+	flag.Parse()
+	m.Run()
+}
+
 func TestE2E(t *testing.T) {
 	RegisterFailHandler(Fail)
 	RunSpecs(t, "Webhosting Operator E2E Test Suite")
@@ -112,6 +121,10 @@ var _ = BeforeEach(func(ctx SpecContext) {
 	log.Info("Created test Namespace", "namespace", namespace.Name)
 
 	DeferCleanup(func(ctx SpecContext) {
+		if ctx.SpecReport().Failed() && skipCleanup {
+			Skip("Leaving state of test failure because --skip-cleanup is set to true")
+		}
+
 		By("Delete all Websites in test Namespace")
 		Expect(testClient.DeleteAllOf(ctx, &webhostingv1alpha1.Website{}, client.InNamespace(namespace.Name))).To(Succeed())
 
@@ -129,6 +142,10 @@ var _ = BeforeEach(func(ctx SpecContext) {
 	scaleController(ctx, 3)
 
 	DeferCleanup(func(ctx SpecContext) {
+		if ctx.SpecReport().Failed() && skipCleanup {
+			Skip("Leaving state of test failure because --skip-cleanup is set to true")
+		}
+
 		By("Scaling webhosting-operator")
 		scaleController(ctx, 3)
 	}, NodeTimeout(ShortTimeout))