From 4a203dcb9368b5e5e3e6066db6fa9ec25d7e669d Mon Sep 17 00:00:00 2001 From: Michal Nowak Date: Wed, 31 Jan 2024 19:11:16 +0100 Subject: [PATCH] Support "source" parameter in isctest.query.(tcp|udp) --- bin/tests/system/isctest/query.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/bin/tests/system/isctest/query.py b/bin/tests/system/isctest/query.py index 5da1ef8423..329558d272 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)