python:tests/dns_base: pass tkey_trans(expected_rcode)

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 27d92fa808c6617353c36fdb230504e880f4925b)

diff --git a/python/samba/tests/dns_base.py b/python/samba/tests/dns_base.py
index 41d34b0da36b34c9159e1d3c72fa690f3dbbda3d..01f980226846cdbf789fa7be2bcd9f7ba341cc0c 100644 (file)
--- a/python/samba/tests/dns_base.py
+++ b/python/samba/tests/dns_base.py
@@ -249,7 +249,8 @@ class DNSTKeyTest(DNSTest):
         self.newrecname = "tkeytsig.%s" % self.get_dns_domain()
 
     def tkey_trans(self, creds=None, algorithm_name="gss-tsig",
-                   tkey_req_in_answers=False):
+                   tkey_req_in_answers=False,
+                   expected_rcode=dns.DNS_RCODE_OK):
         "Do a TKEY transaction and establish a gensec context"
 
         if creds is None:
@@ -307,6 +308,9 @@ class DNSTKeyTest(DNSTest):
 
         (response, response_packet) =\
             self.dns_transaction_tcp(p, self.server_ip)
+        if expected_rcode != dns.DNS_RCODE_OK:
+            self.assert_echoed_dns_error(p, response, response_packet, expected_rcode)
+            return
         self.assert_dns_rcode_equals(response, dns.DNS_RCODE_OK)
 
         tkey_record = response.answers[0].rdata