fix: add DeleteAll() method for TestTriggersClient

kubeshop · Oct 18, 2024 · 6e1256c · 6e1256c
1 parent f4bd2dc
commit 6e1256c
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/pkg/client/testtriggers/v1/mock_testtriggers.go b/pkg/client/testtriggers/v1/mock_testtriggers.go
diff --git a/pkg/client/testtriggers/v1/testtriggers.go b/pkg/client/testtriggers/v1/testtriggers.go
@@ -17,6 +17,7 @@ type Interface interface {
 	Create(trigger *testtriggersv1.TestTrigger) (*testtriggersv1.TestTrigger, error)
 	Update(trigger *testtriggersv1.TestTrigger) (*testtriggersv1.TestTrigger, error)
 	Delete(name, namespace string) error
+	DeleteAll(namespace string) error
 	DeleteByLabels(selector, namespace string) error
 }
 
@@ -89,6 +90,17 @@ func (s TestTriggersClient) Delete(name, namespace string) error {
 	return s.Client.Delete(context.Background(), trigger)
 }
 
+// DeleteAll delete all TestTriggers
+func (s TestTriggersClient) DeleteAll(namespace string) error {
+	if namespace == "" {
+		namespace = s.Namespace
+	}
+	u := &unstructured.Unstructured{}
+	u.SetKind("TestTrigger")
+	u.SetAPIVersion(testtriggersv1.GroupVersion.String())
+	return s.Client.DeleteAllOf(context.Background(), u, client.InNamespace(namespace))
+}
+
 // DeleteByLabels deletes TestTriggers by labels
 func (s TestTriggersClient) DeleteByLabels(selector, namespace string) error {
 	reqs, err := labels.ParseToRequirements(selector)