From: Bob Halley <halley@dnspython.org>
Date: Sat, 25 Jul 2020 14:18:52 +0000 (-0700)
Subject: more message coverage
X-Git-Tag: v2.1.0rc1~136
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=82e4a09607df61a67bf290099b0f2be0e6a1ab7f;p=thirdparty%2Fdnspython.git

more message coverage
---

diff --git a/tests/test_message.py b/tests/test_message.py
index f004b928..02e7cec5 100644
--- a/tests/test_message.py
+++ b/tests/test_message.py
@@ -448,5 +448,11 @@ class MessageTestCase(unittest.TestCase):
         self.assertFalse(isinstance(q2, dns.update.UpdateMessage))
         self.assertEqual(q1, q2)
 
+    def test_truncated_exception_message(self):
+        q = dns.message.Message(id=1)
+        q.flags |= dns.flags.TC
+        te = dns.message.Truncated(message=q)
+        self.assertEqual(te.message(), q)
+
 if __name__ == '__main__':
     unittest.main()
diff --git a/tests/test_tsig.py b/tests/test_tsig.py
index 4b8a395c..f97e53b2 100644
--- a/tests/test_tsig.py
+++ b/tests/test_tsig.py
@@ -30,6 +30,19 @@ class TSIGTestCase(unittest.TestCase):
         with self.assertRaises(NotImplementedError):
             dns.tsig.get_context(bogus)
 
+    def test_tsig_message_properties(self):
+        m = dns.message.make_query('example', 'a')
+        self.assertIsNone(m.keyname)
+        self.assertIsNone(m.keyalgorithm)
+        self.assertIsNone(m.tsig_error)
+        m.use_tsig(keyring, keyname)
+        self.assertEqual(m.keyname, keyname)
+        self.assertEqual(m.keyalgorithm, dns.tsig.default_algorithm)
+        self.assertEqual(m.tsig_error, dns.rcode.NOERROR)
+        m = dns.message.make_query('example', 'a')
+        m.use_tsig(keyring, keyname, tsig_error=dns.rcode.BADKEY)
+        self.assertEqual(m.tsig_error, dns.rcode.BADKEY)
+
     def test_sign_and_validate(self):
         m = dns.message.make_query('example', 'a')
         m.use_tsig(keyring, keyname)
@@ -39,7 +52,6 @@ class TSIGTestCase(unittest.TestCase):
 
     def test_sign_and_validate_with_other_data(self):
         m = dns.message.make_query('example', 'a')
-        other = b'other data'
         m.use_tsig(keyring, keyname, other_data=b'other')
         w = m.to_wire()
         # not raising is passing