From 1db0dcdfec9c927ce83011451c69c2ee5e78289f Mon Sep 17 00:00:00 2001 From: Patrick Hilhorst Date: Wed, 27 Apr 2022 16:23:33 +0200 Subject: nixos/test-driver: add wait_before_entry to polling_condition --- nixos/lib/test-driver/test_driver/driver.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/nixos/lib/test-driver/test_driver/driver.py b/nixos/lib/test-driver/test_driver/driver.py index e32f6810ca8..19b88975988 100644 --- a/nixos/lib/test-driver/test_driver/driver.py +++ b/nixos/lib/test-driver/test_driver/driver.py @@ -202,6 +202,7 @@ class Driver: *, seconds_interval: float = 2.0, description: Optional[str] = None, + wait_before_entry: bool = False, ) -> Union[Callable[[Callable], ContextManager], ContextManager]: driver = self @@ -212,8 +213,15 @@ class Driver: seconds_interval, description, ) + self.wait_before_entry = wait_before_entry def __enter__(self) -> None: + if self.wait_before_entry: + with rootlog.nested( + f"waiting before entering polling condition {self.condition.description}" + ): + retry(lambda x: self.condition.check(force=True)) + driver.polling_conditions.append(self.condition) def __exit__(self, a, b, c) -> None: # type: ignore -- cgit 1.4.1