mirror of
https://github.com/isc-projects/bind9.git
synced 2026-06-11 02:49:59 -04:00
Add retry_with_timeout() utility function
This commit is contained in:
parent
6dd1b3ab38
commit
5694c52f52
2 changed files with 24 additions and 0 deletions
|
|
@ -13,6 +13,7 @@ from . import check
|
|||
from . import instance
|
||||
from . import query
|
||||
from . import rndc
|
||||
from . import run
|
||||
from . import log
|
||||
|
||||
# isctest.mark module is intentionally NOT imported, because it relies on
|
||||
|
|
|
|||
23
bin/tests/system/isctest/run.py
Normal file
23
bin/tests/system/isctest/run.py
Normal file
|
|
@ -0,0 +1,23 @@
|
|||
# Copyright (C) Internet Systems Consortium, Inc. ("ISC")
|
||||
#
|
||||
# SPDX-License-Identifier: MPL-2.0
|
||||
#
|
||||
# This Source Code Form is subject to the terms of the Mozilla Public
|
||||
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
||||
# file, you can obtain one at https://mozilla.org/MPL/2.0/.
|
||||
#
|
||||
# See the COPYRIGHT file distributed with this work for additional
|
||||
# information regarding copyright ownership.
|
||||
|
||||
import time
|
||||
|
||||
|
||||
def retry_with_timeout(func, timeout, delay=1, msg=None):
|
||||
start_time = time.time()
|
||||
while time.time() < start_time + timeout:
|
||||
if func():
|
||||
return
|
||||
time.sleep(delay)
|
||||
if msg is None:
|
||||
msg = f"{func.__module__}.{func.__qualname__} timed out after {timeout} s"
|
||||
assert False, msg
|
||||
Loading…
Reference in a new issue