Skip to content

Commit 0b8091e

Browse files
committed
fixup, remove the add function in TestFrame
1 parent 093bc8d commit 0b8091e

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

tests/containers/kubernetes_utils.py

+3-7
Original file line numberDiff line numberDiff line change
@@ -102,19 +102,15 @@ def defer_resource[T: ocp_resources.resource.Resource](self, resource: T, wait=F
102102
self.defer(resource, destructor)
103103
return result
104104

105-
def add[T](self, resource: T, destructor: Callable[[T], None] = None) -> T:
106-
self.defer(resource, destructor)
107-
return resource
108-
109-
def defer[T](self, resource: T, destructor: Callable[[T], None] = None) -> T:
110-
self.stack.append((resource, destructor))
105+
def defer[T](self, obj: T, destructor: Callable[[T], None] = None) -> T:
106+
self.stack.append((obj, destructor))
111107

112108
def destroy(self, wait=False):
113109
while self.stack:
114110
resource, destructor = self.stack.pop()
115111
if destructor is not None:
116112
destructor(resource)
117-
else:
113+
elif isinstance(resource, ocp_resources.resource.Resource):
118114
resource.clean_up(wait=wait)
119115

120116
def __enter__(self) -> TestFrame:

0 commit comments

Comments
 (0)