WIP: early termination of test-conn-disrupt pods

Signed-off-by: Martynas Pumputis <[email protected]>
cilium · Apr 29, 2024 · 5d01c79 · 5d01c79
1 parent 658e4b5
commit 5d01c79
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/connectivity/builder/no_interrupted_connections.go b/connectivity/builder/no_interrupted_connections.go
@@ -4,6 +4,8 @@
 package builder
 
 import (
+	"context"
+
 	"github.com/cilium/cilium-cli/connectivity/check"
 	"github.com/cilium/cilium-cli/connectivity/tests"
 )
@@ -13,5 +15,11 @@ type noInterruptedConnections struct{}
 func (t noInterruptedConnections) build(ct *check.ConnectivityTest, _ map[string]string) {
 	newTest("no-interrupted-connections", ct).
 		WithCondition(func() bool { return ct.Params().IncludeConnDisruptTest }).
-		WithScenarios(tests.NoInterruptedConnections())
+		WithScenarios(tests.NoInterruptedConnections()).
+		WithFinalizer(func() error {
+			if !ct.Params().ConnDisruptTestSetup {
+				return ct.DeleteConnDisruptTestDeployment(context.Background(), ct.K8sClient())
+			}
+			return nil
+		})
 }