more message coverage

2 files changed, 19 insertions, 1 deletions

diff --git a/tests/test_message.py b/tests/test_message.py
index f004b92..02e7cec 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 4b8a395..f97e53b 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