From: Michal Nowak <mnowak@isc.org>
Date: Wed, 31 Jan 2024 18:11:16 +0000 (+0100)
Subject: Support "source" parameter in isctest.query.(tcp|udp)
X-Git-Tag: v9.19.22~26^2~2
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=4a203dcb9368b5e5e3e6066db6fa9ec25d7e669d;p=thirdparty%2Fbind9.git

Support "source" parameter in isctest.query.(tcp|udp)
---

diff --git a/bin/tests/system/isctest/query.py b/bin/tests/system/isctest/query.py
index 5da1ef84234..329558d2727 100644
--- a/bin/tests/system/isctest/query.py
+++ b/bin/tests/system/isctest/query.py
@@ -20,16 +20,22 @@ QUERY_TIMEOUT = 10
 
 
 def udp(
-    message: dns.message.Message, ip: str, port: Optional[int] = None
+    message: dns.message.Message,
+    ip: str,
+    port: Optional[int] = None,
+    source: Optional[str] = None,
 ) -> dns.message.Message:
     if port is None:
         port = int(os.environ["PORT"])
-    return dns.query.udp(message, ip, QUERY_TIMEOUT, port=port)
+    return dns.query.udp(message, ip, QUERY_TIMEOUT, port=port, source=source)
 
 
 def tcp(
-    message: dns.message.Message, ip: str, port: Optional[int] = None
+    message: dns.message.Message,
+    ip: str,
+    port: Optional[int] = None,
+    source: Optional[str] = None,
 ) -> dns.message.Message:
     if port is None:
         port = int(os.environ["PORT"])
-    return dns.query.tcp(message, ip, QUERY_TIMEOUT, port=port)
+    return dns.query.tcp(message, ip, QUERY_TIMEOUT, port=port, source=source)