s4 dns: Test QTYPE_ALL query
authorKai Blin <kai@samba.org>
Thu, 24 Nov 2011 11:11:26 +0000 (12:11 +0100)
committerKai Blin <kai@samba.org>
Thu, 24 Nov 2011 11:35:08 +0000 (12:35 +0100)
source4/scripting/python/samba/tests/dns.py

index ad0b55a7f8ad9e9f9a670a58e1be8baea909d328..5da96bc0e0849b8bfc703e19a690baceb7903ed9 100644 (file)
@@ -115,6 +115,31 @@ class DNSTest(TestCase):
         response = self.dns_transaction_udp(p)
         self.assert_dns_rcode_equals(response, dns.DNS_RCODE_FORMERR)
 
+    def test_qtype_all_query(self):
+        "create a QTYPE_ALL query"
+        p = self.make_name_packet(dns.DNS_OPCODE_QUERY)
+        questions = []
+
+        name = "%s.%s" % (os.getenv('DC_SERVER'), self.get_dns_domain())
+        q = self.make_name_question(name, dns.DNS_QTYPE_ALL, dns.DNS_QCLASS_IN)
+        print "asking for ", q.name
+        questions.append(q)
+
+        self.finish_name_packet(p, questions)
+        response = self.dns_transaction_udp(p)
+
+        num_answers = 1
+        dc_ipv6 = os.getenv('DC_SERVER_IPV6')
+        if dc_ipv6 is not None:
+            num_answers += 1
+
+        self.assert_dns_rcode_equals(response, dns.DNS_RCODE_OK)
+        self.assert_dns_opcode_equals(response, dns.DNS_OPCODE_QUERY)
+        self.assertEquals(response.ancount, num_answers)
+        self.assertEquals(response.answers[0].rdata,
+                          os.getenv('DC_SERVER_IP'))
+        if dc_ipv6 is not None:
+            self.assertEquals(response.answers[1].rdata, dc_ipv6)
 
 if __name__ == "__main__":
     import unittest