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 5da1ef842348dcf530a59af7a5a1caa7a2e6d266..329558d272775ea2b815a918a4c418e1b9cf6b13 100644 (file)
--- 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)