enable TSIG text format testing

1 files changed, 11 insertions, 2 deletions

diff --git a/tests/test_tsig.py b/tests/test_tsig.py
index b8a38b4..a016cf8 100644
--- a/tests/test_tsig.py
+++ b/tests/test_tsig.py
@@ -246,7 +246,7 @@ class TSIGTestCase(unittest.TestCase):
     def test_hmac_sha512_256(self):
         self._test_truncated_algorithm(dns.tsig.HMAC_SHA512_256, 256)
 
-    def text_format(self):
+    def _test_text_format(self, alg):
         key = dns.tsig.Key('foo', b'abcdefg', algorithm=alg)
         q = dns.message.make_query('example', 'a')
         q.use_tsig(key)
@@ -257,10 +257,19 @@ class TSIGTestCase(unittest.TestCase):
         self.assertEqual(tsig2, q.tsig[0])
 
         q = dns.message.make_query('example', 'a')
-        q.use_tsig(key, other_data='abc')
+        q.use_tsig(key, other_data=b'abc')
         q.use_tsig(key)
         _ = q.to_wire()
 
         text = q.tsig[0].to_text()
         tsig2 = dns.rdata.from_text('ANY', 'TSIG', text)
         self.assertEqual(tsig2, q.tsig[0])
+
+    def test_text_hmac_sha256_128(self):
+        self._test_text_format(dns.tsig.HMAC_SHA256_128)
+
+    def test_text_hmac_sha384_192(self):
+        self._test_text_format(dns.tsig.HMAC_SHA384_192)
+
+    def test_text_hmac_sha512_256(self):
+        self._test_text_format(dns.tsig.HMAC_SHA512_256)