sin/samba-mirror
1
0
Fork 0
mirror of https://github.com/samba-team/samba.git synced 2025-01-11 05:18:09 +03:00
Code

python:tests/dns_base: add self.assert_echoed_dns_error()

Browse Source 
BUG: https://bugzilla.samba.org/show_bug.cgi?id=13019

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Andrew Bartlett <abartlet@samba.org>
(cherry picked from commit ce591464cb)
This commit is contained in:
Stefan Metzmacher 2024-05-29 13:35:58 +02:00 committed by Jule Anger
parent c00749edb3
commit f7f0518b46
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
python/samba/tests/dns_base.py
View File

@ -76,6 +76,24 @@ class DNSTest(TestCaseInTempDir):
self.assertEqual(p_opcode, opcode, "Expected OPCODE %s, got %s" %
(opcode, p_opcode))
def assert_dns_flags_equals(self, packet, flags):
"Helper function to check opcode"
p_flags = packet.operation & (~(dns.DNS_OPCODE|dns.DNS_RCODE))
self.assertEqual(p_flags, flags, "Expected FLAGS %02x, got %02x" %
(flags, p_flags))
def assert_echoed_dns_error(self, request, response, response_p, rcode):
request_p = ndr.ndr_pack(request)
self.assertEqual(response.id, request.id)
self.assert_dns_rcode_equals(response, rcode)
self.assert_dns_opcode_equals(response, request.operation & dns.DNS_OPCODE)
self.assert_dns_flags_equals(response,
(request.operation | dns.DNS_FLAG_REPLY) & (~(dns.DNS_OPCODE|dns.DNS_RCODE)))
self.assertEqual(len(response_p), len(request_p))
self.assertEqual(response_p[4:], request_p[4:])
def make_name_packet(self, opcode, qid=None):
"Helper creating a dns.name_packet"
p = dns.name_packet()