From 81396681f400b871429b6fae6653a17da663fef6 Mon Sep 17 00:00:00 2001 From: Adam Rutkowski Date: Wed, 15 Jun 2022 17:43:49 +0200 Subject: [PATCH] pyocf: extend OcfCompletion 1. add optional timeout and return value to wait() 2. add method to check whether completion was triggered Signed-off-by: Adam Rutkowski --- tests/functional/pyocf/types/shared.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/tests/functional/pyocf/types/shared.py b/tests/functional/pyocf/types/shared.py index 4499eb0..dc23dda 100644 --- a/tests/functional/pyocf/types/shared.py +++ b/tests/functional/pyocf/types/shared.py @@ -103,8 +103,11 @@ class OcfCompletion: return complete - def wait(self): - self.e.wait() + def wait(self, timeout=None): + return self.e.wait(timeout=timeout) + + def completed(self): + return self.e.is_set() class OcfError(BaseException):