From 8cb3eb5d9dd8d96985388ea5abe0b238f00c2261 Mon Sep 17 00:00:00 2001
From: Michal Nowak <mnowak@isc.org>
Date: Fri, 7 Feb 2025 11:26:04 +0100
Subject: [PATCH] Add Dig interface

(cherry picked from commit 7347abd01f38ac77916c1ce8fb1cfff91d32aa94)
---
 bin/tests/system/isctest/run.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)

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: