Jan Včelák · Daniel Salzman
--- a/tests-extra/tools/dnstest/response.py
+++ b/tests-extra/tools/dnstest/response.py
           isset(not(self.resp.ednsflags & flag_val), "NO %s FLAG" % flag)

-   def check_rr(self, section="answer", rname=None, rtype=None):
+   def _check_rr(self, expect, section=None, rname=None, rtype=None):
       """
       Check for a presence of a RR with given name and type.
       """
-       section_rrsets = getattr(self.resp, section)
+       if section is None:
+           section = "answer"
       if rname is not None:
           rname = dns.name.from_text(rname)
       if rtype is not None:
           rtype = dns.rdatatype.from_text(rtype)

+       assert section in ["answer", "authority", "additional"]