fixup, remove the add function in TestFrame

opendatahub-io · Feb 25, 2025 · 0ce8eb5 · 0ce8eb5
1 parent 50db24f
commit 0ce8eb5
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/tests/containers/kubernetes_utils.py b/tests/containers/kubernetes_utils.py
@@ -102,19 +102,15 @@ def defer_resource[T: ocp_resources.resource.Resource](self, resource: T, wait=F
         self.defer(resource, destructor)
         return result
 
-    def add[T](self, resource: T, destructor: Callable[[T], None] = None) -> T:
-        self.defer(resource, destructor)
-        return resource
-
-    def defer[T](self, resource: T, destructor: Callable[[T], None] = None) -> T:
-        self.stack.append((resource, destructor))
+    def defer[T](self, obj: T, destructor: Callable[[T], None] = None) -> T:
+        self.stack.append((obj, destructor))
 
     def destroy(self, wait=False):
         while self.stack:
             resource, destructor = self.stack.pop()
             if destructor is not None:
                 destructor(resource)
-            else:
+            elif isinstance(resource, ocp_resources.resource.Resource):
                 resource.clean_up(wait=wait)
 
     def __enter__(self) -> TestFrame: