upstream/bind9
1
0
Fork 0
mirror of https://github.com/isc-projects/bind9.git synced 2026-05-28 04:34:54 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

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

Browse source 
(cherry picked from commit 4a203dcb93)
This commit is contained in:
Michal Nowak 2024-01-31 19:11:16 +01:00
parent 8fb49c5a8a
commit 16f950e86b
No known key found for this signature in database
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
bin/tests/system/isctest/query.py
View file

@ -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)