]> git.ipfire.org Git - thirdparty/bind9.git/commitdiff
Add retry_with_timeout() utility function
authorMichal Nowak <mnowak@isc.org>
Mon, 19 Feb 2024 17:06:53 +0000 (18:06 +0100)
committerMichal Nowak <mnowak@isc.org>
Fri, 23 Feb 2024 10:04:51 +0000 (11:04 +0100)
bin/tests/system/isctest/__init__.py
bin/tests/system/isctest/run.py [new file with mode: 0644]

index 047ca7762ab95c00eb844cf2565e2ab8f35c3ea5..e0014adbba6eb817245da505a89a8a01e802dbb6 100644 (file)
@@ -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
diff --git a/bin/tests/system/isctest/run.py b/bin/tests/system/isctest/run.py
new file mode 100644 (file)
index 0000000..2efb39d
--- /dev/null
@@ -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