b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

nixos/test-driver: add wait_before_entry to polling_condition

Browse source
This commit is contained in:
Patrick Hilhorst 2022-04-27 16:23:33 +02:00
parent 57beeedbdf
commit 1db0dcdfec
No known key found for this signature in database
GPG key ID: 3BB083304DADC0FF
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
nixos/lib/test-driver/test_driver/driver.py
View file

@ -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