From 0ce8eb5d495422b346b62ce6c67a682157103201 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jiri=20Dan=C4=9Bk?= Date: Mon, 24 Feb 2025 15:52:26 +0100 Subject: [PATCH] fixup, remove the add function in TestFrame --- tests/containers/kubernetes_utils.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/tests/containers/kubernetes_utils.py b/tests/containers/kubernetes_utils.py index 66829829f..4f437a312 100644 --- 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: