From: Bob Halley <halley@dnspython.org>
Date: Fri, 19 Jun 2020 18:15:15 +0000 (-0700)
Subject: add coverage
X-Git-Tag: v2.0.0rc1~14
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=5c4f04caa7f261e81cf5b7a0269d54f994131369;p=thirdparty%2Fdnspython.git

add coverage
---

diff --git a/tests/test_message.py b/tests/test_message.py
index 6fb36693..a45ccf68 100644
--- a/tests/test_message.py
+++ b/tests/test_message.py
@@ -269,5 +269,35 @@ class MessageTestCase(unittest.TestCase):
                                        idna_codec=dns.name.IDNA_2008)
         self.assertEqual(a.answer[0], rrs)
 
+    def test_bad_section_number(self):
+        m = dns.message.make_query('foo', 'A')
+        self.assertRaises(ValueError,
+                          lambda: m.section_number(123))
+
+    def test_section_from_number(self):
+        m = dns.message.make_query('foo', 'A')
+        self.assertEqual(m.section_from_number(dns.message.QUESTION),
+                         m.question)
+        self.assertEqual(m.section_from_number(dns.message.ANSWER),
+                         m.answer)
+        self.assertEqual(m.section_from_number(dns.message.AUTHORITY),
+                         m.authority)
+        self.assertEqual(m.section_from_number(dns.message.ADDITIONAL),
+                         m.additional)
+        self.assertRaises(ValueError,
+                          lambda: m.section_from_number(999))
+
+    def test_wanting_EDNS_true_is_EDNS0(self):
+        m = dns.message.make_query('foo', 'A')
+        self.assertEqual(m.edns, -1)
+        m.use_edns(True)
+        self.assertEqual(m.edns, 0)
+
+    def test_wanting_DNSSEC_turns_on_EDNS(self):
+        m = dns.message.make_query('foo', 'A')
+        self.assertEqual(m.edns, -1)
+        m.want_dnssec()
+        self.assertEqual(m.edns, 0)
+
 if __name__ == '__main__':
     unittest.main()