diff --git a/bin/tests/system/isctest/run.py b/bin/tests/system/isctest/run.py index 09eec76127..cdf0029b8e 100644 --- a/bin/tests/system/isctest/run.py +++ b/bin/tests/system/isctest/run.py @@ -65,6 +65,18 @@ def cmd( return exc +class Dig: + def __init__(self, base_params: str = ""): + self.base_params = base_params + + def __call__(self, params: str) -> str: + """Run the dig command with the given parameters and return the decoded output.""" + return cmd( + [os.environ.get("DIG")] + f"{self.base_params} {params}".split(), + log_stdout=True, + ).stdout.decode("utf-8") + + def retry_with_timeout(func, timeout, delay=1, msg=None): start_time = time.time() while time.time() < start_time + timeout: